Update app.py

app.py

@@ -1,17 +1,11 @@
 # app.py
 import os
 import time
-import json
-import random
+import re
 import streamlit as st
 from dotenv import load_dotenv
 from crewai import Agent, Crew, Process, Task
 
-# Add plotly for interactive charts
-import plotly.graph_objects as go
-import plotly.express as px
-import pandas as pd
-
 # Load environment variables
 load_dotenv()
 
@@ -75,12 +69,34 @@ st.markdown("""
             100% { transform: translateX(100%); opacity: 0; }
         }
         
-        /* Agent message styles */
-        .agent-message {
-            padding: 15px;
+        /* Progress animation */
+        @keyframes progress {
+            0% { width: 0; }
+            100% { width: 100%; }
+        }
+        
+        .progress-bar {
+            height: 20px;
+            background-color: #f0f0f0;
             border-radius: 10px;
+            overflow: hidden;
             margin: 10px 0;
-            animation: slideIn 0.5s ease-out;
+        }
+        
+        .progress-fill {
+            height: 100%;
+            background-color: #4CAF50;
+            animation: progress 2s ease-in-out;
+        }
+        
+        /* Agent status indicators */
+        .agent-status {
+            padding: 10px;
+            margin: 5px 0;
+            border-radius: 5px;
+            background: #f8f9fa;
+            border-left: 4px solid #4CAF50;
+            animation: slideIn 0.3s ease-out;
         }
         
         @keyframes slideIn {
@@ -105,6 +121,15 @@ st.markdown("""
         .stTabs [aria-selected="true"] {
             background-color: #e6f3ff;
         }
+        
+        /* Report container */
+        .report-section {
+            background: white;
+            padding: 20px;
+            border-radius: 10px;
+            margin: 10px 0;
+            box-shadow: 0 2px 4px rgba(0,0,0,0.1);
+        }
     </style>
 """, unsafe_allow_html=True)
 
@@ -124,43 +149,6 @@ def create_agent_flow(container):
         </div>
     """, unsafe_allow_html=True)
 
-def generate_sample_charts(topic, analysis_text):
-    """Generate relevant charts based on the analysis"""
-    charts = []
-    
-    # Extract numerical data using simple patterns
-    # This is a simplified example - in practice, you'd want more sophisticated parsing
-    growth_pattern = r"(\d+(?:\.\d+)?)\s*%"
-    values = [float(x) for x in re.findall(growth_pattern, analysis_text)]
-    
-    if values:
-        # Market Growth Chart
-        fig_growth = go.Figure(data=[
-            go.Scatter(
-                x=list(range(2024, 2024 + len(values))),
-                y=values,
-                mode='lines+markers',
-                name='Projected Growth'
-            )
-        ])
-        fig_growth.update_layout(
-            title=f"{topic} Market Growth Projection",
-            xaxis_title="Year",
-            yaxis_title="Growth Rate (%)"
-        )
-        charts.append(fig_growth)
-        
-        # Market Share Donut Chart
-        fig_share = go.Figure(data=[go.Pie(
-            labels=['Major Players', 'Other Companies'],
-            values=[65, 35],
-            hole=.3
-        )])
-        fig_share.update_layout(title='Market Share Distribution')
-        charts.append(fig_share)
-    
-    return charts
-
 def run_market_research(topic: str, progress_bar, chat_container):
     try:
         # Create agents
@@ -185,41 +173,73 @@ def run_market_research(topic: str, progress_bar, chat_container):
             verbose=True
         )
 
-        # Update progress and display
+        # Update progress
         progress_bar.progress(0, "Initializing research...")
+        chat_container.markdown("""
+            <div class="agent-status">
+                🔍 Research Analyst: Starting market analysis...
+            </div>
+        """, unsafe_allow_html=True)
         time.sleep(1)
         
         # Research Phase
         research_task = Task(
-            description=f"Conduct comprehensive market research on {topic}...",
+            description=f"""
+            Conduct comprehensive market research on {topic}.
+            Focus on:
+            1. Current market size and growth projections
+            2. Key players and their market share
+            3. Consumer adoption trends
+            4. Regulatory environment
+            """,
             agent=researcher,
             expected_output="A comprehensive research document with market analysis."
         )
         
-        progress_bar.progress(25, "Gathering market data...")
+        progress_bar.progress(33, "Analyzing data...")
+        chat_container.markdown("""
+            <div class="agent-status">
+                📊 Data Analyst: Processing research findings...
+            </div>
+        """, unsafe_allow_html=True)
         time.sleep(1)
         
         # Analysis Phase
         analysis_task = Task(
-            description=f"Analyze the research findings for {topic}...",
+            description=f"""
+            Analyze the research findings for {topic} and identify:
+            1. Key market opportunities
+            2. Potential challenges
+            3. Growth drivers
+            4. Competitive dynamics
+            """,
             agent=analyst,
             expected_output="An analytical report with key insights.",
             context=[research_task]
         )
         
-        progress_bar.progress(50, "Analyzing findings...")
+        progress_bar.progress(66, "Generating report...")
+        chat_container.markdown("""
+            <div class="agent-status">
+                ✍️ Report Writer: Compiling final report...
+            </div>
+        """, unsafe_allow_html=True)
         time.sleep(1)
         
         # Report Phase
         report_task = Task(
-            description=f"Create a detailed market research report...",
+            description=f"""
+            Create a detailed market research report that includes:
+            1. Executive summary
+            2. Market overview
+            3. Key findings
+            4. Strategic recommendations
+            """,
             agent=writer,
             expected_output="A complete market research report in markdown format.",
             context=[research_task, analysis_task]
         )
         
-        progress_bar.progress(75, "Generating report...")
-        
         # Create and run the crew
         crew = Crew(
             agents=[researcher, analyst, writer],
@@ -237,14 +257,11 @@ def run_market_research(topic: str, progress_bar, chat_container):
         else:
             final_report = str(result)
             
-        # Generate charts based on the analysis
-        charts = generate_sample_charts(topic, final_report)
-        
-        return final_report, charts
+        return final_report
 
     except Exception as e:
         st.error(f"Error: {str(e)}")
-        return None, []
+        return None
 
 def main():
     st.title("🤖 AI Market Research Generator")
@@ -276,12 +293,11 @@ def main():
                 create_agent_flow(col2)
                 
                 # Run the research
-                report, charts = run_market_research(topic, progress, chat_container)
+                report = run_market_research(topic, progress, chat_container)
                 
                 if report:
                     # Store results in session state
                     st.session_state.report = report
-                    st.session_state.charts = charts
                     st.session_state.current_tab = "report"
                     
                     # Switch to report tab
@@ -291,14 +307,15 @@ def main():
         if hasattr(st.session_state, 'report'):
             st.subheader("📊 Market Research Report")
             
-            # Display report sections
-            st.markdown(st.session_state.report)
-            
-            # Display charts
-            if hasattr(st.session_state, 'charts'):
-                st.subheader("📈 Data Visualization")
-                for chart in st.session_state.charts:
-                    st.plotly_chart(chart, use_container_width=True)
+            # Display report in sections
+            sections = st.session_state.report.split('\n\n')
+            for section in sections:
+                if section.strip():
+                    st.markdown(f"""
+                        <div class="report-section">
+                            {section}
+                        </div>
+                    """, unsafe_allow_html=True)
             
             # Download button
             st.download_button(