Update app.py

app.py

@@ -324,6 +324,9 @@ def query_ai(model, stats, question, df=None):
 def save_plotly_as_image(fig, filename):
     """Convert Plotly figure to PNG for PDF"""
     try:
+        # Check if kaleido is available
+        import kaleido
+        
         temp_dir = tempfile.gettempdir()
         filepath = os.path.join(temp_dir, filename)
         
@@ -335,19 +338,12 @@ def save_plotly_as_image(fig, filename):
             margin=dict(t=50, b=40, l=40, r=40)
         )
         
-        # Try different engines
-        try:
-            pio.write_image(fig, filepath, format='png', width=800, height=400, scale=2, engine='kaleido')
-        except:
-            try:
-                pio.write_image(fig, filepath, format='png', width=800, height=400, scale=1, engine='orca')
-            except:
-                # Fallback: save as HTML and convert manually
-                html_path = filepath.replace('.png', '.html')
-                fig.write_html(html_path)
-                return None  # Skip image if conversion fails
-        
+        pio.write_image(fig, filepath, format='png', width=800, height=400, scale=2)
         return filepath if os.path.exists(filepath) else None
+        
+    except ImportError:
+        st.warning("📸 Kaleido not installed. Charts will not appear in PDF. Run: pip install kaleido")
+        return None
     except Exception as e:
         st.warning(f"Chart save error: {e}")
         return None
@@ -481,32 +477,57 @@ def create_enhanced_pdf_report(df, stats, outliers):
     # Generate and add charts
     elements.append(Paragraph("Production Analysis Charts", subtitle_style))
     
-    with st.spinner("Generating charts for PDF..."):
-        charts = create_pdf_charts(df, stats)
-    
-    # Add charts to PDF
-    if charts.get('pie') and charts['pie'] and os.path.exists(charts['pie']):
-        elements.append(Paragraph("Production Distribution", styles['Heading3']))
-        elements.append(Image(charts['pie'], width=6*inch, height=3*inch))
-        elements.append(Spacer(1, 20))
-    
-    if charts.get('trend') and charts['trend'] and os.path.exists(charts['trend']):
-        elements.append(Paragraph("Production Trend", styles['Heading3']))
-        elements.append(Image(charts['trend'], width=6*inch, height=3*inch))
-        elements.append(Spacer(1, 20))
-    
-    if charts.get('bar') and charts['bar'] and os.path.exists(charts['bar']):
-        elements.append(Paragraph("Material Comparison", styles['Heading3']))
-        elements.append(Image(charts['bar'], width=6*inch, height=3*inch))
-        elements.append(Spacer(1, 20))
-    
-    if charts.get('shift') and charts['shift'] and os.path.exists(charts['shift']):
-        elements.append(Paragraph("Shift Analysis", styles['Heading3']))
-        elements.append(Image(charts['shift'], width=6*inch, height=3*inch))
-        elements.append(Spacer(1, 20))
-    else:
-        # If no charts available, add text note
-        elements.append(Paragraph("Charts not available - install kaleido: pip install kaleido", styles['Normal']))
+    try:
+        # Check if kaleido is available
+        import kaleido
+        
+        with st.spinner("Generating charts for PDF..."):
+            charts = create_pdf_charts(df, stats)
+        
+        # Add charts to PDF
+        charts_added = False
+        
+        if charts.get('pie') and charts['pie'] and os.path.exists(charts['pie']):
+            elements.append(Paragraph("Production Distribution", styles['Heading3']))
+            elements.append(Image(charts['pie'], width=6*inch, height=3*inch))
+            elements.append(Spacer(1, 20))
+            charts_added = True
+        
+        if charts.get('trend') and charts['trend'] and os.path.exists(charts['trend']):
+            elements.append(Paragraph("Production Trend", styles['Heading3']))
+            elements.append(Image(charts['trend'], width=6*inch, height=3*inch))
+            elements.append(Spacer(1, 20))
+            charts_added = True
+        
+        if charts.get('bar') and charts['bar'] and os.path.exists(charts['bar']):
+            elements.append(Paragraph("Material Comparison", styles['Heading3']))
+            elements.append(Image(charts['bar'], width=6*inch, height=3*inch))
+            elements.append(Spacer(1, 20))
+            charts_added = True
+        
+        if charts.get('shift') and charts['shift'] and os.path.exists(charts['shift']):
+            elements.append(Paragraph("Shift Analysis", styles['Heading3']))
+            elements.append(Image(charts['shift'], width=6*inch, height=3*inch))
+            elements.append(Spacer(1, 20))
+            charts_added = True
+        
+        if not charts_added:
+            elements.append(Paragraph("Charts could not be generated. Please install kaleido: pip install kaleido", styles['Normal']))
+            elements.append(Spacer(1, 20))
+            
+    except ImportError:
+        # Kaleido not available - add text summary instead
+        elements.append(Paragraph("Charts Not Available", styles['Heading3']))
+        elements.append(Paragraph("To enable charts in PDF reports, please install kaleido:", styles['Normal']))
+        elements.append(Paragraph("<b>pip install kaleido</b>", styles['Normal']))
+        elements.append(Spacer(1, 10))
+        
+        # Add text-based data summary instead
+        elements.append(Paragraph("Production Summary (Text Format):", styles['Normal']))
+        for material, info in stats.items():
+            if material != '_total_':
+                summary_text = f"• {material.replace('_', ' ').title()}: {info['total']:,.0f} kg ({info['percentage']:.1f}%)"
+                elements.append(Paragraph(summary_text, styles['Normal']))
         elements.append(Spacer(1, 20))
     
     # Quality Analysis