Update app.py

app.py

@@ -141,14 +141,14 @@ if st.session_state.df is not None:
 
     analyze_data = Task(
         description="Analyze the extracted data for query: {query}.",
-        expected_output="Analysis text summarizing findings.",
+        expected_output="Analysis text summarizing findings (without a Conclusion section).",
         agent=data_analyst,
         context=[extract_data],
     )
 
     write_report = Task(
-        description="Summarize the analysis into an executive report.",
-        expected_output="Markdown report of insights.",
+        description="Summarize the analysis into an executive report without a Conclusion.",
+        expected_output="Markdown report of insights without Conclusion.",
         agent=report_writer,
         context=[analyze_data],
     )
@@ -168,12 +168,17 @@ 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 Report WITHOUT Conclusion
+                inputs = {"query": query + " Provide a detailed analysis but DO NOT include a Conclusion."}
+                report_result = crew.kickoff(inputs=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)
 
                 st.markdown("### Analysis Report:")
 
-                # Generate relevant visualizations
+                # Step 3: Generate relevant visualizations
                 visualizations = []
 
                 fig_salary = px.box(st.session_state.df, x="job_title", y="salary_in_usd",
@@ -191,26 +196,17 @@ if st.session_state.df is not None:
                                         title="Salary Distribution by Employment Type")
                 visualizations.append(fig_employment)
 
-                # Split the result and insert visualizations before Conclusion
-                insert_section = "## Conclusion"
-                if insert_section in result:
-                    parts = result.split(insert_section, 1)
-                    st.markdown(parts[0])  # Before Conclusion
-
-                    # Insert Visual Insights Section before Conclusion
-                    st.markdown("## 📊 Visual Insights")
-                    for fig in visualizations:
-                        st.plotly_chart(fig, use_container_width=True)
-
-                    # Show Conclusion
-                    st.markdown(insert_section)
-                    st.markdown(parts[1])
-                else:
-                    # Default if Conclusion not found
-                    st.markdown(result)
-                    st.markdown("## 📊 Visual Insights")
-                    for fig in visualizations:
-                        st.plotly_chart(fig, use_container_width=True)
+                # Step 4: Display report without conclusion
+                st.markdown(report_result)
+
+                # Step 5: Insert Visual Insights
+                st.markdown("## 📊 Visual Insights")
+                for fig in visualizations:
+                    st.plotly_chart(fig, use_container_width=True)
+
+                # Step 6: Append the Conclusion
+                st.markdown("## Conclusion")
+                st.markdown(conclusion_result)
 
     # Tab 2: Full Data Visualization
     with tab2: