Update app.py

app.py

@@ -2,12 +2,19 @@ import gradio as gr
 import pandas as pd
 import matplotlib.pyplot as plt
 import numpy as np
-from fpdf2 import FPDF
 import io
 from datetime import datetime, timedelta
 import sys
 import traceback
 
+# Try to import fpdf2, but allow the app to run without it
+try:
+    from fpdf2 import FPDF
+    FPDF_AVAILABLE = True
+except ImportError:
+    FPDF_AVAILABLE = False
+    FPDF = None
+
 # Check library versions for debugging
 debug_msg = "Library Versions:\n"
 try:
@@ -16,7 +23,10 @@ try:
     debug_msg += f"Pandas: {pd.__version__}\n"
     debug_msg += f"Matplotlib: {matplotlib.__version__}\n"
     debug_msg += f"NumPy: {np.__version__}\n"
-    debug_msg += f"FPDF2: {FPDF.__version__}\n"
+    if FPDF_AVAILABLE:
+        debug_msg += f"FPDF2: {FPDF.__version__}\n"
+    else:
+        debug_msg += "FPDF2: Not installed (PDF download feature disabled)\n"
 except Exception as e:
     debug_msg += f"Error checking library versions: {str(e)}\n"
 
@@ -276,6 +286,10 @@ def filter_and_visualize(selected_lab, selected_type, selected_date_range):
 def download_pdf(selected_lab, selected_type, selected_date_range):
     global df
     try:
+        if not FPDF_AVAILABLE:
+            print("PDF download feature disabled: fpdf2 module not installed.")
+            return None
+        
         if df.empty:
             return None
         
@@ -335,7 +349,9 @@ try:
         anomaly_output = gr.Textbox(label="Anomaly Alerts")
         
         with gr.Row():
-            download_btn = gr.Button("Download PDF Report")
+            download_btn = gr.Button("Download PDF Report", visible=FPDF_AVAILABLE)
+            if not FPDF_AVAILABLE:
+                gr.Markdown("⚠️ PDF download feature disabled: Please install fpdf2 to enable.")
         
         error_box = gr.Textbox(label="Status/Error Message", visible=True, interactive=False)