Update app.py

app.py

@@ -546,69 +546,54 @@ if st.session_state.df is not None:
     )
 
     # Tabs for Query Results and Visualizations
-    tab1, tab2 = st.tabs(["🔍 Query Insights + Viz", "📊 Full Data Viz"])
+    tab1 = st.tabs(["🔍 Query Insights + Viz", "📊 Full Data Viz"])
 
     # Query Insights + Visualization
     with tab1:
         query = st.text_area("Enter Query:", value="Provide insights into the salary of a Principal Data Scientist.")
         if st.button("Submit Query"):
-            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...")
+            with st.spinner("Processing query..."):
+                # Step 1: Generate the 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(40, text="✅ Analysis Report Ready!")
+                report_result = crew_report.kickoff(inputs=report_inputs)
 
-            def generate_conclusion():
-                progress_bar.progress(40, text="📝 Crafting Conclusion...")
+                # Step 2: Generate only the concise 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(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!")
+                conclusion_result = crew_conclusion.kickoff(inputs=conclusion_inputs)
 
-            # 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)
+                # Step 3: Display the report
+                #st.markdown("### Analysis Report:")
+                st.markdown(report_result if report_result else "⚠️ No Report Generated.")
 
-            thread_report.start()
-            thread_conclusion.start()
-            thread_visuals.start()
+                # Step 4: Generate Visualizations
+                visualizations = []
 
-            # Wait for all threads to finish
-            thread_report.join()
-            thread_conclusion.join()
-            thread_visuals.join()
+                fig_salary = px.box(st.session_state.df, x="job_title", y="salary_in_usd",
+                                    title="Salary Distribution by Job Title")
+                visualizations.append(fig_salary)
 
-            progress_bar.progress(100, text="✅ Full Analysis Complete!")
-            time.sleep(0.5)
-            progress_bar.empty()
+                fig_experience = px.bar(
+                    st.session_state.df.groupby("experience_level")["salary_in_usd"].mean().reset_index(),
+                    x="experience_level", y="salary_in_usd",
+                    title="Average Salary by Experience Level"
+                )
+                visualizations.append(fig_experience)
 
-            # Display Report
-            st.markdown("## 📊 Analysis Report")
-            st.markdown(result_container['report'] if result_container['report'] else "⚠️ No Report Generated.")
+                fig_employment = px.box(st.session_state.df, x="employment_type", y="salary_in_usd",
+                                        title="Salary Distribution by Employment Type")
+                visualizations.append(fig_employment)
 
-            # 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.")
+                # Step 5: Insert Visual Insights
+                st.markdown("### Visual Insights")
+                for fig in visualizations:
+                    st.plotly_chart(fig, use_container_width=True)
 
-            # Display Conclusion
-            st.markdown("## 📝 Conclusion")
-            safe_conclusion = escape_markdown(result_container['conclusion'] if result_container['conclusion'] else "⚠️ No Conclusion Generated.")
-            st.markdown(safe_conclusion)
+                # Step 6: Display Concise Conclusion
+                #st.markdown("#### Conclusion")
+                st.markdown(conclusion_result if conclusion_result else "⚠�� No Conclusion Generated.")
 
 
 # Sidebar Reference
 with st.sidebar:
     st.header("📚 Reference:")
-    st.markdown("[SQL Agents w CrewAI & Llama 3 - Plaban Nayak](https://github.com/plaban1981/Agents/blob/main/SQL_Agents_with_CrewAI_and_Llama_3.ipynb)")
+    st.markdown("[SQL Agents w CrewAI & Llama 3 - Plaban Nayak](https://github.com/plaban1981/Agents/blob/main/SQL_Agents_with_CrewAI_and_Llama_3.ipynb)")