Update app.py

app.py

@@ -85,6 +85,36 @@ 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())
 
+# Function to create TXT file
+def generate_txt(content):
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".txt") as tmp_file:
+        tmp_file.write(content.encode('utf-8'))
+        return tmp_file.name
+
+# Function to create PDF with report text and visualizations
+def generate_pdf(report_text, figures):
+    pdf = FPDF()
+    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
+
+
+
 # SQL-RAG Analysis
 if st.session_state.df is not None:
     temp_dir = tempfile.TemporaryDirectory()