Update interim.py

interim.py

@@ -1,8 +1,4 @@
 # add refrence (https://medium.com/@hanan.tabak/user-friendly-open-source-multi-agent-ai-business-consultant-on-crewai-and-streamlit-0f972feb1b74)  + improvize
-#=================
-# Import Libraries
-#=================
-
 import streamlit as st
 from crewai import Agent, Task, Crew
 import os
@@ -12,29 +8,12 @@ import pandas as pd
 import plotly.express as px
 import time
 
-#=================
-# Add Streamlit Components
-#=================
-
-# Background
-page_bg_img = '''
-<style>
-.stApp  {
-background-image: url("https://images.all-free-download.com/images/graphiclarge/abstract_bright_corporate_background_310453.jpg");
-background-size: cover;
-}
-</style>
-'''
-st.markdown(page_bg_img, unsafe_allow_html=True)
-
 # Title and Sidebar
 st.title("Multi-Agent Business Consultant")
 
-image_url = "https://cdn-icons-png.flaticon.com/512/1998/1998614.png"
-st.sidebar.image(image_url, caption="", use_container_width=True)
 st.sidebar.write(
-    "This AI Business Consultant is built using Multi-Agent system. "
-    "It provides business insights, statistical analysis, and professional recommendations!"
+    "This Business Consultant is built using Multi-Agent system. "
+    "Use this application to generate actionable business insights and data-driven analysis!"
 )
 
 # User Inputs
@@ -42,28 +21,32 @@ business = st.text_input('Enter The Required Business Search Area', value="Artif
 stakeholder = st.text_input('Enter The Stakeholder Team', value="Executives")
 
 # Optional Customization
-enable_customization = st.sidebar.checkbox("Enable Advanced Agent Customization")
-if enable_customization:
-    st.sidebar.markdown("### Customize Agent Goals")
-    planner_goal = st.sidebar.text_area(
-        "Planner Goal",
-        value="Plan engaging and factually accurate content about the topic."
-    )
-    writer_goal = st.sidebar.text_area(
-        "Writer Goal",
-        value="Write insightful and engaging content based on the topic."
-    )
-    analyst_goal = st.sidebar.text_area(
-        "Analyst Goal",
-        value="Perform statistical analysis to extract actionable insights."
-    )
-else:
-    planner_goal = "Plan engaging and factually accurate content about the topic."
-    writer_goal = "Write insightful and engaging content based on the topic."
-    analyst_goal = "Perform statistical analysis to extract actionable insights."
+st.sidebar.subheader("Agent Customization")
+
+# Display customization section in a collapsible expander
+with st.sidebar.expander("Customize Agent Goals", expanded=False):
+    enable_customization = st.checkbox("Enable Custom Goals")
+    if enable_customization:
+        planner_goal = st.text_area(
+            "Planner Goal",
+            value="Develop a comprehensive plan focusing on market trends and strategies."
+        )
+        writer_goal = st.text_area(
+            "Writer Goal",
+            value="Craft engaging and actionable content based on analysis."
+        )
+        analyst_goal = st.text_area(
+            "Analyst Goal",
+            value="Perform advanced statistical analysis and generate key insights."
+        )
+    else:
+        planner_goal = "Develop a comprehensive plan focusing on market trends and strategies."
+        writer_goal = "Craft engaging and actionable content based on analysis."
+        analyst_goal = "Perform advanced statistical analysis and generate key insights."
+
 
 #=================
-# LLM Object and API Key
+# LLM Object
 #=================
 llm = ChatGroq(groq_api_key=os.getenv("GROQ_API_KEY"), model="groq/llama-3.3-70b-versatile")
 
@@ -174,38 +157,54 @@ def generate_pdf_report(result):
     pdf.output(report_path)
     return report_path
 
-if st.button("Run Analysis"):
-    with st.spinner('Executing analysis...'):
+
+if st.button("Generate Insights"):
+    with st.spinner('Processing...'):
         try:
             start_time = time.time()
-            result = crew.kickoff(inputs={"topic": business, "stakeholder": stakeholder})
-            execution_time = time.time() - start_time
+            results = crew.kickoff(inputs={"topic": business, "stakeholder": stakeholder})
+            elapsed_time = time.time() - start_time
 
-            # Display Results
+            # Parse and Display Results
             st.markdown("### Insights and Analysis")
-            st.write(result)
+
+            # Handle Raw Output
+            raw_output = getattr(results, "raw", None)
+            if raw_output:
+                st.markdown("#### Executive Summary")
+                st.write(raw_output)
+            else:
+                st.warning("No executive summary available.")
+
+            # Handle Task Outputs
+            tasks_output = getattr(results, "tasks_output", [])
+            if tasks_output:
+                st.markdown("#### Task Outputs")
+                for idx, task in enumerate(tasks_output):
+                    task_description = getattr(task,"description", "No description available.")
+                    task_raw = getattr(task, "raw", "No details available.")
+                    st.subheader(f"Task {idx + 1}")
+                    st.write(f"**Description:** {task_description}")
+                    st.write(task_raw)
+            else:
+                st.warning("No task outputs available.")
+
+            # Display Token Usage
+            token_usage = getattr(results, "token_usage", None)
+            if token_usage:
+                st.markdown("#### Token Usage")
+                st.json(token_usage)
+            else:
+                st.warning("No token usage information available.")
 
             # Display Execution Time
-            st.success(f"Analysis completed in {execution_time:.2f} seconds!")
-
-            # Visualization Example
-            st.markdown("### Data Visualization Example")
-            data = pd.DataFrame({
-                "Metric": ["Trend 1", "Trend 2", "Trend 3"],
-                "Value": [45, 80, 65]
-            })
-            fig = px.bar(data, x="Metric", y="Value", title="Sample Metrics for Analysis")
-            st.plotly_chart(fig)
-
-            # Generate and Provide PDF Report
-            report_path = generate_pdf_report(result)
-            with open(report_path, "rb") as file:
-                st.download_button(
-                    label="Download Report as PDF",
-                    data=file,
-                    file_name="Business_Insights_Report.pdf",
-                    mime="application/pdf"
-                )
+            st.success(f"Analysis completed in {elapsed_time:.2f} seconds.")
+
+            # Generate PDF Report
+            if raw_output:
+                report_path = generate_pdf_report(raw_output)
+                with open(report_path, "rb") as report_file:
+                    st.download_button("Download Report", data=report_file, file_name="Business_Report.pdf")
 
         except Exception as e:
             st.error(f"An error occurred during execution: {e}")