Update app.py

app.py

@@ -85,6 +85,8 @@ 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)
@@ -139,7 +141,10 @@ if st.session_state.df is not None:
     # New Agent for generating ONLY the Conclusion
     conclusion_writer = Agent(
         role="Conclusion Specialist",
-        goal="Summarize findings into a clear and concise Conclusion/Summary section.",
+        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."
+        ),
         backstory="An expert in crafting well-structured and insightful conclusions.",
         llm=llm,
     )
@@ -166,8 +171,11 @@ if st.session_state.df is not None:
     )
 
     write_conclusion = Task(
-        description="Summarize the findings into a concise Conclusion.",
-        expected_output="Markdown-formatted Conclusion section.",
+        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.",
         agent=conclusion_writer,
         context=[analyze_data],
     )
@@ -188,7 +196,7 @@ 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 Report without Conclusion
+                # 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)
 
@@ -196,9 +204,9 @@ if st.session_state.df is not None:
                 conclusion_inputs = {"query": query + " Now, provide only the Conclusion for this analysis."}
                 conclusion_result = crew.kickoff(inputs=conclusion_inputs)
 
-                # 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."
+                # Handle Crew Output safely
+                main_report = getattr(report_result, 'output', "⚠️ No Report Generated.")
+                conclusion = getattr(conclusion_result, 'output', "⚠️ No Conclusion Generated.")
 
                 st.markdown("### Analysis Report:")
                 st.markdown(main_report)
@@ -252,8 +260,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:")
-    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)")