Update app.py

app.py

@@ -85,8 +85,6 @@ if st.session_state.df is not None and st.session_state.show_preview:
 
 # SQL-RAG Analysis
 if st.session_state.df is not None:
-    import numpy as np
-
     temp_dir = tempfile.TemporaryDirectory()
     db_path = os.path.join(temp_dir.name, "data.db")
     connection = sqlite3.connect(db_path)
@@ -141,15 +139,12 @@ if st.session_state.df is not None:
     # New Agent for generating ONLY the Conclusion
     conclusion_writer = Agent(
         role="Conclusion Specialist",
-        goal=(
-            "Generate a concise 3-5 line conclusion including the maximum, minimum, and average salary "
-            "of Principal Data Scientists. Highlight how geography, experience, and employment type impact salary."
-        ),
+        goal="Summarize findings into a clear and concise Conclusion/Summary section.",
         backstory="An expert in crafting well-structured and insightful conclusions.",
         llm=llm,
     )
 
-    # Tasks for Report
+    # Tasks for each agent
     extract_data = Task(
         description="Extract data based on the query: {query}.",
         expected_output="Database results matching the query.",
@@ -170,28 +165,17 @@ 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, "
-            "and highlight the most impactful insights."
-        ),
-        expected_output="Markdown-formatted Conclusion section with key statistics.",
+        description="Summarize the findings into a concise Conclusion.",
+        expected_output="Markdown-formatted Conclusion section.",
         agent=conclusion_writer,
         context=[analyze_data],
     )
 
-    # 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],
+    # 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,
     )
@@ -204,17 +188,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..."):
-                # 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)
+                conclusion_inputs = {"query": query + " Now, provide only the Conclusion for this analysis."}
+                conclusion_result = crew.kickoff(inputs=conclusion_inputs)
 
-                # 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."
+                # 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."
 
                 st.markdown("### Analysis Report:")
                 st.markdown(main_report)
@@ -272,4 +256,5 @@ 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)")
+