Update main.py

main.py

@@ -1,12 +1,12 @@
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from motor.motor_asyncio import AsyncIOMotorClient
-from typing import List, Optional
+from typing import List
 from fastapi.middleware.cors import CORSMiddleware
 import os
 
 # MongoDB connection URI
-MONGO_URI = os.getenv("MONGO_URI")  # Replace with your MongoDB URI if not using environment variables
+MONGO_URI = os.getenv("MONGO_URI")
 
 # Initialize FastAPI app
 app = FastAPI()
@@ -37,6 +37,7 @@ class DiseaseRemedy(BaseModel):
     diseaseRemedy: str
 
 class Disease(BaseModel):
+    key: str
     plantName: str
     botanicalName: str
     diseaseDesc: DiseaseDesc
@@ -50,14 +51,39 @@ def welcome():
 # 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({"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(
+        key=disease["key"],
+        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
+
+# New endpoint to fetch disease details by key
+@app.get("/kotlinback/{key}", response_model=Disease)
+async def get_disease_by_key(key: str):
+    disease = await disease_collection.find_one({"key": key})
+
+    if disease is None:
+        raise HTTPException(status_code=404, detail="Disease not found")
+
+    disease_data = Disease(
+        key=disease["key"],
         plantName=disease["plantName"],
         botanicalName=disease["botanicalName"],
         diseaseDesc=DiseaseDesc(
@@ -77,4 +103,4 @@ async def get_disease(plant_name: str):
 # Run the FastAPI server using Uvicorn
 if __name__ == "__main__":
     import uvicorn
-    uvicorn.run(app, host="0.0.0.0", port=5000)
+    uvicorn.run(app, host="0.0.0.0", port=5000)