96 lines
2.9 KiB
Python
96 lines
2.9 KiB
Python
from fastapi import FastAPI, Request
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from slowapi import _rate_limit_exceeded_handler
|
|
from slowapi.errors import RateLimitExceeded
|
|
|
|
import logging
|
|
|
|
# Set up logging to stdout
|
|
logging.basicConfig(level=logging.INFO)
|
|
logger = logging.getLogger("ai_gateway")
|
|
|
|
from app.api.router import api_router
|
|
from app.core.config import settings
|
|
from app.core.limiter import limiter
|
|
from app.core.database import engine, Base
|
|
from app.models import module # Ensure models are loaded
|
|
|
|
# Create tables
|
|
Base.metadata.create_all(bind=engine)
|
|
|
|
from fastapi.staticfiles import StaticFiles
|
|
from app.api.endpoints import auth
|
|
from app.api import admin_backend
|
|
|
|
def create_application() -> FastAPI:
|
|
# Disable docs and redoc in production
|
|
docs_url = "/docs" if settings.ENVIRONMENT == "development" else None
|
|
redoc_url = "/redoc" if settings.ENVIRONMENT == "development" else None
|
|
|
|
application = FastAPI(
|
|
title=settings.PROJECT_NAME,
|
|
docs_url=docs_url,
|
|
redoc_url=redoc_url
|
|
)
|
|
|
|
|
|
# Mount Static Files
|
|
application.mount("/static", StaticFiles(directory="app/static"), name="static")
|
|
|
|
# Debug Logger Middleware
|
|
@application.middleware("http")
|
|
async def log_requests(request: Request, call_next):
|
|
logger.info(f"DEBUG: Incoming {request.method} {request.url}")
|
|
logger.info(f"DEBUG: Origin: {request.headers.get('origin')}")
|
|
response = await call_next(request)
|
|
logger.info(f"DEBUG: Status Code: {response.status_code}")
|
|
return response
|
|
|
|
# Set up fully permissive CORS for standalone/tunnelling mode
|
|
application.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# Set up Rate Limiter
|
|
application.state.limiter = limiter
|
|
|
|
@application.exception_handler(RateLimitExceeded)
|
|
async def custom_rate_limit_exceeded_handler(request, exc):
|
|
from fastapi.responses import JSONResponse
|
|
return JSONResponse(
|
|
status_code=429,
|
|
content={
|
|
"error": "Too Many Requests",
|
|
"message": "Chill out! You've reached the rate limit. Please wait a moment before trying again."
|
|
}
|
|
)
|
|
|
|
# Include routes
|
|
application.include_router(api_router, prefix="/api/v1")
|
|
application.include_router(auth.router, prefix="/auth", tags=["auth"])
|
|
application.include_router(admin_backend.router, prefix="/internal/admin", tags=["internal-admin"])
|
|
|
|
return application
|
|
|
|
app = create_application()
|
|
|
|
from fastapi.responses import FileResponse
|
|
import os
|
|
|
|
@app.get("/admin")
|
|
async def admin_panel():
|
|
return FileResponse("app/static/admin.html")
|
|
|
|
@app.get("/api/v1/health")
|
|
async def health_check():
|
|
return {"status": "ok", "message": "Global AI Gateway is reachable"}
|
|
|
|
@app.get("/")
|
|
async def root():
|
|
return FileResponse("app/static/index.html")
|
|
|