Update main.py

main.py

@@ -148,10 +148,24 @@ async def predict_plant_disease(plant_name: str, file: UploadFile = File(...)):
             external_data = {"error": str(e)}
 
         return JSONResponse(content={
-            "plant": plant_name,
-            "predicted_disease": class_label,
-            "external_data": external_data
-        })
+    "plant": external_data.get("plantName", plant_name),
+    "botanical_name": external_data.get("botanicalName", "Unknown"),
+    "predicted_disease": external_data.get("diseaseDesc", {}).get("diseaseName", class_label),
+    "symptoms": external_data.get("diseaseDesc", {}).get("symptoms", "Not Available"),
+    "causes": external_data.get("diseaseDesc", {}).get("diseaseCauses", "Not Available"),
+    "remedies": [
+        {
+            "title": remedy.get("title", "Unknown"),
+            "short_desc": remedy.get("diseaseRemedyShortDesc", "Not Available"),
+            "remedy": remedy.get("diseaseRemedy", "Not Available")
+        } for remedy in external_data.get("diseaseRemedyList", [])
+    ]
+})
+        # return JSONResponse(content={
+        #     "plant": plant_name,
+        #     "predicted_disease": class_label,
+        #     "external_data": external_data
+        # })
     finally:
         # Clean up temporary file
         os.remove(temp_path)