Update app.py

app.py

@@ -172,10 +172,17 @@ if st.session_state.df is not None:
         context=[analyze_data],
     )
 
-    # 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],
+    # 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],
         process=Process.sequential,
         verbose=True,
     )
@@ -188,18 +195,23 @@ 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..."):
-                inputs = {"query": query}
-                result = crew.kickoff(inputs=inputs)
+                # Step 1: Generate the main report (without Conclusion)
+                report_inputs = {"query": query}
+                report_result = crew_report.kickoff(inputs=report_inputs)
+
+                # Step 2: Generate ONLY the Conclusion
+                conclusion_inputs = {"query": query}
+                conclusion_result = crew_conclusion.kickoff(inputs=conclusion_inputs)
 
-                # ✅ Corrected result access
-                main_report = result.tasks[2].output if result.tasks[2].output else "⚠️ No Report Generated."
-                conclusion = result.tasks[3].output if result.tasks[3].output else "⚠️ No Conclusion Generated."
+                # Check if results are empty
+                main_report = report_result.output if report_result.output else "⚠️ No Report Generated."
+                conclusion = conclusion_result.output if conclusion_result.output else "⚠️ No Conclusion Generated."
 
-                # Step 1: Display Analysis Report
+                # Step 3: Display the main report
                 st.markdown("### Analysis Report:")
                 st.markdown(main_report)
 
-                # Step 2: Generate relevant visualizations
+                # Step 4: Generate relevant visualizations
                 visualizations = []
 
                 fig_salary = px.box(st.session_state.df, x="job_title", y="salary_in_usd",
@@ -217,12 +229,12 @@ if st.session_state.df is not None:
                                         title="Salary Distribution by Employment Type")
                 visualizations.append(fig_employment)
 
-                # Step 3: Insert Visual Insights
+                # Step 5: Insert Visual Insights
                 st.markdown("## 📊 Visual Insights")
                 for fig in visualizations:
                     st.plotly_chart(fig, use_container_width=True)
 
-                # Step 4: Append the Conclusion
+                # Step 6: Append the Conclusion
                 st.markdown("## Conclusion")
                 st.markdown(conclusion)
 
@@ -248,8 +260,7 @@ 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:")
-    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)")