Update main.py

main.py

@@ -3,9 +3,10 @@ from pydantic import BaseModel
 from motor.motor_asyncio import AsyncIOMotorClient
 from typing import List, Optional
 from fastapi.middleware.cors import CORSMiddleware
-import os 
+import os
+
 # MongoDB connection URI
-MONGO_URI = os.getenv("MONGO_URI") #""
+MONGO_URI = os.getenv("MONGO_URI")  # Replace with your MongoDB URI if not using environment variables
 
 # Initialize FastAPI app
 app = FastAPI()
@@ -25,43 +26,52 @@ db = client.disease  # MongoDB database
 disease_collection = db.diseases  # MongoDB collection
 
 # Pydantic model to represent the Disease schema
-class Disease(BaseModel):
+class DiseaseDesc(BaseModel):
     diseaseName: str
-    description: str
-    cause: str
-    symptoms: List[str]
-    treatments: List[str]
-    preventionTips: Optional[List[str]] = []
-    imageURL: Optional[str] = None
-    createdAt: str
+    symptoms: str
+    diseaseCauses: str
+
+class DiseaseRemedy(BaseModel):
+    title: str
+    diseaseRemedyShortDesc: str
+    diseaseRemedy: str
+
+class Disease(BaseModel):
+    plantName: str
+    botanicalName: str
+    diseaseDesc: DiseaseDesc
+    diseaseRemedyList: List[DiseaseRemedy]
 
 # Root route to show a welcome message
 @app.get("/")
 def welcome():
     return {"message": "Welcome to the Disease Information API! Visit /docs for API documentation."}
 
-
-# API endpoint to fetch disease details by name
-@app.get("/disease/{name}", response_model=Disease)
-async def get_disease(name: str):
+# API endpoint to fetch disease details by plant name
+@app.get("/disease/{plant_name}", response_model=Disease)
+async def get_disease(plant_name: str):
     # Fetch the disease from MongoDB
-    disease = await disease_collection.find_one({"diseaseName": name})
-    
+    disease = await disease_collection.find_one({"plantName": plant_name})
+
     if disease is None:
         raise HTTPException(status_code=404, detail="Disease not found")
-    
+
     # Convert MongoDB document to Pydantic model
     disease_data = Disease(
-        diseaseName=disease["diseaseName"],
-        description=disease["description"],
-        cause=disease["cause"],
-        symptoms=disease["symptoms"],
-        treatments=disease["treatments"],
-        preventionTips=disease.get("preventionTips", []),
-        imageURL=disease.get("imageURL", None),
-        createdAt=disease["createdAt"].isoformat(),
+        plantName=disease["plantName"],
+        botanicalName=disease["botanicalName"],
+        diseaseDesc=DiseaseDesc(
+            diseaseName=disease["diseaseDesc"]["diseaseName"],
+            symptoms=disease["diseaseDesc"]["symptoms"],
+            diseaseCauses=disease["diseaseDesc"]["diseaseCauses"]
+        ),
+        diseaseRemedyList=[DiseaseRemedy(
+            title=remedy["title"],
+            diseaseRemedyShortDesc=remedy["diseaseRemedyShortDesc"],
+            diseaseRemedy=remedy["diseaseRemedy"]
+        ) for remedy in disease["diseaseRemedyList"]]
     )
-    
+
     return disease_data
 
 # Run the FastAPI server using Uvicorn