Spaces:
Runtime error
Runtime error
| import os | |
| import logging | |
| from dotenv import load_dotenv | |
| import dspy | |
| from pathlib import Path | |
| from fastapi import FastAPI | |
| from fastapi.middleware.cors import CORSMiddleware | |
| from starlette.middleware.sessions import SessionMiddleware | |
| # Load environment from backend/.env BEFORE importing modules that read env | |
| _ENV_PATH = Path(__file__).resolve().parent.parent / ".env" | |
| load_dotenv(dotenv_path=_ENV_PATH, override=False) | |
| logging.basicConfig( | |
| level=logging.WARNING, # Changed from INFO to WARNING to reduce logging | |
| format="%(asctime)s - %(name)s - %(levelname)s - %(message)s" | |
| ) | |
| from .core.db import Base, engine | |
| from .routes.health import router as health_router | |
| from .routes.auth import router as auth_router | |
| from .routes.google import router as google_router | |
| from .routes.payment import router as stripe_router | |
| from .routes.export import router as export_router | |
| from .routes.credits import router as credits_router | |
| from .routes.plans import router as plans_router | |
| from .routes.forms import router as forms_router | |
| from .routes.responses import router as responses_router | |
| from .routes.analytics import router as analytics_router | |
| from .routes.users import router as users_router | |
| from .routes.uploads import router as uploads_router | |
| app = FastAPI(title="Backend", version="0.1.0") | |
| default_model = os.getenv("DEFAULT_MODEL", "").lower() | |
| if "anthropic" in default_model: | |
| provider = "ANTHROPIC" | |
| elif "openai" in default_model: | |
| provider = "OPENAI" | |
| elif "gemini" in default_model: | |
| provider = "GEMINI" | |
| else: | |
| provider = "UNKNOWN" | |
| default_lm = dspy.LM(default_model, max_tokens=3200,api_key=os.getenv(provider+'_API_KEY'), temperature=1, cache=False) | |
| dspy.configure(lm=default_lm) | |
| # CORS middleware - allow frontend to access backend | |
| frontend_url = os.getenv("FRONTEND_URL", "http://localhost:5173") | |
| # Support multiple origins (local dev + production) | |
| allowed_origins = [origin.strip() for origin in frontend_url.split(",") if origin.strip()] | |
| app.add_middleware( | |
| CORSMiddleware, | |
| allow_origins=allowed_origins, | |
| allow_credentials=True, | |
| allow_methods=["*"], | |
| allow_headers=["*"], | |
| ) | |
| # Session support for OAuth (Authlib requires request.session) | |
| app.add_middleware( | |
| SessionMiddleware, | |
| secret_key=os.getenv("SESSION_SECRET", "change-this-session-secret"), | |
| same_site="lax", | |
| https_only=bool(int(os.getenv("SESSION_HTTPS_ONLY", "0"))), | |
| ) | |
| from .schemas.auth import LoginRequest | |
| # Optional debug route to verify OAuth env at runtime (masked) | |
| def auth_debug(): | |
| cid = os.getenv("GOOGLE_CLIENT_ID", "") | |
| rid = os.getenv("GOOGLE_REDIRECT_URI", "") | |
| return { | |
| "GOOGLE_CLIENT_ID_present": bool(cid), | |
| "GOOGLE_CLIENT_ID_prefix": cid[:10] + ("..." if cid else ""), | |
| "GOOGLE_REDIRECT_URI": rid, | |
| } | |
| app.include_router(health_router) | |
| # Auth routes | |
| app.include_router(auth_router) | |
| # Payment routes | |
| app.include_router(google_router) | |
| app.include_router(stripe_router) | |
| # Export routes | |
| app.include_router(export_router) | |
| # Credits routes | |
| app.include_router(credits_router) | |
| # Plans routes | |
| app.include_router(plans_router) | |
| # Form routes | |
| app.include_router(forms_router) | |
| app.include_router(responses_router) | |
| app.include_router(analytics_router) | |
| app.include_router(users_router) | |
| app.include_router(uploads_router) | |
| # Initialize DB | |
| if os.getenv("AUTO_MIGRATE", "1") == "1": | |
| Base.metadata.create_all(bind=engine) | |
| # Initialize default subscription plans | |
| from .services.plan_service import plan_service | |
| from .core.db import SessionLocal | |
| try: | |
| db = SessionLocal() | |
| plan_service.initialize_default_plans(db, force=False) | |
| db.close() | |
| logging.info("Default subscription plans initialized") | |
| except Exception as e: | |
| logging.error(f"Failed to initialize subscription plans: {e}") | |