Update app.py

app.py

@@ -110,44 +110,36 @@ def create_pdf_report_with_viz(report, conclusion, visualizations):
     pdf.set_font("Arial", size=12)
     pdf.multi_cell(0, 10, conclusion)
 
-    # Add Visualizations with Descriptions
+    # Add Visualizations
     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)
+    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