Upload app/main.py with huggingface_hub

app/main.py

@@ -0,0 +1,134 @@
+import sys
+import os
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+from fastapi import FastAPI, Request, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi.middleware.gzip import GZipMiddleware
+from fastapi.middleware.trustedhost import TrustedHostMiddleware
+from fastapi.responses import JSONResponse
+from fastapi.staticfiles import StaticFiles
+from slowapi import Limiter, _rate_limit_exceeded_handler
+from slowapi.errors import RateLimitExceeded
+from slowapi.util import get_remote_address
+from starlette import status
+import time
+import logging
+from api import predict, train, status, dashboard, batch_predict, playground
+from api.monitor import router as monitor
+from core.config import settings
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    handlers=[
+        logging.FileHandler('app.log'),
+        logging.StreamHandler()
+    ]
+)
+logger = logging.getLogger(__name__)
+
+# Configure rate limiting
+limiter = Limiter(key_func=get_remote_address)
+
+# Custom rate limit error handler
+async def custom_rate_limit_handler(request: Request, exc: RateLimitExceeded) -> JSONResponse:
+    response = JSONResponse(
+        status_code=status.HTTP_429_TOO_MANY_REQUESTS,
+        content={
+            "error": "Rate limit exceeded",
+            "message": "Too many requests. Please wait a moment before trying again.",
+            "detail": f"Rate limit: {exc.detail}",
+            "retry_after": "Please wait 5-10 seconds before your next request",
+            "tip": "For testing, try clicking more slowly or wait between requests"
+        }
+    )
+    return response
+
+app = FastAPI(
+    title="Hallucination Detection System", 
+    version="1.0.0",
+    description="Production-ready hallucination detection API with ML model"
+)
+
+# Set up rate limiting
+app.state.limiter = limiter
+app.add_exception_handler(RateLimitExceeded, custom_rate_limit_handler)
+
+# Add security and performance middleware
+app.add_middleware(GZipMiddleware, minimum_size=1000)
+app.add_middleware(
+    TrustedHostMiddleware, 
+    allowed_hosts=["localhost", "127.0.0.1", "*.localhost", "testserver"]
+)
+
+# Request timing and monitoring middleware
+@app.middleware("http")
+async def add_process_time_header(request: Request, call_next):
+    start_time = time.time()
+    try:
+        response = await call_next(request)
+        process_time = time.time() - start_time
+        response.headers["X-Process-Time"] = str(process_time)
+        logger.info(f"Request: {request.method} {request.url.path} - Time: {process_time:.3f}s - Status: {response.status_code}")
+        return response
+    except Exception as e:
+        process_time = time.time() - start_time
+        logger.error(f"Request failed: {request.method} {request.url.path} - Time: {process_time:.3f}s - Error: {str(e)}")
+        return JSONResponse(
+            status_code=500,
+            content={"detail": "Internal server error", "request_id": str(time.time())},
+            headers={"X-Process-Time": str(process_time)}
+        )
+
+# Static files
+frontend_path = os.path.join(os.path.dirname(__file__), "frontend")
+app.mount("/frontend", StaticFiles(directory=frontend_path), name="frontend")
+
+# CORS with restricted origins for security
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["http://localhost:8000", "http://127.0.0.1:8000", "http://localhost:3000"],
+    allow_credentials=True,
+    allow_methods=["GET", "POST", "PUT", "DELETE"],
+    allow_headers=["*"],
+)
+
+# Include routers
+app.include_router(predict, prefix="/api", tags=["predictions"])
+app.include_router(batch_predict, prefix="/api", tags=["batch_predictions"])
+app.include_router(train, prefix="/api", tags=["training"])
+app.include_router(status, prefix="/api", tags=["status"])
+app.include_router(dashboard, prefix="/dashboard", tags=["dashboard"])
+app.include_router(monitor, prefix="/api", tags=["monitoring"])
+app.include_router(playground, prefix="", tags=["playground"])
+
+# Competition API
+try:
+    from api.competition import router as competition_router
+    app.include_router(competition_router, prefix="/api", tags=["competition"])
+    logger.info("Competition API loaded successfully")
+except ImportError as e:
+    logger.warning(f"Competition API not available: {e}")
+
+@app.get("/", tags=["health"])
+async def root():
+    return {
+        "message": "Hallucination Detection System is running",
+        "version": "1.0.0",
+        "status": "healthy",
+        "timestamp": time.time()
+    }
+
+@app.get("/health", tags=["health"])
+async def health_check():
+    return {
+        "status": "healthy",
+        "timestamp": time.time(),
+        "service": "hallucination-detector"
+    }
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)