Update app.py

app.py

@@ -14,6 +14,7 @@ class RealTimeMarketData:
         self.symbols = symbols
         self.last_prices = {}
         self.data_history = {symbol: [] for symbol in symbols}
+        self.confidence_history = {symbol: [] for symbol in symbols}
         self.timestamps = []
         self.update_counter = 0
         
@@ -21,6 +22,7 @@ class RealTimeMarketData:
         for symbol in symbols:
             self.last_prices[symbol] = np.random.uniform(150, 250)
             self.data_history[symbol] = [self.last_prices[symbol]]
+            self.confidence_history[symbol] = []
         
         self.timestamps = [datetime.now().strftime('%H:%M:%S')]
         
@@ -31,7 +33,7 @@ class RealTimeMarketData:
         
         # Add new timestamp
         self.timestamps.append(current_time.strftime('%H:%M:%S'))
-        if len(self.timestamps) > 15:
+        if len(self.timestamps) > 20:
             self.timestamps.pop(0)
         
         live_data = {}
@@ -44,7 +46,7 @@ class RealTimeMarketData:
             
             # Add to history
             self.data_history[symbol].append(new_price)
-            if len(self.data_history[symbol]) > 15:
+            if len(self.data_history[symbol]) > 20:
                 self.data_history[symbol].pop(0)
             
             # Calculate metrics
@@ -91,7 +93,6 @@ class AI_TradingAgents:
 • Expanding profit margins
 • Market leadership position
 • Positive institutional sentiment
-• Strong balance sheet
 
 **Recommendation: BUY** (85% confidence)
 **Price Target: ${current_price * 1.15:.2f}**"""
@@ -103,7 +104,6 @@ class AI_TradingAgents:
 • Increasing competitive pressures
 • Wait for better entry point
 • Support at ${current_price * 0.95:.2f}
-• Monitor earnings closely
 
 **Recommendation: HOLD** (70% confidence)"""
             research_conf = 70
@@ -142,8 +142,7 @@ class AI_TradingAgents:
   - Support: ${support:.2f}
   - Resistance: ${resistance:.2f}
 • RSI: {65 if change > 0 else 35} ({rsi_level})
-• Volume: {data['volume']:,}
-• Momentum: {'Positive' if change > 0 else 'Negative'}""",
+• Volume: {data['volume']:,}""",
             'confidence': 75
         }
         
@@ -152,17 +151,14 @@ class AI_TradingAgents:
             risk_level = "HIGH"
             position_size = "1-2%"
             stop_loss = "10%"
-            risk_reward = "1:2"
         elif volatility > 1.5:
             risk_level = "MEDIUM"
             position_size = "2-3%"
             stop_loss = "8%"
-            risk_reward = "1:2.5"
         else:
             risk_level = "LOW"
             position_size = "3-4%"
             stop_loss = "6%"
-            risk_reward = "1:3"
         
         analyses['risk'] = {
             'emoji': '🛡️',
@@ -170,11 +166,9 @@ class AI_TradingAgents:
             'analysis': f"""**{risk_level} RISK PROFILE**
 
 • Volatility: {volatility:.1f}%
-• Recommended Position: {position_size} of portfolio
-• Stop-Loss: {stop_loss} below entry
-• Risk-Reward Ratio: {risk_reward}
-• Maximum Drawdown: 12%
-• Correlation: Low with portfolio
+• Position Size: {position_size}
+• Stop-Loss: {stop_loss}
+• Risk-Reward: 1:{3 if risk_level == 'LOW' else 2}
 • Monitoring: {'Intensive' if risk_level == 'HIGH' else 'Standard'}""",
             'confidence': 80
         }
@@ -183,38 +177,23 @@ class AI_TradingAgents:
         if change > 2 and volatility < 4:
             decision = "BUY"
             confidence = 85
-            reason = "Strong bullish momentum with favorable risk metrics and positive fundamentals"
-            action = "Enter long position with trailing stop at 8%. Target 15-20% upside."
         elif change < -2:
             decision = "SELL"
             confidence = 78
-            reason = "Significant downward pressure with deteriorating technicals"
-            action = "Consider short opportunities or wait for stabilization. Set tight stop-loss."
         else:
             decision = "HOLD"
             confidence = 72
