Update app.py

app.py

@@ -172,6 +172,7 @@ if st.session_state.df is not None:
         context=[analyze_data],
     )
 
+    # ✅ Single Crew to handle both report and conclusion generation
     crew = Crew(
         agents=[sql_dev, data_analyst, report_writer, conclusion_writer],
         tasks=[extract_data, analyze_data, write_report, write_conclusion],
@@ -189,15 +190,16 @@ if st.session_state.df is not None:
             with st.spinner("Processing query..."):
                 inputs = {"query": query}
                 result = crew.kickoff(inputs=inputs)
+                result_dict = result.to_dict()
 
-                # Assuming result is structured as a dictionary:
-                main_report = result.get('write_report', '')
-                conclusion = result.get('write_conclusion', '')
+                main_report = result_dict.get('write_report', '')
+                conclusion = result_dict.get('write_conclusion', '')
 
+                # Step 1: Display Analysis Report
                 st.markdown("### Analysis Report:")
                 st.markdown(main_report)
 
-                # Step 3: Generate relevant visualizations
+                # Step 2: Generate relevant visualizations
                 visualizations = []
 
                 fig_salary = px.box(st.session_state.df, x="job_title", y="salary_in_usd",
@@ -215,12 +217,12 @@ if st.session_state.df is not None:
                                         title="Salary Distribution by Employment Type")
                 visualizations.append(fig_employment)
 
-                # Step 4: Insert Visual Insights
+                # Step 3: Insert Visual Insights
                 st.markdown("## 📊 Visual Insights")
                 for fig in visualizations:
                     st.plotly_chart(fig, use_container_width=True)
 
-                # Step 5: Append the Conclusion
+                # Step 4: Append the Conclusion
                 st.markdown("## Conclusion")
                 st.markdown(conclusion)
 
@@ -249,4 +251,4 @@ else:
 # 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)")