Update app.py

app.py

@@ -457,6 +457,8 @@ def generate_visuals(query, df, llm, result_container, progress_bar, step):
     progress_bar.progress(step + 1, text="✅ Visualizations Ready!")
 
 
+import threading
+
 # SQL-RAG Analysis
 if st.session_state.df is not None:
     temp_dir = tempfile.TemporaryDirectory()
@@ -530,7 +532,7 @@ 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 and Key Insights. DO NOT include any Conclusion or Summary.",
         expected_output="Markdown-formatted report excluding Conclusion.",
         agent=report_writer,
         context=[analyze_data],
@@ -538,14 +540,12 @@ if st.session_state.df is not None:
 
     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.",
+                    "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.",
         agent=conclusion_writer,
         context=[analyze_data],
     )
 
-
-
     # Separate Crews for report and conclusion
     crew_report = Crew(
         agents=[sql_dev, data_analyst, report_writer],
@@ -568,38 +568,60 @@ if st.session_state.df is not None:
     with tab1:
         query = st.text_area("Enter Query:", value="Provide insights into the salary of a Principal Data Scientist.")
         if st.button("Submit Query"):
-            with st.spinner("Processing query..."):
-                # Step 1: Generate the analysis report
+            result_container = {"report": None, "conclusion": None, "visuals": None}
+            progress_bar = st.progress(0, text="🚀 Starting Analysis...")
+
+            # Define parallel tasks
+            def generate_report():
+                progress_bar.progress(20, text="📝 Generating Analysis Report...")
                 report_inputs = {"query": query + " Provide detailed analysis but DO NOT include Conclusion."}
-                report_result = crew_report.kickoff(inputs=report_inputs)
+                result_container['report'] = crew_report.kickoff(inputs=report_inputs)
+                progress_bar.progress(40, text="✅ Analysis Report Ready!")
 
-                # Step 2: Generate only the concise conclusion
+            def generate_conclusion():
+                progress_bar.progress(40, text="📝 Crafting Conclusion...")
                 conclusion_inputs = {"query": query + " Provide ONLY the most important insights in 3-5 concise lines."}
-                conclusion_result = crew_conclusion.kickoff(inputs=conclusion_inputs)
-
-                # Step 3: Display the report
-                #st.markdown("### Analysis Report:")
-                st.markdown(report_result if report_result else "⚠️ No Report Generated.")
-
-                # Step 4: Suggest Visualizations
-                visualization_suggestions = ask_gpt4o_for_visualization(query, st.session_state.df, llm)
-
-
-
-                # Step 5: Insert Visual Insights
-                st.markdown("### Visual Insights")
-                if visualization_suggestions:
-                    handle_visualization_suggestions(visualization_suggestions, st.session_state.df)
-                else:
-                    st.warning("⚠️ No suitable visualizations to display.")
-
-
-
-                # Step 6: Display Concise Conclusion
-                #st.markdown("#### Conclusion")
-
-                safe_conclusion = escape_markdown(conclusion_result if conclusion_result else "⚠️ No Conclusion Generated.")
-                st.markdown(safe_conclusion)
+                result_container['conclusion'] = crew_conclusion.kickoff(inputs=conclusion_inputs)
+                progress_bar.progress(60, text="✅ Conclusion Ready!")
+
+            def generate_visuals():
+                progress_bar.progress(60, text="📊 Creating Visualizations...")
+                result_container['visuals'] = ask_gpt4o_for_visualization(query, st.session_state.df, llm)
+                progress_bar.progress(80, text="✅ Visualizations Ready!")
+
+            # Run tasks in parallel
+            thread_report = threading.Thread(target=generate_report)
+            thread_conclusion = threading.Thread(target=generate_conclusion)
+            thread_visuals = threading.Thread(target=generate_visuals)
+
+            thread_report.start()
+            thread_conclusion.start()
+            thread_visuals.start()
+
+            # Wait for all threads to finish
+            thread_report.join()
+            thread_conclusion.join()
+            thread_visuals.join()
+
+            progress_bar.progress(100, text="✅ Full Analysis Complete!")
+            time.sleep(0.5)
+            progress_bar.empty()
+
+            # Display Report
+            st.markdown("## 📊 Analysis Report")
+            st.markdown(result_container['report'] if result_container['report'] else "⚠️ No Report Generated.")
+
+            # Display Visual Insights
+            st.markdown("## 📈 Visual Insights")
+            if result_container['visuals']:
+                handle_visualization_suggestions(result_container['visuals'], st.session_state.df)
+            else:
+                st.warning("⚠️ No suitable visualizations to display.")
+
+            # Display Conclusion
+            st.markdown("## 📝 Conclusion")
+            safe_conclusion = escape_markdown(result_container['conclusion'] if result_container['conclusion'] else "⚠️ No Conclusion Generated.")
+            st.markdown(safe_conclusion)
 
 
 # Sidebar Reference