Update app.py

app.py

@@ -143,12 +143,24 @@ def health_check():
     return {"status": "API is running"}
 
 @app.post("/predict")
-@app.post("/run/predict")
 def predict(request: PredictionRequest):
     disease = predict_disease(request.symptoms)
     return {"predicted_disease": disease}
 
+class GradioRequest(BaseModel):
+    data: list
 
+@app.post("/run/predict")
+def predict_gradio(request: GradioRequest):
+    # Gradio sends data as a list of arguments
+    # We expect the first argument to be the comma-separated symptoms string
+    if not request.data:
+        raise HTTPException(status_code=400, detail="No data provided")
+    
+    symptoms = request.data[0]
+    disease = predict_disease(symptoms)
+    # Gradio expects the return value wrapped in a "data" list
+    return {"data": [disease]}
 
 if __name__ == "__main__":
     import uvicorn