Update app.py

app.py

@@ -52,14 +52,21 @@ health_knowledge_base = {
 def get_health_info(query):
     query = query.lower()
     
+    # List to store matching results
+    matching_conditions = []
+    
     # Search for conditions that match the query
     for condition, info in health_knowledge_base.items():
-        if condition in query:
-            return f"Condition: {condition.title()}\n" \
-                   f"Symptoms: {', '.join(info['symptoms'])}\n" \
-                   f"Treatment: {info['treatment']}"
+        if condition in query or any(symptom in query for symptom in info['symptoms']):
+            matching_conditions.append(f"Condition: {condition.title()}\n"
+                                       f"Symptoms: {', '.join(info['symptoms'])}\n"
+                                       f"Treatment: {info['treatment']}\n")
     
-    return "Sorry, I couldn't find information on that condition. Please try another query."
+    # If we have matching conditions, return all of them, otherwise return a message
+    if matching_conditions:
+        return "\n\n".join(matching_conditions)
+    else:
+        return "Sorry, I couldn't find information on that condition. Please try another query."
 
 # Set up the Streamlit app interface
 st.title("Healthcare Knowledge Chatbot")