Update app.py

app.py

@@ -111,6 +111,7 @@ if st.session_state.df is not None:
         """Validate the SQL query syntax and structure before execution."""
         return QuerySQLCheckerTool(db=db, llm=llm).invoke({"query": sql_query})
 
+    # Agent for SQL data extraction
     sql_dev = Agent(
         role="Senior Database Developer",
         goal="Extract data using optimized SQL queries.",
@@ -119,6 +120,7 @@ if st.session_state.df is not None:
         tools=[list_tables, tables_schema, execute_sql, check_sql],
     )
 
+    # Agent for data analysis
     data_analyst = Agent(
         role="Senior Data Analyst",
         goal="Analyze the data and produce insights.",
@@ -126,13 +128,23 @@ if st.session_state.df is not None:
         llm=llm,
     )
 
+    # Agent for generating the main report (without Conclusion)
     report_writer = Agent(
         role="Technical Report Writer",
-        goal="Summarize the insights into a clear report.",
-        backstory="An expert in summarizing data insights into readable reports.",
+        goal="Write a clear, structured report containing ONLY Key Insights and Analysis. NO Introduction, Summary, or Conclusion.",
+        backstory="An expert in crafting data-driven reports with clear insights.",
         llm=llm,
     )
 
+    # New Agent for generating ONLY the Conclusion
+    conclusion_writer = Agent(
+        role="Conclusion Specialist",
+        goal="Summarize findings into a clear and concise Conclusion section.",
+        backstory="An expert in crafting well-structured and insightful conclusions.",
+        llm=llm,
+    )
+
+    # Tasks for each agent
     extract_data = Task(
         description="Extract data based on the query: {query}.",
         expected_output="Database results matching the query.",
@@ -141,25 +153,40 @@ if st.session_state.df is not None:
 
     analyze_data = Task(
         description="Analyze the extracted data for query: {query}.",
-        expected_output="Provide only the Introduction, Key Insights, and Analysis sections. Do NOT include any conclusion, summary, or closing remarks.",
+        expected_output="Provide ONLY Key Insights and Analysis. Exclude Introduction and Conclusion.",
         agent=data_analyst,
         context=[extract_data],
     )
 
     write_report = Task(
-        description="Summarize the analysis into a structured report with Introduction, Key Insights, and Analysis. DO NOT include any conclusion or summary.",
-        expected_output="Markdown report excluding Conclusion and Summary.",
+        description="Write the report with ONLY Key Insights and Analysis. DO NOT include Introduction or Conclusion.",
+        expected_output="Markdown report excluding Introduction and Conclusion.",
         agent=report_writer,
         context=[analyze_data],
     )
 
-    crew = Crew(
+    write_conclusion = Task(
+        description="Summarize the findings into a concise Conclusion.",
+        expected_output="Markdown-formatted Conclusion section.",
+        agent=conclusion_writer,
+        context=[analyze_data],
+    )
+
+    # Crew with separate tasks 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],
+        process=Process.sequential,
+        verbose=True,
+    )
+
     # Tabs for Query Results and General Insights
     tab1, tab2 = st.tabs(["🔍 Query Insights + Viz", "📊 Full Data Viz"])
 
@@ -168,15 +195,13 @@ 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 structured report (without Conclusion)
-                report_prompt = f"{query} Provide only the Introduction, Key Insights, and Analysis sections. Do NOT include Conclusion or Summary."
-                inputs = {"query": report_prompt}
-                report_result = crew.kickoff(inputs=inputs)
+                # Step 1: Generate the main report (without Conclusion)
+                report_inputs = {"query": query}
+                report_result = crew_report.kickoff(inputs=report_inputs)
 
                 # Step 2: Generate ONLY the Conclusion
-                conclusion_prompt = f"{query} Provide ONLY the Conclusion for this analysis."
-                conclusion_inputs = {"query": conclusion_prompt}
-                conclusion_result = crew.kickoff(inputs=conclusion_inputs)
+                conclusion_inputs = {"query": query}
+                conclusion_result = crew_conclusion.kickoff(inputs=conclusion_inputs)
 
                 st.markdown("### Analysis Report:")
 
@@ -198,7 +223,7 @@ if st.session_state.df is not None:
                                         title="Salary Distribution by Employment Type")
                 visualizations.append(fig_employment)
 
-                # Step 4: Display report (without Conclusion)
+                # Step 4: Display the main report
                 st.markdown(report_result)
 
                 # Step 5: Insert Visual Insights
@@ -232,6 +257,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:")