config.py

# Configuration file for Flask application

import os

class Config:
    """Base configuration"""
    SECRET_KEY = os.environ.get('SECRET_KEY') or 'dev-secret-key-change-in-production'
    MAX_CONTENT_LENGTH = 16 * 1024 * 1024  # 16MB max file size
    UPLOAD_FOLDER = 'static/uploads'
    ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif', 'bmp', 'webp'}
    
    # TensorFlow Configuration
    TF_CPP_MIN_LOG_LEVEL = '2'  # Reduce TensorFlow logging
    INTRA_OP_PARALLELISM_THREADS = 4
    INTER_OP_PARALLELISM_THREADS = 4
    
    # Model configuration
    MODEL_PATH = 'model/cifar10_advanced_cnn.h5'
    BATCH_SIZE = 32  # For inference
    
    # Class Names
    CLASS_NAMES = [
        '0: airplane', '1: automobile', '2: bird', '3: cat', '4: deer',
        '5: dog', '6: frog', '7: horse', '8: ship', '9: truck'
    ]


class DevelopmentConfig(Config):
    """Development configuration"""
    DEBUG = True


class ProductionConfig(Config):
    """Production configuration"""
    DEBUG = False
    # In production, always set SECRET_KEY via environment variable
    SECRET_KEY = os.environ.get('SECRET_KEY')
    if not SECRET_KEY:
        raise ValueError("No SECRET_KEY set for production")


# Configuration dictionary
config = {
    'development': DevelopmentConfig,
    'production': ProductionConfig,
    'default': DevelopmentConfig
}