Update main.py

main.py

@@ -78,25 +78,27 @@ async def get_disease(plant_name: str):
 
 # New endpoint to fetch disease details by key
 @app.get("/kotlinback/{key}", response_model=Disease)
-async def get_disease_by_key(key: str):
+async def get_disease_by_key(key: str, lang: str = "en"):
     disease = await disease_collection.find_one({"key": key})
 
     if disease is None:
         raise HTTPException(status_code=404, detail="Disease not found")
-
+    def get_localized(d, field):
+        localized_field = f"{field}_{lang}"
+        return d.get(localized_field) or d.get(field) or "N/A"
     disease_data = Disease(
         key=disease["key"],
-        plantName=disease["plantName"],
-        botanicalName=disease["botanicalName"],
+        plantName=get_localized(disease, "plantName"),
+        botanicalName=get_localized(disease, "botanicalName"),
         diseaseDesc=DiseaseDesc(
-            diseaseName=disease["diseaseDesc"]["diseaseName"],
-            symptoms=disease["diseaseDesc"]["symptoms"],
-            diseaseCauses=disease["diseaseDesc"]["diseaseCauses"]
+            diseaseName=get_localized(disease["diseaseDesc"], "diseaseName"),
+            symptoms=get_localized(disease["diseaseDesc"], "symptoms"),
+            diseaseCauses=get_localized(disease["diseaseDesc"], "diseaseCauses")
         ),
         diseaseRemedyList=[DiseaseRemedy(
-            title=remedy["title"],
-            diseaseRemedyShortDesc=remedy["diseaseRemedyShortDesc"],
-            diseaseRemedy=remedy["diseaseRemedy"]
+            title=get_localized(remedy, "title"),
+            diseaseRemedyShortDesc=get_localized(remedy, "diseaseRemedyShortDesc"),
+            diseaseRemedy=get_localized(remedy, "diseaseRemedy")
         ) for remedy in disease["diseaseRemedyList"]]
     )