Update app.py

app.py

@@ -35,18 +35,38 @@ def load_logs(file_obj):
     except Exception as e:
         raise ValueError(f"Failed to load CSV: {e}")
 
+# ----------------------------
+# Filter DataFrame by Lab and Type
+# ----------------------------
+def filter_df(df, selected_lab, selected_type):
+    filtered = df.copy()
+    if selected_lab and selected_lab != "All":
+        filtered = filtered[filtered['Lab'] == selected_lab]
+    if selected_type and selected_type != "All":
+        filtered = filtered[filtered['Type'] == selected_type]
+    return filtered
+
 # ----------------------------
 # Summarize Log Data
 # ----------------------------
 def summarize_logs(df):
+    if df.empty:
+        # Return empty DataFrame with expected columns for plotting
+        return pd.DataFrame(columns=['OK', 'DOWN'])
     summary = df.groupby(['Lab', 'Type'])['Status'].value_counts().unstack().fillna(0)
     return summary
 
 # ----------------------------
 # Generate Chart
 # ----------------------------
-def generate_chart(df):
-    summary = summarize_logs(df)
+def generate_chart(df, selected_lab, selected_type):
+    filtered = filter_df(df, selected_lab, selected_type)
+    summary = summarize_logs(filtered)
+    if summary.empty:
+        fig, ax = plt.subplots(figsize=(8, 4))
+        ax.text(0.5, 0.5, 'No data for selected filters', ha='center', va='center', fontsize=14)
+        ax.axis('off')
+        return fig
     fig, ax = plt.subplots(figsize=(8, 4))
     summary.plot(kind='bar', stacked=True, ax=ax)
     ax.set_title("Device Uptime/Downtime Summary")
@@ -59,8 +79,9 @@ def generate_chart(df):
 # ----------------------------
 # Export PDF using ReportLab
 # ----------------------------
-def export_pdf(df):
-    summary = summarize_logs(df)
+def export_pdf(df, selected_lab, selected_type):
+    filtered = filter_df(df, selected_lab, selected_type)
+    summary = summarize_logs(filtered)
     buffer = io.BytesIO()
     pdf = canvas.Canvas(buffer, pagesize=letter)
     width, height = letter
@@ -94,23 +115,39 @@ def export_pdf(df):
     return ("LabOps_Summary.pdf", buffer.read())
 
 # ----------------------------
-# Dashboard Display
+# Update dropdown options based on uploaded file
+# ----------------------------
+def update_dropdown_options(file_obj):
+    try:
+        df = load_logs(file_obj)
+        labs = sorted(df['Lab'].dropna().unique().tolist())
+        types = sorted(df['Type'].dropna().unique().tolist())
+        # Add "All" option for no filtering
+        labs = ["All"] + labs
+        types = ["All"] + types
+        return labs, types, gr.update(visible=True)
+    except Exception as e:
+        return ["All"], ["All"], gr.update(visible=False)
+
+# ----------------------------
+# Dashboard Display with filters
 # ----------------------------
-def dashboard(file_obj):
+def dashboard(file_obj, selected_lab, selected_type):
     try:
         df = load_logs(file_obj)
-        return generate_chart(df)
+        fig = generate_chart(df, selected_lab, selected_type)
+        return fig, gr.update(visible=False)  # Hide error
     except Exception as e:
-        return gr.update(visible=True), gr.Textbox.update(value=f"❌ Error: {e}", visible=True)
+        return None, gr.update(value=f"❌ Error: {e}", visible=True)
 
 # ----------------------------
-# PDF Generator
+# PDF Generator with filters
 # ----------------------------
-def generate_pdf_button(file_obj):
+def generate_pdf_button(file_obj, selected_lab, selected_type):
     try:
         df = load_logs(file_obj)
-        filename, pdf_bytes = export_pdf(df)
-        return gr.File.update(value=(filename, pdf_bytes), visible=True)
+        filename, pdf_bytes = export_pdf(df, selected_lab, selected_type)
+        return gr.File.update(value=(filename, pdf_bytes), visible=True), gr.Textbox.update(visible=False)
     except Exception as e:
         return gr.File.update(visible=False), gr.Textbox.update(value=f"❌ Error: {e}", visible=True)
 
@@ -123,13 +160,35 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         file_input = gr.File(label="Upload Log CSV", file_types=[".csv"])
+        lab_dropdown = gr.Dropdown(label="Select Lab", choices=["All"], value="All", interactive=True)
+        type_dropdown = gr.Dropdown(label="Select Equipment Type", choices=["All"], value="All", interactive=True)
         download_button = gr.Button("Download PDF Summary")
         download_file = gr.File(label="Download PDF", visible=False)
 
     plot_output = gr.Plot()
     error_output = gr.Textbox(visible=False, label="Errors")
 
-    file_input.change(fn=dashboard, inputs=file_input, outputs=plot_output)
-    download_button.click(fn=generate_pdf_button, inputs=file_input, outputs=[download_file, error_output])
+    # Update dropdowns on file upload
+    file_input.change(
+        fn=update_dropdown_options,
+        inputs=file_input,
+        outputs=[lab_dropdown, type_dropdown, error_output]
+    )
+
+    # Update plot when file or dropdowns change
+    inputs_for_dashboard = [file_input, lab_dropdown, type_dropdown]
+    for input_component in inputs_for_dashboard:
+        input_component.change(
+            fn=dashboard,
+            inputs=inputs_for_dashboard,
+            outputs=[plot_output, error_output]
+        )
+
+    # Generate PDF on button click
+    download_button.click(
+        fn=generate_pdf_button,
+        inputs=[file_input, lab_dropdown, type_dropdown],
+        outputs=[download_file, error_output]
+    )
 
 demo.launch()