Update app.py

app.py

@@ -5,7 +5,6 @@ import yfinance as yf
 import plotly.graph_objects as go
 from plotly.subplots import make_subplots
 from datetime import datetime, timedelta
-import threading
 import time
 import warnings
 warnings.filterwarnings('ignore')
@@ -109,13 +108,14 @@ class RealTimeMarketData:
     
     def get_single_stock_data(self, symbol):
         """Get data for a single stock with real-time updates"""
-        if symbol.upper() not in self.symbols and symbol:
-            # Add new symbol temporarily
-            self.symbols.append(symbol.upper())
+        symbol = symbol.upper().strip()
+        if symbol not in self.symbols and symbol:
+            # Add new symbol temporarily for this session
+            self.symbols.append(symbol)
             self._update_all_data()
         
         data = self.get_live_data()
-        return data.get(symbol.upper(), self._generate_simulated_data(symbol))
+        return data.get(symbol, self._generate_simulated_data(symbol))
 
 class AI_TradingAgents:
     """
@@ -468,11 +468,12 @@ def generate_agent_analysis(symbol, market_data_dict):
     
     else:
         # Analyze specific symbol
-        analysis = trading_agents.analyze_market(symbol.upper(), market_data_dict)
+        symbol = symbol.upper().strip()
+        analysis = trading_agents.analyze_market(symbol, market_data_dict)
         
-        analysis_text = f"# 🎯 {symbol.upper()} - AI Trading Analysis\n\n"
-        analysis_text += f"**Current Price:** ${market_data_dict.get(symbol.upper(), {}).get('current_price', 'N/A'):.2f}\n"
-        analysis_text += f"**24h Change:** {market_data_dict.get(symbol.upper(), {}).get('change', 'N/A'):+.2f}%\n"
+        analysis_text = f"# 🎯 {symbol} - AI Trading Analysis\n\n"
+        analysis_text += f"**Current Price:** ${market_data_dict.get(symbol, {}).get('current_price', 'N/A'):.2f}\n"
+        analysis_text += f"**24h Change:** {market_data_dict.get(symbol, {}).get('change', 'N/A'):+.2f}%\n"
         analysis_text += f"**Last Updated:** {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n\n"
         
         for agent_type, agent_analysis in analysis.items():
@@ -481,18 +482,38 @@ def generate_agent_analysis(symbol, market_data_dict):
         
         return analysis_text
 
-def update_interface(symbol_input):
+def update_interface(symbol_input=""):
     """Main function to update the entire interface"""
-    # Get fresh market data
-    current_market_data = market_data.get_live_data()
-    
-    # Create visualizations
-    dashboard = create_real_time_dashboard(current_market_data, symbol_input)
-    
-    # Generate analysis
-    analysis = generate_agent_analysis(symbol_input, current_market_data)
-    
-    return dashboard, analysis
+    try:
+        # Get fresh market data
+        current_market_data = market_data.get_live_data()
+        
+        # Create visualizations
+        dashboard = create_real_time_dashboard(current_market_data, symbol_input)
+        
+        # Generate analysis
+        analysis = generate_agent_analysis(symbol_input, current_market_data)
+        
+        return dashboard, analysis
+        
+    except Exception as e:
+        # Return error state
+        error_fig = go.Figure()
+        error_fig.add_annotation(
+            text="⚠️ System temporarily unavailable<br>Please try again in a moment",
+            xref="paper", yref="paper",
+            x=0.5, y=0.5,
+            showarrow=False,
+            font=dict(size=20)
+        )
+        error_fig.update_layout(
+            height=400,
+            template='plotly_dark'
+        )
+        
+        error_analysis = "# ❌ System Error\n\nWe're experiencing temporary technical difficulties. Please refresh the page and try again."
+        
+        return error_fig, error_analysis
 
 # Create Gradio Interface
 with gr.Blocks(
@@ -500,17 +521,7 @@ with gr.Blocks(
         primary_hue="blue",
         secondary_hue="slate"
     ),
-    title="🤖 AI Trading Agents - Real-Time System",
-    css="""
-    .gradio-container {
-        max-width: 1200px !important;
-    }
-    .analysis-box {
-        border-left: 4px solid #4ECDC4;
-        padding-left: 15px;
-        margin: 10px 0;
-    }
-    """
+    title="🤖 AI Trading Agents - Real-Time System"
 ) as demo:
     
     gr.Markdown("""
@@ -536,7 +547,7 @@ with gr.Blocks(
             gr.Markdown("""
             **Tracked Stocks:** AAPL, GOOGL, MSFT, TSLA
             
-            *Data updates every 30 seconds automatically*
+            *Data updates automatically on each analysis*
             """)
             
         with gr.Column(scale=2):
@@ -553,12 +564,11 @@ with gr.Blocks(
         with gr.TabItem("🤖 AI Analysis"):
             with gr.Row():
                 analysis_output = gr.Markdown(
-                    label="Multi-Agent Analysis Report",
-                    show_label=True
+                    label="Multi-Agent Analysis Report"
                 )
     
     with gr.Row():
-        gr.Markdown("""
+        gr.Markdown(f"""
         ---
         **🔍 System Features:**
         - 📊 **Research Agent**: Fundamental analysis & market sentiment
@@ -566,8 +576,8 @@ with gr.Blocks(
         - 🛡️ **Risk Agent**: Position sizing & risk management  
         - 🎯 **Decision Engine**: Final trading recommendations
         
-        *Last System Update: {time}*
-        """.format(time=datetime.now().strftime("%Y-%m-%d %H:%M:%S")))
+        *Last System Update: {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}*
+        """)
     
     # Event handlers
     analyze_btn.click(
@@ -576,12 +586,11 @@ with gr.Blocks(
         outputs=[chart_output, analysis_output]
     )
     
-    # Auto-refresh every 30 seconds
+    # Load initial data when app starts
     demo.load(
-        fn=update_interface,
-        inputs=[symbol_input],
-        outputs=[chart_output, analysis_output],
-        every=30  # Refresh every 30 seconds
+        fn=lambda: update_interface(""),
+        inputs=[],
+        outputs=[chart_output, analysis_output]
     )
 
 # Launch for Hugging Face Spaces
@@ -589,6 +598,5 @@ if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=True,
-        show_error=True
+        share=True
     )