bug fix

app.py

@@ -80,38 +80,21 @@ 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 = "Wetin My Eye See So?"
-                results = KB.search(search_query)
-                remedy = results[0]['content'] if results else "I'm not sure what I'm seeing!"
-                return f"""## Diagnosis Report
-
-**Condition Identified:**
-### Unknown
-
----
-
-## Suggested Remedy
-
-{remedy}"""
-
-            # Default to comic relief
-            search_query = "Wetin My Eye See So?"
-            report_title = diagnosis # Default title
+                search_query = "Comic Relief"
+                report_title = "Unknown Condition"
+            else:
+                search_query = diagnosis
+                report_title = diagnosis
 
-            healthy_keywords = ["Healthy", "harvested", "drying"]
-            if any(keyword in diagnosis for keyword in healthy_keywords):
-                search_query = "Better Ways to Enjoy Your Healthy Corn (Maize)"
-                report_title = "Healthy Maize Plant"
-            elif "phosphorus deficiency" in diagnosis.lower():
-                search_query = "How You Fit Solve Phosphorus Problem for Your Corn (Maize)"
-                report_title = "Phosphorus Deficiency"
-            
             results = KB.search(search_query)
             
             if not results:
                 # Fallback to comic relief if no specific remedy is found
-                results = KB.search("Wetin My Eye See So?")
+                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."
             else:
                 remedy = results[0]['content']
@@ -368,11 +351,18 @@ def create_kb_management_ui():
         def rebuild_kb():
             yield "Rebuilding knowledge base..."
             try:
-                docs = {}
-                for filename in os.listdir("knowledge_base_data"):
-                    if filename.endswith(".txt"):
-                        with open(os.path.join("knowledge_base_data", filename)) as f:
-                            docs[filename] = f.read()
+                docs = {
+                    "Healthy Maize Plant": "",
+                    "Maize Phosphorus Deficiency": "",
+                    "Comic Relief": ""
+                }
+                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: