Update main.py

main.py

@@ -1,6 +1,5 @@
 import base64
-import binascii
-from fastapi import FastAPI, HTTPException, Depends, Header, Body
+from fastapi import FastAPI, HTTPException, Depends, Header
 from schemas import AudioInput, DetectionResult
 from model_service import get_model_service, ModelService
 
@@ -23,34 +22,25 @@ async def verify_api_key(x_api_key: str = Header(...)):
 
 @app.post("/detect", response_model=DetectionResult)
 async def detect_voice(
-    data: dict = Body(..., example={"audio_base64": "PASTE_BASE64_HERE"}),
+    input_data: AudioInput, 
     service: ModelService = Depends(get_model_service),
     api_key: str = Depends(verify_api_key)
 ):
     try:
-        audio_b64 = None
-        possible_keys = ["audio_base64", "audio", "data", "file", "encoded_audio", "mp3"]
-        for k in possible_keys:
-            if k in data and data[k]:
-                audio_b64 = data[k]
-                break
+        audio_b64 = input_data.audio_base64
         if not audio_b64:
-            for v in data.values():
-                if isinstance(v, str) and len(v) > 100:
-                    audio_b64 = v
-                    break
-        if not audio_b64:
-             raise HTTPException(status_code=422, detail="No valid audio data found.")
+             raise HTTPException(status_code=422, detail="No audio data found.")
         if "," in audio_b64:
             audio_b64 = audio_b64.split(",")[1]
         audio_bytes = base64.b64decode(audio_b64)
     except Exception as e:
-        raise HTTPException(status_code=400, detail=str(e))
+        raise HTTPException(status_code=400, detail=f"Parsing error: {str(e)}")
+
     try:
         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=str(e))
+        raise HTTPException(status_code=500, detail=f"Model error: {str(e)}")
 
 @app.get("/")
 def read_root():