Update app.py

app.py

@@ -12,12 +12,12 @@ def upload_csv(file):
         df = pd.read_csv(file.name)
         required_columns = {'DeviceID', 'Lab', 'Type', 'Timestamp', 'Status', 'UsageCount'}
         if not required_columns.issubset(df.columns):
-            return None, "CSV must contain columns: " + ", ".join(required_columns)
+            return gr.update(choices=["All"]), gr.update(choices=["All"]), "❌ CSV must contain columns: " + ", ".join(required_columns)
         labs = ['All'] + sorted(df['Lab'].dropna().unique().tolist())
         types = ['All'] + sorted(df['Type'].dropna().unique().tolist())
-        return (labs, types), ""
+        return gr.update(choices=labs, value="All"), gr.update(choices=types, value="All"), ""
     except Exception as e:
-        return None, f"Failed to load CSV: {str(e)}"
+        return gr.update(choices=["All"]), gr.update(choices=["All"]), f"❌ Failed to load CSV: {str(e)}"
 
 def filter_and_plot(selected_lab, selected_type):
     global df
@@ -29,11 +29,11 @@ def filter_and_plot(selected_lab, selected_type):
     if selected_type != "All":
         filtered_df = filtered_df[filtered_df["Type"] == selected_type]
     
-    # Prepare plot
+    # Plotting
     plt.figure(figsize=(8, 4))
     status_counts = filtered_df["Status"].value_counts()
     status_counts.plot(kind="bar", color=["green", "red"])
-    plt.title("Status Counts")
+    plt.title("Device Status Overview")
     plt.xlabel("Status")
     plt.ylabel("Count")
     
@@ -74,14 +74,16 @@ with gr.Blocks() as demo:
         csv_input = gr.File(label="Upload Log CSV", file_types=[".csv"])
     
     with gr.Row():
-        lab_dropdown = gr.Dropdown(label="Select Lab", choices=["All"])
-        type_dropdown = gr.Dropdown(label="Select Equipment Type", choices=["All"])
+        lab_dropdown = gr.Dropdown(label="Select Lab", choices=["All"], value="All")
+        type_dropdown = gr.Dropdown(label="Select Equipment Type", choices=["All"], value="All")
 
-    plot_output = gr.Image(label="Plot")
+    error_box = gr.Textbox(label="Message", visible=True, interactive=False)
+
+    plot_output = gr.Image(label="Device Status Plot")
     download_btn = gr.Button("Download PDF Summary")
-    error_box = gr.Textbox(visible=False)
 
-    csv_input.change(fn=upload_csv, inputs=csv_input, outputs=[(lab_dropdown, type_dropdown), error_box])
+    # Hook up event logic
+    csv_input.change(fn=upload_csv, inputs=csv_input, outputs=[lab_dropdown, type_dropdown, error_box])
     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())