app.py

import os
import logging
from ui import create_interface
from huggingface_hub import login

# Setup logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# Environment variables check
required_vars = ["HF_TOKEN", "GROQ_API_KEY"]
missing_vars = [var for var in required_vars if not os.getenv(var)]
if missing_vars:
    raise ValueError(f"Missing required environment variables: {', '.join(missing_vars)}")

# Hugging Face login
try:
    login(token=os.getenv("HF_TOKEN"))
    logger.info("Successfully logged in to Hugging Face")
except Exception as e:
    logger.error(f"Failed to login to Hugging Face: {str(e)}")
    raise

if __name__ == "__main__":
    try:
        demo = create_interface()
        demo.queue(max_size=5).launch()
    except Exception as e:
        logger.error(f"Application startup error: {str(e)}")
        raise