Update app.py

app.py

@@ -20,10 +20,7 @@ class ConnectionManager:
     def _monitor(self):
         while not self.stop_event.is_set():
             try:
-                response = requests.get(
-                    f"{COLAB_URL}/health",
-                    timeout=5
-                )
+                response = requests.get(f"{COLAB_URL}/health", timeout=5)
                 self.active = response.status_code == 200
             except:
                 self.active = False
@@ -63,24 +60,19 @@ with gr.Blocks(title="Phi-3 Chatbot") as demo:
     def update_status():
         return "🟢 Connected" if conn.active else "🔴 Backend offline"
     
-    # Periodically update status
-    demo.load(
-        lambda: None,
-        inputs=None,
-        outputs=None,
-        _js="""
-        function() {
-            setInterval(() => {
-                document.getElementById('status').dispatchEvent(new Event('change'));
-            }, 3000);
-            return [];
-        }
-        """
-    )
-    status.change(
-        update_status,
-        inputs=None,
-        outputs=status
+    # Add dummy hidden component for polling
+    dummy = gr.Textbox(visible=False)
+    
+    # Set up polling
+    def poll_status():
+        while True:
+            time.sleep(POLL_INTERVAL)
+            yield update_status()
+    
+    dummy.change(
+        fn=poll_status,
+        outputs=status,
+        show_progress=False
     )
     
     chatbot = gr.Chatbot(height=350)