Update app.py

app.py

@@ -149,7 +149,7 @@ if st.session_state.df is not None:
         llm=llm,
     )
 
-    # Tasks for each agent
+    # Tasks for Report
     extract_data = Task(
         description="Extract data based on the query: {query}.",
         expected_output="Database results matching the query.",
@@ -170,6 +170,7 @@ if st.session_state.df is not None:
         context=[analyze_data],
     )
 
+    # Task for Conclusion
     write_conclusion = Task(
         description=(
             "Summarize the findings into a concise Conclusion. Include the max, min, and average salary in USD, "
@@ -180,10 +181,17 @@ if st.session_state.df is not None:
         context=[analyze_data],
     )
 
-    # Crew setup
-    crew = Crew(
-        agents=[sql_dev, data_analyst, report_writer, conclusion_writer],
-        tasks=[extract_data, analyze_data, write_report, write_conclusion],
+    # Separate Crews 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,
     )
@@ -197,16 +205,16 @@ if st.session_state.df is not None:
         if st.button("Submit Query"):
             with st.spinner("Processing query..."):
                 # Step 1: Generate the main report (without Conclusion)
-                report_inputs = {"query": query + " Provide a detailed analysis but DO NOT include a Conclusion."}
-                report_result = crew.kickoff(inputs=report_inputs)
+                report_inputs = {"query": query}
+                report_result = crew_report.kickoff(inputs=report_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)
+                conclusion_inputs = {"query": query}
+                conclusion_result = crew_conclusion.kickoff(inputs=conclusion_inputs)
 
-                # Handle Crew Output safely
-                main_report = getattr(report_result, 'output', "⚠️ No Report Generated.")
-                conclusion = getattr(conclusion_result, 'output', "⚠️ No Conclusion Generated.")
+                # Extract results safely
+                main_report = report_result[0].output if report_result and hasattr(report_result[0], 'output') else "⚠️ No Report Generated."
+                conclusion = conclusion_result[0].output if conclusion_result and hasattr(conclusion_result[0], 'output') else "⚠️ No Conclusion Generated."
 
                 st.markdown("### Analysis Report:")
                 st.markdown(main_report)
@@ -260,6 +268,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:")