Update app.py

app.py

@@ -6,26 +6,34 @@ from datetime import datetime
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 
+REQUIRED_COLUMNS = ['DeviceID', 'Lab', 'Type', 'Timestamp', 'Status', 'UsageCount']
+
 # ----------------------------
-# Load Logs
+# Load Logs Safely
 # ----------------------------
 def load_logs(file_obj):
-    if file_obj is not None:
-        content = file_obj.read().decode('utf-8')
-        df = pd.read_csv(io.StringIO(content), parse_dates=['Timestamp'])
-    else:
-        sample_data = {
-            'DeviceID': ['D001', 'D002', 'D003'],
-            'Lab': ['Lab A', 'Lab B', 'Lab A'],
-            'Type': ['UV', 'Weight', 'Cell'],
-            'Timestamp': [pd.Timestamp('2025-06-01 09:00:00'),
-                          pd.Timestamp('2025-06-01 10:00:00'),
-                          pd.Timestamp('2025-06-01 11:00:00')],
-            'Status': ['OK', 'DOWN', 'OK'],
-            'UsageCount': [120, 85, 100]
-        }
-        df = pd.DataFrame(sample_data)
-    return df
+    try:
+        if file_obj is not None:
+            content = file_obj.read().decode('utf-8')
+            df = pd.read_csv(io.StringIO(content))
+            if not all(col in df.columns for col in REQUIRED_COLUMNS):
+                raise ValueError(f"CSV must contain columns: {', '.join(REQUIRED_COLUMNS)}")
+            df['Timestamp'] = pd.to_datetime(df['Timestamp'], errors='coerce')
+            df.dropna(subset=['Timestamp'], inplace=True)
+        else:
+            df = pd.DataFrame({
+                'DeviceID': ['D001', 'D002', 'D003'],
+                'Lab': ['Lab A', 'Lab B', 'Lab A'],
+                'Type': ['UV', 'Weight', 'Cell'],
+                'Timestamp': [pd.Timestamp('2025-06-01 09:00:00'),
+                              pd.Timestamp('2025-06-01 10:00:00'),
+                              pd.Timestamp('2025-06-01 11:00:00')],
+                'Status': ['OK', 'DOWN', 'OK'],
+                'UsageCount': [120, 85, 100]
+            })
+        return df
+    except Exception as e:
+        raise ValueError(f"Failed to load CSV: {e}")
 
 # ----------------------------
 # Summarize Log Data
@@ -49,11 +57,10 @@ def generate_chart(df):
     return fig
 
 # ----------------------------
-# Export PDF with ReportLab
+# Export PDF using ReportLab
 # ----------------------------
 def export_pdf(df):
     summary = summarize_logs(df)
-
     buffer = io.BytesIO()
     pdf = canvas.Canvas(buffer, pagesize=letter)
     width, height = letter
@@ -73,12 +80,10 @@ def export_pdf(df):
     pdf.setFont("Helvetica", 10)
 
     for (lab, dev_type), row in summary.iterrows():
-        ok = int(row.get('OK', 0))
-        down = int(row.get('DOWN', 0))
         pdf.drawString(50, y, str(lab))
         pdf.drawString(150, y, str(dev_type))
-        pdf.drawString(300, y, str(ok))
-        pdf.drawString(350, y, str(down))
+        pdf.drawString(300, y, str(int(row.get('OK', 0))))
+        pdf.drawString(350, y, str(int(row.get('DOWN', 0))))
         y -= 20
         if y < 50:
             pdf.showPage()
@@ -89,29 +94,42 @@ def export_pdf(df):
     return ("LabOps_Summary.pdf", buffer.read())
 
 # ----------------------------
-# Gradio App
+# Dashboard Display
 # ----------------------------
 def dashboard(file_obj):
-    df = load_logs(file_obj)
-    fig = generate_chart(df)
-    return fig
+    try:
+        df = load_logs(file_obj)
+        return generate_chart(df)
+    except Exception as e:
+        return gr.update(visible=True), gr.Textbox.update(value=f"❌ Error: {e}", visible=True)
 
+# ----------------------------
+# PDF Generator
+# ----------------------------
 def generate_pdf_button(file_obj):
-    df = load_logs(file_obj)
-    return export_pdf(df)
+    try:
+        df = load_logs(file_obj)
+        filename, pdf_bytes = export_pdf(df)
+        return gr.File.update(value=(filename, pdf_bytes), visible=True)
+    except Exception as e:
+        return gr.File.update(visible=False), gr.Textbox.update(value=f"❌ Error: {e}", visible=True)
 
+# ----------------------------
+# Gradio Interface
+# ----------------------------
 with gr.Blocks() as demo:
     gr.Markdown("## 🧪 LabOps Dashboard")
-    gr.Markdown("Monitor and analyze device logs across SmartLabs.")
+    gr.Markdown("Upload lab device logs to visualize uptime/downtime & generate PDF reports.")
 
     with gr.Row():
         file_input = gr.File(label="Upload Log CSV", file_types=[".csv"])
         download_button = gr.Button("Download PDF Summary")
-        download_file = gr.File(label="Download PDF", visible=True)
+        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)
+    download_button.click(fn=generate_pdf_button, inputs=file_input, outputs=[download_file, error_output])
 
 demo.launch()