Update app.py

app.py

@@ -194,14 +194,17 @@ def display_confusion_matrices_los(percentage_idx):
 # Main function to handle user choice
 def handle_user_choice(choice, percentage_idx=None, uploaded_file=None):
     if choice == "Use Default Dataset":
-        return display_confusion_matrices_los(percentage_idx)
+        raw_img, embeddings_img = display_confusion_matrices_los(percentage_idx)
+        return raw_img, embeddings_img, ""  # Return empty string for console output
     elif choice == "Upload Dataset":
         if uploaded_file is not None:
-            return process_hdf5_file(uploaded_file, percentage_idx)
+            raw_img, embeddings_img, console_output = process_hdf5_file(uploaded_file, percentage_idx)
+            return raw_img, embeddings_img, console_output
         else:
-            return "Please upload a dataset", "Please upload a dataset"
+            return "Please upload a dataset", "Please upload a dataset", ""  # Return empty string for console output
     else:
-        return "Invalid choice", "Invalid choice"
+        return "Invalid choice", "Invalid choice", ""  # Return empty string for console output
+