Update main.py

main.py

@@ -12,7 +12,7 @@ app = FastAPI(
 
 @app.on_event("startup")
 async def startup_event():
-    # EXTREMELY IMPORTANT: Instant startup to prevent 504 Timeout
+    # Instant startup to prevent 504 Timeout
     print("API is starting up...")
 
 API_KEY = "my_secret_key_123"
@@ -25,34 +25,37 @@ async def verify_api_key(x_api_key: str = Header(...)):
 @app.post("/detect", response_model=DetectionResult)
 async def detect_voice(
     request: Request,
-    # This line ensures the "Request body" box appears in your /docs test page
     payload: dict = Body(..., example={"audioBase64": "PASTE_BASE64_HERE"}),
     service: ModelService = Depends(get_model_service),
     api_key: str = Depends(verify_api_key)
 ):
     try:
-        # 1. Flexible parsing for portal compatibility (audioBase64, audio_base_64, etc.)
-        body = await request.json()
+        # 1. Parse body manually to be flexible
         audio_b64 = None
-        
-        # Check all possible key variants including camelCase for the portal
-        possible_keys = ["audioBase64", "audio_base_64", "audio_base64", "audio", "data", "file"]
-        for k in possible_keys:
-            if k in body and body[k]:
-                audio_b64 = body[k]
-                break
-        
-        # Fallback: search for any long string value
-        if not audio_b64:
-            for v in body.values():
-                if isinstance(v, str) and len(v) > 100:
-                    audio_b64 = v
+        try:
+            # Try JSON first
+            body = await request.json()
+            possible_keys = ["audioBase64", "audio_base_64", "audio", "data", "file"]
+            for k in possible_keys:
+                if k in body and body[k]:
+                    audio_b64 = body[k]
                     break
+            
+            if not audio_b64:
+                for v in body.values():
+                    if isinstance(v, str) and len(v) > 100:
+                        audio_b64 = v
+                        break
+        except Exception:
+            # FALLBACK: If not JSON, the body might be the raw Base64 string
+            raw_body = await request.body()
+            audio_b64 = raw_body.decode('utf-8')
         
-        if not audio_b64:
-             raise HTTPException(status_code=422, detail="No audio data found.")
+        if not audio_b64 or len(str(audio_b64)) < 20:
+             raise HTTPException(status_code=422, detail="No valid audio data found.")
 
-        # 2. Cleanup (Remove Data URI prefix like "data:audio/mp3;base64,")
+        # 2. Cleanup whitespace or quotes from copy-pasting
+        audio_b64 = str(audio_b64).strip().strip('"').strip("'")
         if "," in audio_b64:
             audio_b64 = audio_b64.split(",")[1]
 
@@ -60,16 +63,13 @@ async def detect_voice(
         audio_bytes = base64.b64decode(audio_b64)
         
     except Exception as e:
+        if isinstance(e, HTTPException): raise e
         raise HTTPException(status_code=400, detail=f"Request parsing error: {str(e)}")
 
     try:
-        # 4. AI Inference (Returns label: HUMAN/AI_GENERATED and score)
+        # 4. AI Inference
         label, confidence = service.predict(audio_bytes)
-        return DetectionResult(
-            label=label, 
-            confidence=confidence, 
-            message="Analysis successful"
-        )
+        return DetectionResult(label=label, confidence=confidence, message="Analysis successful")
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Model error: {str(e)}")