Update app.py

app.py

@@ -5,6 +5,7 @@ import yfinance as yf
 import plotly.graph_objects as go
 from plotly.subplots import make_subplots
 from datetime import datetime, timedelta
+import asyncio
 import time
 import warnings
 warnings.filterwarnings('ignore')
@@ -22,10 +23,7 @@ class RealTimeMarketData:
         """Generate realistic live market data with actual changes"""
         current_time = datetime.now()
         
-        # Update every 10 seconds
-        if (current_time - self.last_update).seconds < 10 and self.update_counter > 0:
-            return self.get_current_data()
-            
+        # Always generate new data for demo purposes
         self.update_counter += 1
         self.last_update = current_time
         
@@ -69,7 +67,7 @@ class RealTimeMarketData:
                 # Ensure timestamps match
                 current_timestamps = self.timestamps[-len(self.data_history[symbol]):]
                 if len(current_timestamps) != len(self.data_history[symbol]):
-                    current_timestamps = [f"{(datetime.now() - timedelta(seconds=i*10)).strftime('%H:%M:%S')}" 
+                    current_timestamps = [f"{(datetime.now() - timedelta(seconds=i*5)).strftime('%H:%M:%S')}" 
                                         for i in range(len(self.data_history[symbol])-1, -1, -1)]
                 
                 # Calculate change
@@ -115,7 +113,7 @@ class RealTimeMarketData:
             self.data_history[symbol].pop(0)
         
         # Generate timestamps
-        current_timestamps = [f"{(datetime.now() - timedelta(seconds=i*10)).strftime('%H:%M:%S')}" 
+        current_timestamps = [f"{(datetime.now() - timedelta(seconds=i*5)).strftime('%H:%M:%S')}" 
                             for i in range(len(self.data_history[symbol])-1, -1, -1)]
         
         change = ((new_price - self.data_history[symbol][0]) / self.data_history[symbol][0]) * 100
@@ -129,23 +127,6 @@ class RealTimeMarketData:
             'update_count': self.update_counter,
             'last_updated': datetime.now().strftime('%H:%M:%S')
         }
-    
-    def get_current_data(self):
-        """Return current data without updating"""
-        live_data = {}
-        for symbol in self.symbols:
-            if symbol in self.data_history and self.data_history[symbol]:
-                live_data[symbol] = {
-                    'prices': self.data_history[symbol].copy(),
-                    'timestamps': self.timestamps[-len(self.data_history[symbol]):],
-                    'current_price': self.last_prices.get(symbol, 150),
-                    'change': ((self.last_prices.get(symbol, 150) - self.data_history[symbol][0]) / 
-                             self.data_history[symbol][0] * 100) if self.data_history[symbol] else 0,
-                    'volume': np.random.randint(1000000, 5000000),
-                    'update_count': self.update_counter,
-                    'last_updated': self.last_update.strftime('%H:%M:%S')
-                }
-        return live_data
 
 class AI_TradingAgents:
     def __init__(self):
@@ -273,7 +254,7 @@ def create_live_dashboard():
         subplot_titles=[
             '📈 Live Price Movement (Auto-Refresh)',
             '📊 Real-Time Performance', 
-            '🔄 Minute-by-Minute Changes',
+            '🔄 Recent Price Changes',
             '🎯 Market Sentiment'
         ],
         specs=[
@@ -407,9 +388,22 @@ def generate_analysis(symbol_input, live_data):
 
 def update_interface(symbol_input=""):
     """Main update function called automatically"""
-    dashboard, live_data = create_live_dashboard()
-    analysis = generate_analysis(symbol_input, live_data)
-    return dashboard, analysis
+    try:
+        dashboard, live_data = create_live_dashboard()
+        analysis = generate_analysis(symbol_input, live_data)
+        return dashboard, analysis
+    except Exception as e:
+        # Return error state
+        error_fig = go.Figure()
+        error_fig.add_annotation(
+            text="🔄 Updating... Please wait",
+            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')
+        return error_fig, "# 🔄 Updating... Please wait"
 
 # Create auto-refreshing interface
 with gr.Blocks(
@@ -417,13 +411,12 @@ with gr.Blocks(
         primary_hue="blue",
         secondary_hue="slate"
     ),
-    title="🤖 Auto-Refresh AI Trading System",
-    refresh_interval=10  # Auto-refresh every 10 seconds
+    title="🤖 Auto-Refresh AI Trading System"
 ) as demo:
     
     gr.Markdown("""
     # 🤖 Real-Time AI Trading System
-    ## *Auto-Refresh Every 10 Seconds*
+    ## *Auto-Refresh Every 5 Seconds*
     
     🔄 **Charts and data update automatically - no clicking needed!**
     """)
@@ -439,7 +432,7 @@ with gr.Blocks(
             gr.Markdown("""
             **Tracked Stocks:** AAPL, GOOGL, MSFT, TSLA
             
-            ⏰ **Auto-Refresh:** Every 10 seconds
+            ⏰ **Auto-Refresh:** Every 5 seconds
             �� **Live Data:** Real-time price movements
             🤖 **AI Analysis:** Dynamic recommendations
             """)
@@ -450,28 +443,37 @@ with gr.Blocks(
     with gr.Tabs():
         with gr.TabItem("📈 Live Charts"):
             chart_output = gr.Plot(
-                label="Auto-Refreshing Market Data",
-                every=10  # Auto-refresh every 10 seconds
+                label="Auto-Refreshing Market Data"
             )
         
         with gr.TabItem("🤖 AI Analysis"):
-            analysis_output = gr.Markdown(
-                every=10  # Auto-refresh every 10 seconds
-            )
+            analysis_output = gr.Markdown()
     
     gr.Markdown(f"""
     ---
     **🔄 System Status:** Auto-Refresh Active • **📅 Last Refresh:** {datetime.now().strftime('%H:%M:%S')}
-    **💡 Tip:** The page automatically updates every 10 seconds with live data
+    **💡 Tip:** The page automatically updates every 5 seconds with live data
     **🎯 Tracking:** {len(market_data.symbols)} stocks with real-time AI analysis
     """)
     
-    # Set up auto-refresh
+    # Set up auto-refresh using JavaScript
     demo.load(
         fn=update_interface,
         inputs=[symbol_input],
         outputs=[chart_output, analysis_output]
     )
+    
+    # Add JavaScript for auto-refresh
+    demo.js = """
+    <script>
+    function autoRefresh() {
+        setTimeout(function() {
+            location.reload();
+        }, 5000); // Refresh every 5 seconds
+    }
+    window.onload = autoRefresh;
+    </script>
+    """
 
 # Launch the application
 if __name__ == "__main__":