from fastapi import FastAPI from fastapi.middleware.cors import CORSMiddleware import logging import os from dotenv import load_dotenv # Load environment variables load_dotenv() from app.database import init_db # Import init_db from app.auth import router as auth_router from app.upload import router as upload_router from app.dashboard import router as dashboard_router from app.agent.custom_chatbot import router as custom_chatbot_router from app.feeds import router as feeds_router # from app.routes import pdf_ingestions # Initialize logger logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) app = FastAPI( title="Dubsway Video AI", description="Production-ready API for auth, video upload, and analysis with Groq LLM integration", version="1.0.0", docs_url="/docs", # Optional: secure this in prod redoc_url=None, ) # Allow frontend (adjust in prod!) app.add_middleware( CORSMiddleware, allow_origins=["*"], # REPLACE with frontend URL in production! allow_credentials=True, allow_methods=["*"], allow_headers=["*"], ) # Include API routes app.include_router(auth_router, prefix="/api", tags=["Auth"]) app.include_router(upload_router, prefix="/api", tags=["Upload"]) app.include_router(dashboard_router, prefix="/api", tags=["Dashboard"]) app.include_router(custom_chatbot_router, prefix="/api", tags=["Custom Chatbot"]) app.include_router(feeds_router, prefix="/api", tags=["Feeds"]) # app.include_router(pdf_ingestion.router, prefix="/api", tags=["PDF Ingestion"]) @app.get("/") async def root(): """Root endpoint for Hugging Face Spaces""" return { "message": "🎥 Dubsway Video AI API", "version": "1.0.0", "docs": "/docs", "status": "running" } @app.on_event("startup") async def startup_event(): logger.info("✅ FastAPI app started") logger.info(f"Environment: {os.getenv('ENVIRONMENT', 'development')}") logger.info(f"Database URL configured: {'DATABASE_URL' in os.environ}") await init_db() @app.on_event("shutdown") async def shutdown_event(): logger.info("🛑 FastAPI app shutdown")