Update app.py

app.py

@@ -92,26 +92,61 @@ def generate_txt(content):
         return tmp_file.name
 
 # Function to create PDF with report text and visualizations
-def generate_pdf(report_text, figures):
+def create_pdf_report_with_viz(report, conclusion, visualizations):
     pdf = FPDF()
+    pdf.set_auto_page_break(auto=True, margin=15)
     pdf.add_page()
     pdf.set_font("Arial", size=12)
-    
-    # Add report text
-    for line in report_text.split('\n'):
-        pdf.multi_cell(0, 10, line)
-    
-    # Add visualizations
-    for fig in figures:
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as img_file:
-            fig.write_image(img_file.name)
-            pdf.add_page()
-            pdf.image(img_file.name, x=10, y=10, w=190)
-    
-    # Save PDF
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_pdf:
-        pdf.output(tmp_pdf.name)
-        return tmp_pdf.name
+
+    # Title
+    pdf.set_font("Arial", style="B", size=18)
+    pdf.cell(0, 10, "📊 Analysis Report", ln=True, align="C")
+    pdf.ln(10)
+
+    # Report Content
+    pdf.set_font("Arial", style="B", size=14)
+    pdf.cell(0, 10, "Analysis", ln=True)
+    pdf.set_font("Arial", size=12)
+    pdf.multi_cell(0, 10, report)
+
+    pdf.ln(10)
+    pdf.set_font("Arial", style="B", size=14)
+    pdf.cell(0, 10, "Conclusion", ln=True)
+    pdf.set_font("Arial", size=12)
+    pdf.multi_cell(0, 10, conclusion)
+
+    # Add Visualizations
+    pdf.add_page()
+    pdf.set_font("Arial", style="B", size=16)
+    pdf.cell(0, 10, "📈 Visualizations", ln=True)
+    pdf.ln(5)
+
+    with tempfile.TemporaryDirectory() as temp_dir:
+        for i, fig in enumerate(visualizations, start=1):
+            fig_title = fig.layout.title.text if fig.layout.title.text else f"Visualization {i}"
+            x_axis = fig.layout.xaxis.title.text if fig.layout.xaxis.title.text else "X-axis"
+            y_axis = fig.layout.yaxis.title.text if fig.layout.yaxis.title.text else "Y-axis"
+
+            # Save each visualization as a PNG image
+            img_path = os.path.join(temp_dir, f"viz_{i}.png")
+            fig.write_image(img_path)
+
+            # Insert Title and Description
+            pdf.set_font("Arial", style="B", size=14)
+            pdf.multi_cell(0, 10, f"{i}. {fig_title}")
+            pdf.set_font("Arial", size=12)
+            pdf.multi_cell(0, 10, f"X-axis: {x_axis} | Y-axis: {y_axis}")
+            pdf.ln(3)
+
+            # Embed Visualization
+            pdf.image(img_path, w=170)
+            pdf.ln(10)
+
+        # Save PDF
+        temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+        pdf.output(temp_pdf.name)
+
+    return temp_pdf