Update app.py

app.py

@@ -172,17 +172,10 @@ if st.session_state.df is not None:
         context=[analyze_data],
     )
 
-    # Crew with separate tasks for report and conclusion
-    crew_report = Crew(
-        agents=[sql_dev, data_analyst, report_writer],
-        tasks=[extract_data, analyze_data, write_report],
-        process=Process.sequential,
-        verbose=True,
-    )
-
-    crew_conclusion = Crew(
-        agents=[data_analyst, conclusion_writer],
-        tasks=[write_conclusion],
+    # ✅ Optimized Single Crew for Report and Conclusion
+    crew = Crew(
+        agents=[sql_dev, data_analyst, report_writer, conclusion_writer],
+        tasks=[extract_data, analyze_data, write_report, write_conclusion],
         process=Process.sequential,
         verbose=True,
     )
@@ -195,15 +188,15 @@ if st.session_state.df is not None:
         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 main report (without Conclusion)
-                report_inputs = {"query": query}
-                report_result = crew_report.kickoff(inputs=report_inputs)
+                inputs = {"query": query}
+                result = crew.kickoff(inputs=inputs)
 
-                # Step 2: Generate ONLY the Conclusion
-                conclusion_inputs = {"query": query}
-                conclusion_result = crew_conclusion.kickoff(inputs=conclusion_inputs)
+                # Assuming result is structured as a dictionary:
+                main_report = result.get('write_report', '')
+                conclusion = result.get('write_conclusion', '')
 
                 st.markdown("### Analysis Report:")
+                st.markdown(main_report)
 
                 # Step 3: Generate relevant visualizations
                 visualizations = []
@@ -223,17 +216,14 @@ if st.session_state.df is not None:
                                         title="Salary Distribution by Employment Type")
                 visualizations.append(fig_employment)
 
-                # Step 4: Display the main report
-                st.markdown(report_result)
-
-                # Step 5: Insert Visual Insights
+                # Step 4: Insert Visual Insights
                 st.markdown("## 📊 Visual Insights")
                 for fig in visualizations:
                     st.plotly_chart(fig, use_container_width=True)
 
-                # Step 6: Append the Conclusion
+                # Step 5: Append the Conclusion
                 st.markdown("## Conclusion")
-                st.markdown(conclusion_result)
+                st.markdown(conclusion)
 
     # Tab 2: Full Data Visualization
     with tab2:
@@ -257,7 +247,6 @@ if st.session_state.df is not None:
 else:
     st.info("Please load a dataset to proceed.")
 
-
 # Sidebar Reference
 with st.sidebar:
     st.header("📚 Reference:")