-            reason = "Consolidation phase with mixed signals. Awaiting clearer market direction."
-            action = "Monitor for breakout above resistance or breakdown below support."
         
         analyses['decision'] = {
             'emoji': '🎯',
             'title': 'Trading Decision',
             'analysis': f"""**FINAL DECISION: {decision}** 🎯
 
-**Confidence Level:** {confidence}%
-**Current Price:** ${current_price:.2f}
-**Price Change:** {change:+.2f}%
+**Confidence:** {confidence}%
+**Price:** ${current_price:.2f}
+**Change:** {change:+.2f}%
 
-**Rationale:**
-{reason}
-
-**Execution Plan:**
-{action}
-
-**Key Factors:**
-• Fundamental Score: {analyses['research']['confidence']}%
-• Technical Score: {analyses['technical']['confidence']}%
-• Risk Assessment: {analyses['risk']['confidence']}%""",
+**Action:** {'Enter long position' if decision == 'BUY' else 'Wait for setup'}""",
             'confidence': confidence,
             'decision': decision
         }
@@ -222,7 +201,7 @@ class AI_TradingAgents:
         return analyses
     
     def _get_error_analysis(self, symbol):
-        error_msg = "Data temporarily unavailable. Please check symbol spelling or try again later."
+        error_msg = "Data temporarily unavailable."
         return {
             'research': {'emoji': '📊', 'title': 'Research', 'analysis': error_msg, 'confidence': 0},
             'technical': {'emoji': '📈', 'title': 'Technical', 'analysis': error_msg, 'confidence': 0},
@@ -234,146 +213,268 @@ class AI_TradingAgents:
 market_data = RealTimeMarketData()
 trading_agents = AI_TradingAgents()
 
+def create_live_price_chart():
+    """Create live price movement chart"""
+    live_data = market_data.generate_live_data()
+    
+    fig = go.Figure()
+    
+    colors = ['#FF6B6B', '#4ECDC4', '#45B7D1', '#96CEB4']
+    for i, (symbol, data) in enumerate(live_data.items()):
+        fig.add_trace(go.Scatter(
+            x=data['timestamps'],
+            y=data['prices'],
+            mode='lines+markers',
+            name=symbol,
+            line=dict(color=colors[i], width=3),
+            marker=dict(size=6),
+            hovertemplate=f'<b>{symbol}</b><br>%{{x}}<br>$%{{y:.2f}}<extra></extra>'
+        ))
+    
+    fig.update_layout(
+        title=f"📈 Live Price Movement - Update #{market_data.update_counter}",
+        template='plotly_dark',
+        height=400,
+        xaxis_title="Time",
+        yaxis_title="Price ($)",
+        showlegend=True
+    )
+    
+    return fig
+
+def create_confidence_chart(symbol_input=""):
+    """Create AI agents confidence comparison chart"""
+    live_data = market_data.generate_live_data()
+    
+    if not symbol_input:
+        symbol = 'AAPL'
+    else:
+        symbol = symbol_input.upper()
+    
+    if symbol not in live_data:
+        symbol = 'AAPL'
+    
+    analysis = trading_agents.analyze_market(symbol, live_data)
+    
+    agents = list(analysis.keys())
+    confidences = [analysis[agent]['confidence'] for agent in agents]
+    colors = ['#FF6B6B', '#4ECDC4', '#45B7D1', '#96CEB4']
+    
+    fig = go.Figure()
+    
+    fig.add_trace(go.Bar(
+        x=agents,
+        y=confidences,
+        marker_color=colors,
+        text=[f"{c}%" for c in confidences],
+        textposition='auto',
+        hovertemplate='<b>%{x}</b><br>Confidence: %{y}%<extra></extra>'
+    ))
+    
+    fig.update_layout(
+        title=f"🤖 AI Agents Confidence - {symbol}",
+        template='plotly_dark',
+        height=400,
+        xaxis_title="AI Agents",
+        yaxis_title="Confidence Level (%)",
+        yaxis_range=[0, 100]
+    )
+    
+    return fig
+
+def create_market_sentiment_chart():
+    """Create market sentiment pie chart"""
+    live_data = market_data.generate_live_data()
+    
+    decisions = {'BUY': 0, 'SELL': 0, 'HOLD': 0}
+    for symbol in live_data.keys():
+        analysis = trading_agents.analyze_market(symbol, live_data)
+        decision = analysis['decision']['decision']
+        decisions[decision] += 1
+    
+    fig = go.Figure()
+    
+    fig.add_trace(go.Pie(
+        labels=list(decisions.keys()),
+        values=list(decisions.values()),
+        hole=0.4,
+        marker_colors=['#00CC96', '#EF553B', '#636EFA'],
+        hovertemplate='<b>%{label}</b><br>%{value} stocks<extra></extra>'
+    ))
+    
+    fig.update_layout(
+        title="🎯 Market Sentiment - AI Recommendations",
+        template='plotly_dark',
+        height=400
+    )
+    
+    return fig
+
+def create_performance_chart():
+    """Create stock performance comparison chart"""
+    live_data = market_data.generate_live_data()
+    
+    symbols = list(live_data.keys())
+    changes = [live_data[symbol]['change'] for symbol in symbols]
+    prices = [live_data[symbol]['current_price'] for symbol in symbols]
+    
+    fig = make_subplots(
+        rows=1, cols=2,
+        subplot_titles=['24h Performance (%)', 'Current Prices ($)'],
+        specs=[[{"type": "bar"}, {"type": "bar"}]]
+    )
+    
+    # Performance bars
+    fig.add_trace(go.Bar(
+        x=symbols,
+        y=changes,
+        name='24h Change',
+        marker_color=['#00CC96' if c > 0 else '#EF553B' for c in changes],
+        text=[f"{c:+.2f}%" for c in changes],
+        textposition='auto'
+    ), row=1, col=1)
+    
+    # Price bars
+    fig.add_trace(go.Bar(
+        x=symbols,
+        y=prices,
+        name='Current Price',
+        marker_color='#636EFA',
+        text=[f"${p:.2f}" for p in prices],
+        textposition='auto'
+    ), row=1, col=2)
+    
+    fig.update_layout(
+        title="📊 Market Performance Overview",
+        template='plotly_dark',
+        height=400,
+        showlegend=False
+    )
+    
+    return fig
+
 def get_ai_agents_analysis(symbol_input=""):
     """Get detailed analysis from all AI agents"""
     live_data = market_data.generate_live_data()
     
     if not symbol_input:
-        return "# 🤖 AI Agents Analysis\n\n**Please enter a stock symbol to see detailed analysis from all AI agents**"
+        return "# 🤖 AI Agents Analysis\n\n**Please enter a stock symbol to see detailed analysis**"
     
     symbol = symbol_input.upper()
     if symbol not in live_data:
-        return f"# ❌ Symbol Not Found\n\n'{symbol}' is not in our tracked symbols. Try: {', '.join(market_data.symbols)}"
+        return f"# ❌ Symbol Not Found\n\n'{symbol}' not in tracked symbols."
     
-    # Get analysis from all agents
     analysis = trading_agents.analyze_market(symbol, live_data)
     
     detailed_report = f"""
-# 🤖 AI Agents Specialized Analysis
+# 🤖 AI Agents Analysis - {symbol}
 
-## 📈 {symbol} - Real-time Analysis
 **Current Price:** ${live_data[symbol]['current_price']:.2f}  
 **24h Change:** {live_data[symbol]['change']:+.2f}%  
-**Last Update:** {datetime.now().strftime('%H:%M:%S')}  
-**Analysis Cycle:** #{market_data.update_counter}
+**Last Update:** {datetime.now().strftime('%H:%M:%S')}
 
 ---
 
 ## {analysis['research']['emoji']} {analysis['research']['title']}
-**Confidence Level:** {analysis['research']['confidence']}%
+**Confidence:** {analysis['research']['confidence']}%
 
 {analysis['research']['analysis']}
 
 ---
 
 ## {analysis['technical']['emoji']} {analysis['technical']['title']}
-**Confidence Level:** {analysis['technical']['confidence']}%
+**Confidence:** {analysis['technical']['confidence']}%
 
 {analysis['technical']['analysis']}
 
 ---
 
 ## {analysis['risk']['emoji']} {analysis['risk']['title']}
-**Confidence Level:** {analysis['risk']['confidence']}%
+**Confidence:** {analysis['risk']['confidence']}%
 
 {analysis['risk']['analysis']}
 
 ---
 
 ## {analysis['decision']['emoji']} {analysis['decision']['title']}
-**Confidence Level:** {analysis['decision']['confidence']}%
+**Confidence:** {analysis['decision']['confidence']}%
 
 {analysis['decision']['analysis']}
-
----
-
-### 🎯 Multi-Agent Consensus
-- **Overall Confidence:** {np.mean([a['confidence'] for a in analysis.values()]):.1f}%
-- **Final Recommendation:** {analysis['decision']['decision']}
-- **Risk Level:** {'High' if analysis['risk']['confidence'] < 70 else 'Medium' if analysis['risk']['confidence'] < 80 else 'Low'}
-- **Market Outlook:** {'Bullish' if analysis['research']['confidence'] > 80 else 'Neutral' if analysis['research']['confidence'] > 65 else 'Bearish'}
-
-*Analysis generated by Multi-Agent AI Trading System*
 """
     return detailed_report
 
-def get_agents_performance():
-    """Get performance overview of all agents"""
-    live_data = market_data.generate_live_data()
-    
-    performance_data = []
-    for symbol in market_data.symbols:
-        analysis = trading_agents.analyze_market(symbol, live_data)
-        performance_data.append({
-            'Symbol': symbol,
-            'Price': f"${live_data[symbol]['current_price']:.2f}",
-            'Change': f"{live_data[symbol]['change']:+.2f}%",
-            'Research': f"{analysis['research']['confidence']}%",
-            'Technical': f"{analysis['technical']['confidence']}%",
-            'Risk': f"{analysis['risk']['confidence']}%",
-            'Decision': analysis['decision']['decision'],
-            'Confidence': f"{analysis['decision']['confidence']}%"
-        })
-    
-    df = pd.DataFrame(performance_data)
-    return df
-
 # Create the interface
-with gr.Blocks(theme=gr.themes.Soft(), title="AI Agents Analysis Dashboard") as demo:
+with gr.Blocks(theme=gr.themes.Soft(), title="AI Trading Dashboard") as demo:
     
     gr.Markdown("""
-    # 🤖 AI Agents Specialized Analysis
-    ## *Multi-Agent AI Trading System - Professional Analysis*
+    # 🤖 AI Trading Dashboard
+    ## *Live Charts & Real-time Analysis*
     
-    **Real-time analysis from 4 specialized AI agents working in coordination**
+    **Interactive charts with live data updates every 5 seconds**
     """)
     
     with gr.Row():
         with gr.Column(scale=1):
             symbol_input = gr.Textbox(
-                label="Enter Stock Symbol",
-                placeholder="e.g., AAPL, TSLA, GOOGL...",
+                label="Stock Symbol",
+                placeholder="AAPL, TSLA...",
                 max_lines=1
             )
             gr.Markdown("""
-            **Available Symbols:** AAPL, GOOGL, MSFT, TSLA
-            
-            **🤖 AI Agents:**
-            - 📊 Research Agent: Fundamental Analysis
-            - 📈 Technical Agent: Price & Patterns  
-            - 🛡️ Risk Agent: Risk Management
-            - 🎯 Decision Engine: Final Recommendation
+            **Tracked Stocks:** AAPL, GOOGL, MSFT, TSLA
+            **Auto-Refresh:** Every 5 seconds
             """)
-        
-        with gr.Column(scale=2):
-            gr.Markdown("### 📋 Agents Performance Overview")
-            agents_performance = gr.DataFrame(
-                label="AI Agents Confidence Levels",
-                every=5,
-                value=get_agents_performance
-            )
     
-    with gr.Row():
-        # AI Agents Detailed Analysis
-        agents_analysis = gr.Markdown(
-            label="Specialized AI Agents Analysis",
-            every=5,
-            value=get_ai_agents_analysis
-        )
+    with gr.Tabs():
+        with gr.TabItem("📈 Live Charts"):
+            with gr.Row():
+                price_chart = gr.Plot(
+                    label="Live Price Movement",
+                    every=5,
+                    value=create_live_price_chart
+                )
+            
+            with gr.Row():
+                confidence_chart = gr.Plot(
+                    label="AI Agents Confidence",
+                    every=5,
+                    value=lambda: create_confidence_chart(symbol_input.value if symbol_input.value else "")
+                )
+            
+            with gr.Row():
+                sentiment_chart = gr.Plot(
+                    label="Market Sentiment",
+                    every=5,
+                    value=create_market_sentiment_chart
+                )
+                
+                performance_chart = gr.Plot(
+                    label="Performance Overview",
+                    every=5,
+                    value=create_performance_chart
+                )
+        
+        with gr.TabItem("🤖 AI Analysis"):
+            with gr.Row():
+                agents_analysis = gr.Markdown(
+                    label="AI Agents Analysis",
+                    every=5,
+                    value=get_ai_agents_analysis
+                )
     
     gr.Markdown(f"""
     ---
-    **🔄 Live AI Analysis Active** • **Last Analysis:** {datetime.now().strftime('%H:%M:%S')}
-    *Each agent provides specialized analysis updated every 5 seconds*
-    
-    **📊 Analysis Includes:**
-    - Fundamental company research and growth prospects
-    - Technical price patterns and market structure  
-    - Comprehensive risk assessment and position sizing
-    - Final trading decision with execution plan
+    **🔄 Live Dashboard Active** • **Last Update:** {datetime.now().strftime('%H:%M:%S')}
+    *All charts and analysis update automatically every 5 seconds*
     """)
     
-    # Update analysis when symbol changes
+    # Update confidence chart when symbol changes
+    symbol_input.change(
+        fn=lambda s: create_confidence_chart(s),
+        inputs=[symbol_input],
+        outputs=[confidence_chart]
+    )
+    
     symbol_input.change(
         fn=get_ai_agents_analysis,
         inputs=[symbol_input],