Create config/agent_config.py

config/agent_config.py

@@ -0,0 +1,83 @@
+# config/agent_config.py
+
+AGENT_PERSONAS = {
+    'Collection Agent': {
+        'emoji': '🔍',
+        'activities': [
+            'Scanning social platforms...',
+            'Collecting trending posts...',
+            'Processing engagement metrics...',
+            'Analyzing hashtag patterns...',
+            'Gathering location data...',
+            'Validating trend sources...'
+        ]
+    },
+    'Research Agent': {
+        'emoji': '📊',
+        'activities': [
+            'Analyzing trend context...',
+            'Evaluating engagement patterns...',
+            'Processing geographic data...',
+            'Calculating growth metrics...',
+            'Generating predictions...'
+        ]
+    },
+    'Analysis Agent': {
+        'emoji': '🧪',
+        'activities': [
+            'Calculating trend scores...',
+            'Mapping geographic spread...',
+            'Generating visualizations...',
+            'Processing demographics...',
+            'Creating forecasts...'
+        ]
+    }
+}
+
+def get_agent_status(agent_name: str) -> str:
+    """Get current status for an agent"""
+    if agent_name not in AGENT_PERSONAS:
+        return "Working..."
+    
+    from datetime import datetime
+    import streamlit as st
+    
+    activities = AGENT_PERSONAS[agent_name]['activities']
+    current_time = datetime.now()
+    start_time = st.session_state.get('start_time', current_time)
+    index = int((current_time - start_time).total_seconds() / 2) % len(activities)
+    
+    return activities[index]
+
+def display_agent_progress():
+    """Display agent progress with visual enhancements"""
+    import streamlit as st
+    
+    st.markdown("### 🤖 Analysis Team Progress")
+    
+    for agent_name, persona in AGENT_PERSONAS.items():
+        emoji = persona['emoji']
+        
+        if agent_name in st.session_state.completed_agents:
+            status = "✅ Completed"
+            color = "green"
+        elif agent_name == st.session_state.current_agent:
+            status = get_agent_status(agent_name)
+            color = "blue"
+        else:
+            status = "Waiting..."
+            color = "gray"
+        
+        st.markdown(f"""
+        <div style='padding: 10px; margin: 5px; border-radius: 5px; border: 1px solid {color}'>
+            <div style='font-weight: bold; color: {color}'>{emoji} {agent_name}</div>
+            <div style='color: {color}; font-size: 0.9em'>{status}</div>
+        </div>
+        """, unsafe_allow_html=True)
+    
+    completed = len(st.session_state.completed_agents)
+    total = len(AGENT_PERSONAS)
+    progress = completed / total
+    
+    st.progress(progress)
+    st.markdown(f"**Overall Progress:** {progress * 100:.0f}%")