Update app.py

app.py

@@ -86,6 +86,71 @@ if st.session_state.df is not None and st.session_state.show_preview:
     st.subheader("📂 Dataset Preview")
     st.dataframe(st.session_state.df.head())
 
+# Helper Function to Create a PDF Report with Visualizations and Descriptions
+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)
+
+    # 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 with Descriptions
+    pdf.add_page()
+    pdf.set_font("Arial", style="B", size=16)
+    pdf.cell(0, 10, "📈 Visualizations", ln=True)
+    pdf.ln(5)
+
+    temp_images = []  # Store temp image paths for cleanup
+
+    for i, fig in enumerate(visualizations, start=1):
+        # Extract Figure Title and Axis Labels
+        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 the figure temporarily
+        img_temp = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
+        fig.write_image(img_temp.name)
+        temp_images.append(img_temp.name)  # Keep track for cleanup
+
+        # 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 with Scaling
+        pdf.image(img_temp.name, w=170)  # Auto-scaled width to fit the page
+        pdf.ln(10)
+
+    # Save the PDF to a temporary file
+    temp_pdf = tempfile.NamedTemporaryFile(delete=False, suffix=".pdf")
+    pdf.output(temp_pdf.name)
+
+    # Clean up temporary images
+    for img_path in temp_images:
+        if os.path.exists(img_path):
+            os.remove(img_path)
+
+    return temp_pdf
+
     
 # Helper function to create a plain text report with visualization summaries
 def create_text_report_with_viz(report, conclusion, visualizations):