Update server.py

server.py

@@ -94,15 +94,37 @@ def health_check():
         
     return status
 
+# @app.post("/api/v1/inference")
+# async def inference(request: AudioRequest) -> AudioResponse:
+#     """Run inference on audio input"""
+#     if not INITIALIZATION_STATUS["model_loaded"]:
+#         raise HTTPException(
+#             status_code=503,
+#             detail=f"Model not ready. Status: {INITIALIZATION_STATUS}"
+#         )
+        
+#     try:
+#         # Decode audio from base64
+#         audio_bytes = base64.b64decode(request.audio_data) 
+#         audio_array = np.load(io.BytesIO(audio_bytes))
+        
+#         # Run inference
+#         result = model.inference(audio_array, request.sample_rate)
+        
+#         # Encode output audio
+#         buffer = io.BytesIO()
+#         np.save(buffer, result['audio'])
+#         audio_b64 = base64.b64encode(buffer.getvalue()).decode()
+        
+#         return AudioResponse(
+#             audio_data=audio_b64,
+#             text=result.get("text", "")
+#         )
+#     except Exception as e:
+#         logger.error(f"Inference failed: {str(e)}")
+#         raise HTTPException(status_code=500, detail=str(e))
 @app.post("/api/v1/inference")
 async def inference(request: AudioRequest) -> AudioResponse:
-    """Run inference on audio input"""
-    if not INITIALIZATION_STATUS["model_loaded"]:
-        raise HTTPException(
-            status_code=503,
-            detail=f"Model not ready. Status: {INITIALIZATION_STATUS}"
-        )
-        
     try:
         # Decode audio from base64
         audio_bytes = base64.b64decode(request.audio_data) 
@@ -120,10 +142,6 @@ async def inference(request: AudioRequest) -> AudioResponse:
             audio_data=audio_b64,
             text=result.get("text", "")
         )
-    except Exception as e:
-        logger.error(f"Inference failed: {str(e)}")
-        raise HTTPException(status_code=500, detail=str(e))
-
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000)