app.py

"""Hugging Face Spaces wrapper for the multi-agent system."""
import os
import sys
from pathlib import Path

# Set environment variables from HF Secrets
os.environ["OPENROUTER_API_KEY"] = os.getenv("OPENROUTER_API_KEY", "")
os.environ["HUGGINGFACE_TOKEN"] = os.getenv("HUGGINGFACE_TOKEN", "")
os.environ["HUGGINGFACE_DATASET"] = os.getenv("HUGGINGFACE_DATASET", "factorstudios/Pipeline")
os.environ["OPENROUTER_BASE_URL"] = os.getenv("OPENROUTER_BASE_URL", "https://openrouter.ai/api/v1")
os.environ["MODEL_NAME"] = os.getenv("MODEL_NAME", "poolside/laguna-m.1:free")

# Import and run the FastAPI app
from main import app

if __name__ == "__main__":
    import uvicorn
    
    # Get port from environment or use default
    port = int(os.getenv("PORT", 7860))
    
    # Run the server
    uvicorn.run(
        app,
        host="0.0.0.0",
        port=port,
        log_level="info",
    )