Update app.py

app.py

@@ -172,7 +172,7 @@ if st.session_state.df is not None:
         context=[analyze_data],
     )
 
-    # ✅ Single Crew to handle both report and conclusion generation
+    # 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],
@@ -190,10 +190,10 @@ 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()
 
-                main_report = result_dict.get('write_report', '')
-                conclusion = result_dict.get('write_conclusion', '')
+                # ✅ 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."
 
                 # Step 1: Display Analysis Report
                 st.markdown("### Analysis Report:")
@@ -248,6 +248,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:")