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],
+    # Crew setup
+    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,23 +188,22 @@ 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)
+                # Step 1: Generate Report without Conclusion
+                report_inputs = {"query": query + " Provide a detailed analysis but DO NOT include a Conclusion."}
+                report_result = crew.kickoff(inputs=report_inputs)
 
-                # Step 2: Generate ONLY the Conclusion
-                conclusion_inputs = {"query": query}
-                conclusion_result = crew_conclusion.kickoff(inputs=conclusion_inputs)
+                # Step 2: Generate only the Conclusion
+                conclusion_inputs = {"query": query + " Now, provide only the Conclusion for this analysis."}
+                conclusion_result = crew.kickoff(inputs=conclusion_inputs)
 
-                # ✅ Correct way to access results
-                main_report = report_result.get("write_report", "⚠️ No Report Generated.")
-                conclusion = conclusion_result.get("write_conclusion", "⚠️ No Conclusion Generated.")
+                # Directly use the outputs 
+                main_report = report_result if report_result else "⚠️ No Report Generated."
+                conclusion = conclusion_result if conclusion_result else "⚠️ No Conclusion Generated."
 
-                # Step 3: Display the main report
                 st.markdown("### Analysis Report:")
                 st.markdown(main_report)
 
-                # Step 4: Generate relevant visualizations
+                # Step 3: Generate relevant visualizations
                 visualizations = []
 
                 fig_salary = px.box(st.session_state.df, x="job_title", y="salary_in_usd",
@@ -229,12 +221,12 @@ if st.session_state.df is not None:
                                         title="Salary Distribution by Employment Type")
                 visualizations.append(fig_employment)
 
-                # 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)