config.py

"""
Configuration file for the Inclusive World Curriculum Assistant
"""

# Model Configuration
MODEL_CONFIG = {
    "model_name": "microsoft/DialoGPT-medium",
    "embedding_model": "sentence-transformers/all-MiniLM-L6-v2",
    "max_new_tokens": 256,
    "temperature": 0.7,
    "top_p": 0.95,
    "repetition_penalty": 1.15,
    "torch_dtype": "float16",
    "device_map": "auto",
    "trust_remote_code": True
}

# Vector Database Configuration
VECTOR_DB_CONFIG = {
    "chunk_size": 1000,
    "chunk_overlap": 200,
    "persist_directory": "./chroma_db",
    "search_kwargs": {"k": 3}
}

# File Processing Configuration
FILE_CONFIG = {
    "slides_directory": "Slides",
    "supported_formats": [".pdf"],
    "max_preview_length": 500
}

# UI Configuration
UI_CONFIG = {
    "page_title": "Inclusive World Curriculum Assistant",
    "page_icon": "🎓",
    "layout": "wide",
    "initial_sidebar_state": "expanded"
}

# Curriculum Topics (for quick access)
CURRICULUM_TOPICS = [
    "Variables and Data Types",
    "Control Structures (if/else)",
    "Loops (for, while)",
    "Functions and Methods",
    "Arrays and Lists",
    "Object-Oriented Programming",
    "Error Handling",
    "File Operations",
    "Web Development Basics",
    "Database Fundamentals",
    "API Development",
    "Testing Strategies"
]