Update app.py

app.py

@@ -124,15 +124,15 @@ class SelectedSymptomsQuery(BaseModel):
 def find_disease(query: SelectedSymptomsQuery):
     SelectedSymptoms = query.selected_symptoms
     all_selected_symptoms.extend(SelectedSymptoms)
-    
+
     disease_list = []
     symptoms_list = []
-    unique_symptoms_list = []
-    
+    unique_symptoms_set = set()  # Use a set for unique symptoms
+
     # Combine all the symptoms we already know (all_symptoms + selected symptoms)
     known_symptoms = {symptom.lower() for symptom_set in all_symptoms for symptom in symptom_set}
     known_symptoms.update([symptom.lower() for symptom in SelectedSymptoms])
-    
+
     # Generate combinations of symptoms from all_symptoms and selected symptoms
     for symptoms_set in all_symptoms:
         for i in range(1, len(symptoms_set) + 1):  # Generate combinations with all lengths
@@ -141,7 +141,7 @@ def find_disease(query: SelectedSymptomsQuery):
                 
                 # Search for diseases that match the combination
                 matching_diseases = df[df['Symptoms'].apply(lambda x: all(s in x for s in temp))]
-                
+
                 for _, row in matching_diseases.iterrows():
                     disease_info = {
                         'Disease': row['Name'],
@@ -149,17 +149,15 @@ def find_disease(query: SelectedSymptomsQuery):
                         'Treatments': row['Treatments']
                     }
                     disease_list.append(disease_info)
-                    
-                    symptoms_info = row['Symptoms']
-                    symptoms_list.append(symptoms_info)
-    
-    # Flatten the list of symptoms and remove duplicates, excluding known symptoms
-    for symptoms in symptoms_list:
-        for symptom in symptoms:
-            symptom_lower = symptom.lower()
-            if symptom_lower not in known_symptoms and symptom_lower not in unique_symptoms_list:
-                unique_symptoms_list.append(symptom_lower)
-    
+
+                    # Add each symptom in lowercase to the set
+                    for symptom in row['Symptoms']:
+                        if symptom.lower() not in known_symptoms:
+                            unique_symptoms_set.add(symptom.lower())
+
+    # Convert the set back to a list for the response
+    unique_symptoms_list = list(unique_symptoms_set)
+
     return {
         "unique_symptoms_list": unique_symptoms_list,
         "all_selected_symptoms": all_selected_symptoms,