Update app.py

app.py

@@ -221,7 +221,7 @@ crew = Crew(
 )
 
 # PDF Report Generation
-def generate_pdf_report(result, charts=None, table_data=None, metadata=None):
+def generate_pdf_report(result, charts=None, table_data=None, metadata=None, key_insights=None):
     with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as temp_pdf:
         pdf = FPDF()
         pdf.add_page()
@@ -240,25 +240,37 @@ def generate_pdf_report(result, charts=None, table_data=None, metadata=None):
         pdf.set_font("Arial", size=12)
         pdf.multi_cell(0, 10, txt=result)
 
+        # Insert Key Insights Section
+        if key_insights:
+            pdf.add_page()
+            pdf.set_font("Arial", size=14, style="B")
+            pdf.cell(200, 10, txt="Key Strategic Insights", ln=True)
+            pdf.ln(5)
+            pdf.set_font("Arial", size=12)
+            for insight in key_insights:
+                pdf.multi_cell(0, 10, txt=f"- {insight}")
+
+        # Insert Charts
         if charts:
             for chart_path in charts:
                 try:
                     pdf.add_page()
                     pdf.image(chart_path, x=10, y=20, w=180)
-                    logging.info(f"Successfully included chart: {chart_path}")
                 except Exception as e:
-                    logging.error(f"Failed to include chart in PDF: {chart_path}. Error: {e}")
+                    logging.error(f"Error including chart: {e}")
 
+        # Insert Tables
         if table_data:
             pdf.add_page()
             pdf.set_font("Arial", size=10)
-            pdf.cell(200, 10, txt="Consolidated Table:", ln=True, align="L")
+            pdf.cell(200, 10, txt="Consolidated Data Table:", ln=True, align="L")
             for row in table_data:
                 pdf.cell(200, 10, txt=str(row), ln=True)
 
         pdf.output(temp_pdf.name)
         return temp_pdf.name
 
+
 # Data Validation
 def validate_analyst_output(analyst_output):
     if not analyst_output: