Update app.py

app.py

@@ -12,14 +12,14 @@ def upload_csv(file):
     try:
         # Check if a file was actually uploaded
         if file is None:
-            return ["All"], ["All"], "No file uploaded. Please upload a CSV file."
+            return ["All"], ["All"], "No file uploaded. Please upload a CSV file.", "All", "All"
         
         # Read the CSV file
         df = pd.read_csv(file.name)
         
         # Check if the DataFrame is empty
         if df.empty:
-            return ["All"], ["All"], "The uploaded CSV file is empty."
+            return ["All"], ["All"], "The uploaded CSV file is empty.", "All", "All"
         
         # Define required columns
         required_columns = {'DeviceID', 'Lab', 'Type', 'Timestamp', 'Status', 'UsageCount'}
@@ -27,16 +27,17 @@ def upload_csv(file):
         # Check if all required columns are present
         if not required_columns.issubset(df.columns):
             missing_cols = required_columns - set(df.columns)
-            return ["All"], ["All"], f"CSV is missing required columns: {', '.join(missing_cols)}"
+            return ["All"], ["All"], f"CSV is missing required columns: {', '.join(missing_cols)}", "All", "All"
         
         # Extract unique values for dropdowns
         labs = ['All'] + sorted(df['Lab'].dropna().unique().tolist())
         types = ['All'] + sorted(df['Type'].dropna().unique().tolist())
         
-        return labs, types, ""
+        # Return the updated choices and reset the dropdown values to "All"
+        return labs, types, "", "All", "All"
     except Exception as e:
         # Handle any other errors (e.g., file not readable, invalid CSV format)
-        return ["All"], ["All"], f"Failed to load CSV: {str(e)}"
+        return ["All"], ["All"], f"Failed to load CSV: {str(e)}", "All", "All"
 
 def filter_and_plot(selected_lab, selected_type):
     global df
@@ -124,8 +125,12 @@ with gr.Blocks() as demo:
     download_btn = gr.Button("Download PDF Summary")
     error_box = gr.Textbox(label="Status/Error Message", visible=True, interactive=False)
 
-    # Connect the components - Fixed the outputs argument
-    csv_input.change(fn=upload_csv, inputs=csv_input, outputs=[lab_dropdown, type_dropdown, error_box])
+    # Connect the components
+    csv_input.change(
+        fn=upload_csv,
+        inputs=csv_input,
+        outputs=[lab_dropdown, type_dropdown, error_box, lab_dropdown, type_dropdown]
+    )
     lab_dropdown.change(fn=filter_and_plot, inputs=[lab_dropdown, type_dropdown], outputs=plot_output)
     type_dropdown.change(fn=filter_and_plot, inputs=[lab_dropdown, type_dropdown], outputs=plot_output)
     download_btn.click(fn=download_pdf, inputs=[lab_dropdown, type_dropdown], outputs=gr.File(label="Download PDF"))