Update app.py

app.py

@@ -323,6 +323,28 @@ if st.session_state.df is not None:
                 #st.markdown("#### 6. Conclusion")
                 st.markdown(conclusion_result if conclusion_result else "⚠️ No Conclusion Generated.")
 
+                # Step 7: PDF and TXT Download Buttons for Query Insights + Viz
+                if report_result and conclusion_result and visualizations:
+                    # PDF Download
+                    pdf_file = create_pdf_report_with_viz(report_result, conclusion_result, visualizations)
+                    with open(pdf_file.name, "rb") as f:
+                        st.download_button(
+                            label="📥 Download Full Report (PDF)",
+                            data=f,
+                            file_name="query_insights_report.pdf",
+                            mime="application/pdf"
+                        )
+
+                # TXT Download
+                text_file = create_text_report_with_viz(report_result, conclusion_result, visualizations)
+                st.download_button(
+                    label="📥 Download Full Report (TXT)",
+                    data=text_file,
+                    file_name="query_insights_report.txt",
+                    mime="text/plain"
+                )
+
+
     # Full Data Visualization Tab
     with tab2:
         st.subheader("📊 Comprehensive Data Visualizations")