yep

app.py

@@ -80,22 +80,16 @@ def create_field_mode_ui():
             diagnosis = DIAGNOSIS_TOOL(temp_file_path)
             print(f"Diagnosis received: {diagnosis}")
 
-            # The diagnosis from the tool should ideally be one of:
-            # "Healthy Maize Plant", "Maize Phosphorus Deficiency", or an error.
-            
             if "Could not parse" in diagnosis:
-                search_query = "Comic Relief"
-                report_title = "Unknown Condition"
-            else:
-                search_query = diagnosis
-                report_title = diagnosis
+                return f"Sorry, I couldn't identify the condition from the image. Raw output: {diagnosis}"
+
+            search_query = diagnosis
+            report_title = diagnosis
 
             results = KB.search(search_query)
             
             if not results:
-                # Fallback to comic relief if no specific remedy is found
-                results = KB.search("Comic Relief")
-                remedy = results[0]['content'] if results else "I couldn't find a specific remedy, and I'm out of jokes."
+                remedy = "No remedy found in the local knowledge base."
             else:
                 remedy = results[0]['content']
 
@@ -352,17 +346,9 @@ def create_kb_management_ui():
             yield "Rebuilding knowledge base..."
             try:
                 docs = {
-                    "Healthy Maize Plant": "",
-                    "Maize Phosphorus Deficiency": "",
-                    "Comic Relief": ""
+                    "Healthy Maize Plant": "For a Healthy Maize Plant, ensure proper watering and sunlight. No special remedy is needed. Continue good farming practices.",
+                    "Maize Phosphorus Deficiency": "Phosphorus deficiency in maize is characterized by stunted growth and purplish discoloration of leaves. To remedy this, apply a phosphorus-rich fertilizer like DAP (Di-Ammonium Phosphate) or bone meal to the soil. Follow package instructions for application rates."
                 }
-                with open(os.path.join("knowledge_base_data", "healthy_maize_remedy.txt")) as f:
-                    docs["Healthy Maize Plant"] = f.read()
-                with open(os.path.join("knowledge_base_data", "maize_phosphorus_deficiency_remedy.txt")) as f:
-                    docs["Maize Phosphorus Deficiency"] = f.read()
-                with open(os.path.join("knowledge_base_data", "comic_relief.txt")) as f:
-                    docs["Comic Relief"] = f.read()
-                
                 KB.create_initial_index(docs)
                 yield "Knowledge base rebuilt successfully."
             except Exception as e: