Update app.py

app.py

@@ -38,9 +38,83 @@ st.markdown("""
             color: white;
             transition: width 0.5s ease;
         }
+
+        /* Agent Log Styles */
+        .agent-log {
+            background: #1a1a1a;
+            color: #fff;
+            padding: 20px;
+            border-radius: 10px;
+            font-family: 'Courier New', monospace;
+            height: 500px;
+            overflow-y: auto;
+        }
+        
+        .log-entry {
+            margin: 10px 0;
+            padding: 8px;
+            border-left: 3px solid;
+            animation: fadeIn 0.5s ease-in;
+        }
+        
+        .log-entry.researcher {
+            border-color: #4CAF50;
+            background: rgba(76, 175, 80, 0.1);
+        }
+        
+        .log-entry.analyst {
+            border-color: #2196F3;
+            background: rgba(33, 150, 243, 0.1);
+        }
+        
+        .log-entry.writer {
+            border-color: #FF9800;
+            background: rgba(255, 152, 0, 0.1);
+        }
+        
+        .timestamp {
+            color: #888;
+            font-size: 0.8em;
+        }
+        
+        @keyframes fadeIn {
+            from { opacity: 0; transform: translateY(10px); }
+            to { opacity: 1; transform: translateY(0); }
+        }
     </style>
 """, unsafe_allow_html=True)
 
+# Initialize session state
+if 'agent_logs' not in st.session_state:
+    st.session_state.agent_logs = []
+
+def log_agent_activity(agent_type: str, message: str):
+    """Log agent activity with timestamp"""
+    timestamp = time.strftime("%H:%M:%S")
+    st.session_state.agent_logs.append({
+        'timestamp': timestamp,
+        'agent': agent_type,
+        'message': message
+    })
+
+def display_agent_logs():
+    """Display all agent logs"""
+    if not st.session_state.agent_logs:
+        st.info("No agent activity logs yet. Generate a report to see agents in action!")
+        return
+
+    st.markdown('<div class="agent-log">', unsafe_allow_html=True)
+    
+    for log in st.session_state.agent_logs:
+        st.markdown(f"""
+            <div class="log-entry {log['agent'].lower()}">
+                <span class="timestamp">[{log['timestamp']}]</span>
+                <strong>{log['agent']}:</strong> {log['message']}
+            </div>
+        """, unsafe_allow_html=True)
+    
+    st.markdown('</div>', unsafe_allow_html=True)
+
 def update_progress(progress_container, percentage, message=""):
     """Update the progress bar."""
     progress_html = f"""
@@ -141,32 +215,56 @@ def create_research_crew(topic: str):
 def run_market_research(topic: str, progress_container, chat_container, log_container):
     """Run the market research process"""
     try:
+        # Clear previous logs when starting new research
+        st.session_state.agent_logs = []
+        
         # Step 1: Initialize research team
         update_progress(progress_container, 0, "Initializing research team...")
-        log_container.write("Initializing research team...")
+        log_agent_activity("System", f"Starting market research for: {topic}")
         time.sleep(1)
         
         # Step 2: Gathering market data
         update_progress(progress_container, 25, "Gathering market data...")
-        log_container.write("Research Analyst is gathering market data...")
+        log_agent_activity("Research Analyst", f"Beginning comprehensive research on {topic}")
+        display_agent_message(chat_container, "researcher", 
+            f"Starting comprehensive research on {topic}...")
+        log_agent_activity("Research Analyst", "Collecting market size data...")
+        time.sleep(1)
+        log_agent_activity("Research Analyst", "Analyzing competitive landscape...")
+        time.sleep(1)
+        log_agent_activity("Research Analyst", "Gathering growth projections...")
         time.sleep(1)
         
         # Step 3: Analyzing findings
         update_progress(progress_container, 50, "Analyzing findings...")
-        log_container.write("Data Analyst is analyzing findings...")
+        log_agent_activity("Data Analyst", "Beginning data analysis phase")
+        display_agent_message(chat_container, "analyst", 
+            "Processing research data and identifying key insights...")
+        log_agent_activity("Data Analyst", "Processing market size metrics...")
+        time.sleep(1)
+        log_agent_activity("Data Analyst", "Calculating growth rates...")
+        time.sleep(1)
+        log_agent_activity("Data Analyst", "Generating market share analysis...")
         time.sleep(1)
         
         # Step 4: Compiling final report
         update_progress(progress_container, 75, "Compiling final report...")
-        log_container.write("Report Writer is compiling the final report...")
-        
+        log_agent_activity("Report Writer", "Starting report compilation")
+        display_agent_message(chat_container, "writer", 
+            "Compiling final report with executive summary and detailed analysis...")
+        log_agent_activity("Report Writer", "Creating executive summary...")
+        time.sleep(1)
+        log_agent_activity("Report Writer", "Drafting detailed analysis...")
+        time.sleep(1)
+        log_agent_activity("Report Writer", "Adding citations and sources...")
+        time.sleep(1)
+
         # Create and run the crew
         crew = create_research_crew(topic)
         result = crew.kickoff()
         
-        # Ensure result is a dictionary
-        if isinstance(result, list):
-            result = result[0]
+        # Log completion
+        log_agent_activity("System", "Report generation completed successfully")
         
         # Format and process the result
         report_data = format_json_output(result)
@@ -177,23 +275,27 @@ def run_market_research(topic: str, progress_container, chat_container, log_cont
         
         update_progress(progress_container, 100, "Report completed!")
         st.success("✨ Report generation completed! You can now view the full report.")
-        log_container.write("Report generation completed!")
+        log_agent_activity("System", "Report generation completed!")
         
         # Store the report data in session state
         st.session_state.current_report = report_data
         st.session_state.current_topic = topic
         
     except Exception as e:
-        st.error(f"Error during research: {str(e)}")
-        log_container.write(f"Error during research: {str(e)}")
+        error_msg = f"Error during research: {str(e)}"
+        log_agent_activity("System", f"Error: {error_msg}")
+        st.error(error_msg)
 
 def main():
     st.title("🤖 AI Market Research Generator")
     
-    # Initialize session state for button control
+    # Initialize session states
     if 'generating' not in st.session_state:
         st.session_state.generating = False
+    if 'agent_logs' not in st.session_state:
+        st.session_state.agent_logs = []
     
+    # Create tabs including the new Live Agent Log tab
     tab1, tab2, tab3 = st.tabs(["Generate Report", "View Reports", "Live Agent Log"])
     
     with tab1:
@@ -287,9 +389,30 @@ def main():
                     st.markdown(f"- {source}")
     
     with tab3:
-        st.subheader("Live Agent Log")
-        if 'log_container' in st.session_state:
-            st.session_state['log_container'].write("Live log will appear here...")
+        st.subheader("📋 Agent Activity Log")
+        st.markdown("""
+            This log shows the detailed activities of our AI agents as they work on your market research.
+            Each agent's actions are tracked and timestamped for transparency.
+        """)
+        
+        # Add log controls
+        col1, col2 = st.columns([6, 1])
+        with col1:
+            if st.session_state.agent_logs:
+                st.download_button(
+                    "Download Log",
+                    data="\n".join([f"[{log['timestamp']}] {log['agent']}: {log['message']}" 
+                                  for log in st.session_state.agent_logs]),
+                    file_name="agent_activity_log.txt",
+                    mime="text/plain"
+                )
+        with col2:
+            if st.button("Clear Log"):
+                st.session_state.agent_logs = []
+                st.rerun()
+        
+        # Display the logs
+        display_agent_logs()
 
 def format_json_output(raw_output):
     """Format CrewOutput or raw string into proper JSON structure"""
@@ -322,79 +445,4 @@ def format_json_output(raw_output):
             "metrics": {
                 "market_size_data": [],
                 "growth_rates": [],
-                "market_shares": []
-            }
-        }
-    except Exception as e:
-        st.error(f"Error formatting output: {str(e)}")
-        return {
-            "exec_summary": {
-                "summary": "Error formatting report",
-                "market_size": "N/A",
-                "growth_rate": "N/A",
-                "key_players": "N/A"
-            },
-            "detailed_report": raw_text if 'raw_text' in locals() else str(raw_output),
-            "sources": [],
-            "metrics": {
-                "market_size_data": [],
-                "growth_rates": [],
-                "market_shares": []
-            }
-        }
-
-def extract_section(text, section_name):
-    """Extract a section from the text"""
-    pattern = f"{section_name}.*?\n(.*?)(?=\n\n|$)"
-    match = re.search(pattern, text, re.DOTALL | re.IGNORECASE)
-    return match.group(1).strip() if match else ""
-
-def extract_sources(text):
-    """Extract sources from the text"""
-    sources = []
-    source_pattern = r"Source:.*?(?:\n|$)|\[.*?\]|\(https?://.*?\)"
-    matches = re.finditer(source_pattern, text, re.MULTILINE)
-    return [match.group().strip() for match in matches]
-
-def generate_visualizations(data):
-    """Generate visualizations based on the report data"""
-    charts = []
-    
-    # Market Size Chart
-    if data['metrics']['market_size_data']:
-        fig = go.Figure(data=[
-            go.Bar(
-                x=['Current', 'Projected'],
-                y=data['metrics']['market_size_data'],
-                text=data['metrics']['market_size_data'],
-                textposition='auto',
-            )
-        ])
-        fig.update_layout(title='Market Size Projection')
-        charts.append(fig)
-    
-    # Growth Rate Chart
-    if data['metrics']['growth_rates']:
-        fig = go.Figure(data=[
-            go.Scatter(
-                x=list(range(len(data['metrics']['growth_rates']))),
-                y=data['metrics']['growth_rates'],
-                mode='lines+markers'
-            )
-        ])
-        fig.update_layout(title='Growth Rate Trends')
-        charts.append(fig)
-    
-    # Market Share Chart
-    if data['metrics']['market_shares']:
-        fig = go.Figure(data=[go.Pie(
-            labels=list(data['metrics']['market_shares'].keys()),
-            values=list(data['metrics']['market_shares'].values())
-        )])
-        fig.update_layout(title='Market Share Distribution')
-        charts.append(fig)
-    
-    return charts
-
-if __name__ == "__main__":
-    main()
+                "market_shares":