Update main.py

main.py

@@ -1,87 +1,76 @@
-import base64
-import binascii
-from fastapi import FastAPI, HTTPException, Depends, Header
-from schemas import AudioInput, DetectionResult
-from model_service import get_model_service, ModelService
-
-app = FastAPI(
-    title="AI Voice Detection API",
-    description="Detects whether a voice sample is AI-generated or Human-spoken.",
-    version="1.0.0"
-)
-
-@app.on_event("startup")
-async def startup_event():
-    # Initialize model on startup
-    get_model_service()
-
-API_KEY = "my_secret_key_123"  # Simple hardcoded key for submission
-
-async def verify_api_key(x_api_key: str = Header(...)):
-    if x_api_key != API_KEY:
-        raise HTTPException(status_code=401, detail="Invalid API Key")
-    return x_api_key
-
-from fastapi import FastAPI, HTTPException, Depends, Header, Request
-
-# ... (Previous imports stay, schema can stay unused or updated)
-
-@app.post("/detect", response_model=DetectionResult)
-async def detect_voice(
-    request: Request,
-    service: ModelService = Depends(get_model_service),
-    api_key: str = Depends(verify_api_key)
-):
-    try:
-        # 1. Parse JSON body manually to be flexible
-        body = await request.json()
-        print(f"DEBUG: Received Body Keys: {list(body.keys())}")
-        
-        # 2. Look for the base64 string in common keys using a priority list
-        # OR just grab the first string value that looks like base64
-        audio_b64 = None
-        
-        # Check specific keys first
-        possible_keys = ["audio_base64", "audio", "data", "file", "encoded_audio", "mp3"]
-        for k in possible_keys:
-            if k in body and body[k]:
-                audio_b64 = body[k]
-                print(f"DEBUG: Found audio in key: '{k}'")
-                break
-        
-        # Fallback: Search ALL values for a long string
-        if not audio_b64:
-            for k, v in body.items():
-                if isinstance(v, str) and len(v) > 100:
-                    audio_b64 = v
-                    print(f"DEBUG: Found audio in generic key: '{k}'")
-                    break
-        
-        if not audio_b64:
-             raise HTTPException(status_code=422, detail=f"Could not find audio data. Received keys: {list(body.keys())}")
-
-        # Decode Base64 string
-        # Handle data URI scheme if present (e.g. "data:audio/mp3;base64,...")
-        if "," in audio_b64:
-            audio_b64 = audio_b64.split(",")[1]
-
-        audio_bytes = base64.b64decode(audio_b64)
-    except Exception as e:
-        print(f"Error parsing request: {e}")
-        raise HTTPException(status_code=400, detail=f"Invalid Request: {str(e)}")
-
-    try:
-        label, confidence = service.predict(audio_bytes)
-        return DetectionResult(
-            label=label,
-            confidence=confidence,
-            message="Analysis successful"
-        )
-    except ValueError as ve:
-        raise HTTPException(status_code=400, detail=str(ve))
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Internal Server Error: {str(e)}")
-
-@app.get("/")
-def read_root():
-    return {"message": "AI Voice Detection API is running. Use /detect endpoint."}
+import base64
+import binascii
+from fastapi import FastAPI, HTTPException, Depends, Header, Request, Body
+from schemas import AudioInput, DetectionResult
+from model_service import get_model_service, ModelService
+
+app = FastAPI(
+    title="AI Voice Detection API",
+    description="Detects whether a voice sample is AI-generated or Human-spoken.",
+    version="1.0.0"
+)
+
+@app.on_event("startup")
+async def startup_event():
+    # Pre-load model on startup
+    get_model_service()
+
+API_KEY = "my_secret_key_123"
+
+async def verify_api_key(x_api_key: str = Header(...)):
+    if x_api_key != API_KEY:
+        raise HTTPException(status_code=401, detail="Invalid API Key")
+    return x_api_key
+
+@app.post("/detect", response_model=DetectionResult)
+async def detect_voice(
+    # 'Body' forces the input box to appear in Swagger/Docs
+    body: dict = Body(..., example={"audio_base64": "PASTE_YOUR_BASE64_HERE"}),
+    service: ModelService = Depends(get_model_service),
+    api_key: str = Depends(verify_api_key)
+):
+    try:
+        # 1. Universal Search: find the base64 string in the body
+        audio_b64 = None
+        possible_keys = ["audio_base64", "audio", "data", "file", "encoded_audio", "mp3"]
+        
+        # Check priority keys
+        for k in possible_keys:
+            if k in body and body[k]:
+                audio_b64 = body[k]
+                break
+        
+        # Fallback: search all values for a long string if no common key is found
+        if not audio_b64:
+            for k, v in body.items():
+                if isinstance(v, str) and len(v) > 100:
+                    audio_b64 = v
+                    break
+        
+        if not audio_b64:
+             raise HTTPException(status_code=422, detail="Could not find audio data in your request.")
+
+        # 2. Cleanup Base64 (remove data URI prefixes if they exist)
+        if "," in audio_b64:
+            audio_b64 = audio_b64.split(",")[1]
+
+        # 3. Decode
+        audio_bytes = base64.b64decode(audio_b64)
+        
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f"Request formatting error: {str(e)}")
+
+    try:
+        # 4. Predict
+        label, confidence = service.predict(audio_bytes)
+        return DetectionResult(
+            label=label,
+            confidence=confidence,
+            message="Analysis successful"
+        )
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Model error: {str(e)}")
+
+@app.get("/")
+def read_root():
+    return {"message": "AI Voice Detection API is running. Use /detect endpoint."}