Update app.py

app.py

@@ -2,8 +2,6 @@ import gradio as gr
 import pandas as pd
 from datetime import datetime, timedelta
 import logging
-import plotly.express as px
-import plotly.graph_objects as go
 from sklearn.ensemble import IsolationForest
 from concurrent.futures import ThreadPoolExecutor
 import os
@@ -15,18 +13,15 @@ from simple_salesforce import Salesforce
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-# Salesforce configuration
+# Try to import plotly
 try:
-    sf = Salesforce(
-        username='multi-devicelabopsdashboard@sathkrutha.com',
-        password='Team@1234',
-        security_token=os.getenv('SF_SECURITY_TOKEN', ''),
-        domain='login'
-    )
-    logging.info("Salesforce connection established")
-except Exception as e:
-    logging.error(f"Failed to connect to Salesforce: {str(e)}")
-    sf = None
+    import plotly.express as px
+    import plotly.graph_objects as go
+    plotly_available = True
+    logging.info("plotly module successfully imported")
+except ImportError:
+    logging.warning("plotly module not found. Chart generation disabled.")
+    plotly_available = False
 
 # Try to import reportlab
 try:
@@ -40,6 +35,19 @@ except ImportError:
     logging.warning("reportlab module not found. PDF generation disabled.")
     reportlab_available = False
 
+# Salesforce configuration
+try:
+    sf = Salesforce(
+        username='multi-devicelabopsdashboard@sathkrutha.com',
+        password='Team@1234',
+        security_token=os.getenv('SF_SECURITY_TOKEN', ''),
+        domain='login'
+    )
+    logging.info("Salesforce connection established")
+except Exception as e:
+    logging.error(f"Failed to connect to Salesforce: {str(e)}")
+    sf = None
+
 # Cache picklist values at startup
 def get_picklist_values(field_name):
     if sf is None:
@@ -254,20 +262,30 @@ def generate_dashboard_insights(df):
         logging.error(f"Dashboard insights generation failed: {str(e)}")
         return "Failed to generate insights."
 
-# Placeholder chart for empty data
+# Placeholder chart for empty data or missing plotly
 def create_placeholder_chart(title):
-    fig = go.Figure()
-    fig.add_annotation(
-        text="No data available for this chart",
-        xref="paper", yref="paper",
-        x=0.5, y=0.5, showarrow=False,
-        font=dict(size=16)
-    )
-    fig.update_layout(title=title, margin=dict(l=20, r=20, t=40, b=20))
-    return fig
+    if not plotly_available:
+        logging.warning(f"Cannot create chart '{title}': plotly not available")
+        return None
+    try:
+        fig = go.Figure()
+        fig.add_annotation(
+            text="No data available for this chart",
+            xref="paper", yref="paper",
+            x=0.5, y=0.5, showarrow=False,
+            font=dict(size=16)
+        )
+        fig.update_layout(title=title, margin=dict(l=20, r=20, t=40, b=20))
+        return fig
+    except Exception as e:
+        logging.error(f"Failed to create placeholder chart '{title}': {str(e)}")
+        return None
 
 # Create usage chart
 def create_usage_chart(df):
+    if not plotly_available:
+        logging.warning("Cannot create usage chart: plotly not available")
+        return None
     try:
         if df.empty or "usage_hours" not in df.columns or "device_id" not in df.columns:
             logging.warning("Insufficient data for usage chart")
@@ -290,6 +308,9 @@ def create_usage_chart(df):
 
 # Create downtime chart
 def create_downtime_chart(df):
+    if not plotly_available:
+        logging.warning("Cannot create downtime chart: plotly not available")
+        return None
     try:
         if df.empty or "downtime" not in df.columns or "device_id" not in df.columns:
             logging.warning("Insufficient data for downtime chart")
@@ -312,6 +333,9 @@ def create_downtime_chart(df):
 
 # Create daily log trends chart
 def create_daily_log_trends_chart(df):
+    if not plotly_available:
+        logging.warning("Cannot create daily log trends chart: plotly not available")
+        return None
     try:
         if df.empty or "timestamp" not in df.columns:
             logging.warning("Insufficient data for daily log trends chart")
@@ -335,6 +359,9 @@ def create_daily_log_trends_chart(df):
 
 # Create weekly uptime chart
 def create_weekly_uptime_chart(df):
+    if not plotly_available:
+        logging.warning("Cannot create weekly uptime chart: plotly not available")
+        return None
     try:
         if df.empty or "timestamp" not in df.columns or "usage_hours" not in df.columns or "downtime" not in df.columns:
             logging.warning("Insufficient data for weekly uptime chart")
@@ -364,6 +391,9 @@ def create_weekly_uptime_chart(df):
 
 # Create anomaly alerts chart
 def create_anomaly_alerts_chart(anomalies_df):
+    if not plotly_available:
+        logging.warning("Cannot create anomaly alerts chart: plotly not available")
+        return None
     try:
         if anomalies_df is None or anomalies_df.empty or "timestamp" not in anomalies_df.columns:
             logging.warning("Insufficient data for anomaly alerts chart")
@@ -396,9 +426,9 @@ def generate_device_cards(df):
         }).reset_index()
         device_stats['count'] = df.groupby('device_id').size().reindex(device_stats['device_id']).values
         device_stats['health'] = device_stats['status'].map({
-(b'Active', b'Healthy'),
-(b'Inactive', b'Unhealthy'),
-(b'Pending', b'Warning')
+            'Active': 'Healthy',
+            'Inactive': 'Unhealthy',
+            'Pending': 'Warning'
         }).fillna('Unknown')
         cards_html = '<div style="display: flex; flex-wrap: wrap; gap: 20px;">'
         for _, row in device_stats.iterrows():
@@ -421,6 +451,7 @@ def generate_device_cards(df):
 # Generate PDF content
 def generate_pdf_content(summary, preview_df, anomalies, amc_reminders, insights, device_cards_html, daily_log_chart, weekly_uptime_chart, anomaly_alerts_chart, downtime_chart):
     if not reportlab_available:
+        logging.warning("PDF generation disabled: reportlab not available")
         return None
     try:
         pdf_path = f"status_report_{datetime.now().strftime('%Y%m%d_%H%M%S')}.pdf"
@@ -479,7 +510,10 @@ def generate_pdf_content(summary, preview_df, anomalies, amc_reminders, insights
         story.append(Spacer(1, 12))
 
         story.append(Paragraph("Charts", styles['Heading2']))
-        story.append(Paragraph("[Chart placeholders - see dashboard for visuals]", styles['Normal']))
+        if not plotly_available:
+            story.append(safe_paragraph("Charts unavailable: plotly not installed.", styles['Normal']))
+        else:
+            story.append(safe_paragraph("[Chart placeholders - see dashboard for visuals]", styles['Normal']))
 
         doc.build(story)
         logging.info(f"PDF generated at {pdf_path}")