Update app.py

app.py

@@ -184,12 +184,13 @@ async def check_worker_health(worker_url: str) -> bool:
 
 async def health_check_loop():
     while True:
+        # Check all workers
         for worker_url in WORKER_URLS:
             healthy = await check_worker_health(worker_url)
             worker_health[worker_url]["healthy"] = healthy
             worker_health[worker_url]["last_check"] = time.time()
         
-        # Broadcast to dashboard
+        # Always broadcast stats to connected clients
         await broadcast_stats()
         
         await asyncio.sleep(HEALTH_CHECK_INTERVAL)
@@ -694,16 +695,91 @@ async def dashboard():
     </div>
     
     <script>
-        const ws = new WebSocket(`ws://${window.location.host}/ws`);
+        // Use wss:// for HTTPS, ws:// for HTTP
+        const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
+        let ws;
+        let usePolling = false;
+        
+        function connectWebSocket() {
+            try {
+                ws = new WebSocket(`${protocol}//${window.location.host}/ws`);
+                
+                ws.onopen = () => {
+                    console.log('✅ WebSocket connected');
+                    usePolling = false;
+                };
+                
+                ws.onmessage = (event) => {
+                    const data = JSON.parse(event.data);
+                    updateDashboard(data);
+                };
+                
+                ws.onerror = (error) => {
+                    console.error('❌ WebSocket error, switching to polling');
+                    usePolling = true;
+                    startPolling();
+                };
+                
+                ws.onclose = () => {
+                    console.log('🔌 WebSocket disconnected');
+                    if (!usePolling) {
+                        setTimeout(connectWebSocket, 3000);
+                    }
+                };
+            } catch (e) {
+                console.error('Failed to connect WebSocket, using polling');
+                usePolling = true;
+                startPolling();
+            }
+        }
+        
+        async function pollStats() {
+            if (!usePolling) return;
+            
+            try {
+                const response = await fetch('/api/status');
+                const data = await response.json();
+                
+                // Fetch worker stats too
+                const workersRes = await fetch('/workers');
+                const workersData = await workersRes.json();
+                
+                // Format data like WebSocket
+                const formattedData = {
+                    timestamp: Date.now() / 1000,
+                    mode: data.mode,
+                    load: data.current_load,
+                    workers: workersData.workers.map(w => ({
+                        url: w.url.split("//")[1].split(".")[0],
+                        healthy: w.healthy,
+                        active: w.active_requests || 0,
+                        total: 0,
+                        tokens: 0,
+                        latency: 0,
+                        role: "idle"
+                    })),
+                    cluster: {
+                        total_requests: 0,
+                        successful: 0,
+                        failed: 0,
+                        uptime: 0,
+                        rps: 0
+                    }
+                };
+                
+                updateDashboard(formattedData);
+            } catch (e) {
+                console.error('Polling error:', e);
+            }
+        }
         
-        ws.onmessage = (event) => {
-            const data = JSON.parse(event.data);
-            updateDashboard(data);
-        };
+        function startPolling() {
+            pollStats();
+            setInterval(pollStats, 1000);
+        }
         
-        ws.onerror = () => {
-            console.error('WebSocket error');
-        };
+        // Try WebSocket first
+        connectWebSocket();
         
         function updateDashboard(data) {
             // Mode