Update app.py

app.py

@@ -3,6 +3,7 @@ import pandas as pd
 import sqlite3
 import tempfile
 from fpdf import FPDF
+import time
 import os
 import re
 import json
@@ -436,6 +437,24 @@ def escape_markdown(text):
     escape_chars = r"(\*|_|`|~)"
     return re.sub(escape_chars, r"\\\1", text)
 
+
+def generate_report(query, crew_report, result_container, progress_bar, step):
+    progress_bar.progress(step, text="📝 Generating Analysis Report...")
+    report_inputs = {"query": query + " Provide detailed analysis but DO NOT include Conclusion."}
+    result_container['report'] = crew_report.kickoff(inputs=report_inputs)
+    progress_bar.progress(step + 1, text="✅ Analysis Report Ready!")
+
+def generate_conclusion(query, crew_conclusion, result_container, progress_bar, step):
+    progress_bar.progress(step, text="📝 Crafting Conclusion...")
+    conclusion_inputs = {"query": query + " Provide ONLY the most important insights in 3-5 concise lines."}
+    result_container['conclusion'] = crew_conclusion.kickoff(inputs=conclusion_inputs)
+    progress_bar.progress(step + 1, text="✅ Conclusion Ready!")
+
+def generate_visuals(query, df, llm, result_container, progress_bar, step):
+    progress_bar.progress(step, text="📊 Creating Visualizations...")
+    result_container['visuals'] = ask_gpt4o_for_visualization(query, df, llm)
+    progress_bar.progress(step + 1, text="✅ Visualizations Ready!")
+
 # SQL-RAG Analysis
 if st.session_state.df is not None:
     temp_dir = tempfile.TemporaryDirectory()