Upload 40 files

ai_voice_detection/api/main.py

@@ -43,8 +43,28 @@ class ErrorResponse(BaseModel):
     message: str
 
 
+from fastapi import FastAPI, HTTPException, Security, Request
+from fastapi.exceptions import RequestValidationError
+from fastapi.responses import JSONResponse
+from fastapi.security import APIKeyHeader
+
 app = FastAPI(title="AI-Generated Voice Detection API")
 
+@app.exception_handler(HTTPException)
+async def http_exception_handler(request: Request, exc: HTTPException):
+    return JSONResponse(
+        status_code=exc.status_code,
+        content={"status": "error", "message": str(exc.detail)},
+    )
+
+@app.exception_handler(RequestValidationError)
+async def validation_exception_handler(request: Request, exc: RequestValidationError):
+    # Simplify validation error message
+    return JSONResponse(
+        status_code=400,
+        content={"status": "error", "message": "Malformed request or invalid data"},
+    )
+
 
 # --- Dependencies ---
 async def verify_api_key(api_key: str = Security(api_key_header)):