Initial Commit

app.py

@@ -83,18 +83,42 @@ custom_symptom_weightage = {
     "Family Working in Public Exposed Places": 0.5,
 }
 
+# Create a bar chart to represent symptom weightage
+# def show_symptom_weightage(symptoms, weights):
+#     plt.figure(figsize=(10, 6))
+#     # Filter out the non-selected symptoms and their corresponding weights
+#     selected_symptoms = [symptom for symptom, weight in zip(symptoms, weights) if weight > 0]
+#     selected_weights = [weight for weight in weights if weight > 0]
+#     plt.bar(selected_symptoms, selected_weights)
+#     plt.xlabel("Symptom")
+#     plt.ylabel("Weight")
+#     plt.title("Weightage/Importance of Symptoms")
+#     plt.xticks(rotation=45, ha='right')
+#     plt.tight_layout()
+#     # Save the plot as an image and return the file path
+#     file_path = tempfile.NamedTemporaryFile(suffix=".png").name
+#     plt.savefig(file_path)
+#     plt.close()
+#     return file_path
+
 # Create a bar chart to represent symptom weightage
 def show_symptom_weightage(symptoms, weights):
-    plt.figure(figsize=(10, 6))
-    # Filter out the non-selected symptoms and their corresponding weights
     selected_symptoms = [symptom for symptom, weight in zip(symptoms, weights) if weight > 0]
     selected_weights = [weight for weight in weights if weight > 0]
+
+    # Check if any symptom is selected by the user
+    if len(selected_symptoms) == 0:
+        # If no symptom is selected, return None to indicate no bar graph should be displayed
+        return None
+
+    plt.figure(figsize=(10, 6))
     plt.bar(selected_symptoms, selected_weights)
     plt.xlabel("Symptom")
     plt.ylabel("Weight")
-    plt.title("Weightage/Importance of Symptoms")
+    plt.title("Weightage/Importance of Selected Symptoms")
     plt.xticks(rotation=45, ha='right')
     plt.tight_layout()
+
     # Save the plot as an image and return the file path
     file_path = tempfile.NamedTemporaryFile(suffix=".png").name
     plt.savefig(file_path)