Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import pandas as pd
 import sqlite3
 import os
+import io
 import json
 from pathlib import Path
 import plotly.express as px
@@ -83,6 +84,37 @@ 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 plain text report with visualization summaries
+def create_text_report_with_viz(report, conclusion, visualizations):
+    content = f"### Analysis Report\n\n{report}\n\n### Conclusion\n\n{conclusion}\n\n### Visualizations\n"
+
+    # Dynamically add descriptions for each visualization
+    for i, fig in enumerate(visualizations, start=1):
+        # Extract the title of the Plotly figure if available
+        fig_title = fig.layout.title.text if fig.layout.title.text else f"Visualization {i}"
+        
+        # Add title and figure details
+        content += f"\n{i}. {fig_title}\n"
+
+        # Extract x and y axis titles if available
+        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"
+
+        content += f"   - X-axis: {x_axis}\n"
+        content += f"   - Y-axis: {y_axis}\n"
+
+        # If figure has data, summarize it
+        if fig.data:
+            trace_types = set(trace.type for trace in fig.data)
+            content += f"   - Chart Type(s): {', '.join(trace_types)}\n"
+        else:
+            content += "   - No data available in this visualization.\n"
+
+    # Return the content as a downloadable text stream
+    return io.BytesIO(content.encode("utf-8"))
+    
+
 # SQL-RAG Analysis
 if st.session_state.df is not None:
     temp_dir = tempfile.TemporaryDirectory()