Update app.py

app.py

@@ -2,11 +2,8 @@ import streamlit as st
 import pandas as pd
 import sqlite3
 import os
-import io
 import json
 from pathlib import Path
-import tempfile
-from fpdf import FPDF
 import plotly.express as px
 from datetime import datetime, timezone
 from crewai import Agent, Crew, Process, Task
@@ -86,94 +83,6 @@ 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
-    pdf.add_page()
-    pdf.set_font("Arial", style="B", size=16)
-    pdf.cell(0, 10, "📈 Visualizations", ln=True)
-    pdf.ln(5)
-
-    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
-
-    
-# 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()
@@ -220,7 +129,7 @@ if st.session_state.df is not None:
 
     report_writer = Agent(
         role="Technical Report Writer",
-        goal="Write a structured report with Introduction and Key Insights. DO NOT include any Conclusion or Summary.",
+        goal="Write a structured report with Introduction, Key Insights, and Analysis. DO NOT include any Conclusion or Summary.",
         backstory="Specializes in detailed analytical reports without conclusions.",
         llm=llm,
     )
@@ -247,16 +156,16 @@ if st.session_state.df is not None:
     )
 
     write_report = Task(
-        description="Write the analysis report with  Introduction and Key Insights. DO NOT include any Conclusion or Summary.",
+        description="Write the analysis report with  Introduction, Key Insights, and Analysis. DO NOT include any Conclusion or Summary.",
         expected_output="Markdown-formatted report excluding Conclusion.",
         agent=report_writer,
         context=[analyze_data],
     )
 
     write_conclusion = Task(
-        description="Summarize the key findings in 3-5 impactful lines, highlighting the maximum, minimum, and average salaries."
-        "Emphasize significant insights on salary distribution and influential compensation trends for strategic decision-making.",
-        expected_output="Markdown-formatted Conclusion section with key insights and statistics.",
+        description="Write a brief and impactful 3-5 line Conclusion summarizing only the most important insights/findings. Include the max, min, and average salary"
+            "and highlight the most impactful insights.",
+        expected_output="Markdown-formatted Conclusion/Summary section with key insights and statistics.",
         agent=conclusion_writer,
         context=[analyze_data],
     )
@@ -315,7 +224,7 @@ if st.session_state.df is not None:
                 visualizations.append(fig_employment)
 
                 # Step 5: Insert Visual Insights
-                st.markdown("### Visual Insights")
+                st.markdown("#### 5. Visual Insights")
                 for fig in visualizations:
                     st.plotly_chart(fig, use_container_width=True)
 
@@ -323,28 +232,6 @@ 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")