Update app.py

app.py

@@ -5,8 +5,11 @@ import gradio as gr
 import openstack
 from openstack import exceptions
 
-# Global counter
+# GLOBAL SHARED STATE
 active_user_count = 0
+is_comfit_alive = False  # Track if software is reachable across all sessions
+clean_link = "comfit-copilot.mch250095.projects.jetstream-cloud.org:3000"
+full_url = f"http://{clean_link}"
 
 def get_conn():
     return openstack.connect(
@@ -19,7 +22,7 @@ def get_conn():
     )
 
 def get_current_status():
-    global active_user_count
+    global active_user_count, is_comfit_alive
     try:
         conn = get_conn()
         instance_id = os.getenv("OS_INSTANCE_ID")
@@ -36,43 +39,60 @@ def get_current_status():
             }
             status_text = status_map.get(server.status, f"❓ STATE: {server.status}")
         
-        return status_text, f"👥 Active Users: {active_user_count}"
+        # Background health check to update the global 'is_comfit_alive' flag
+        if "RUNNING" in status_text:
+            try:
+                if requests.get(full_url, timeout=2).status_code < 500:
+                    is_comfit_alive = True
+                else:
+                    is_comfit_alive = False
+            except:
+                is_comfit_alive = False
+        else:
+            is_comfit_alive = False
+
+        # Determine link visibility for the live timer update
+        link_update = gr.update(value=f"### [Visit: {clean_link}]({full_url})", visible=is_comfit_alive)
+        
+        return status_text, f"👥 Active Users: {active_user_count}", link_update
     except Exception as e:
-        return f"❌ Connection Error: {str(e)}", f"👥 Active Users: {active_user_count}"
+        return f"❌ Error: {str(e)}", f"👥 Active Users: {active_user_count}", gr.update()
 
 def user_joined():
-    global active_user_count
+    global active_user_count, is_comfit_alive
     active_user_count += 1
-    return f"👥 Active Users: {active_user_count}"
+    # If already alive, show the link immediately on page load
+    link_visibility = gr.update(value=f"### [Visit: {clean_link}]({full_url})", visible=is_comfit_alive)
+    return f"👥 Active Users: {active_user_count}", link_visibility
 
 def user_left():
     global active_user_count
     active_user_count = max(0, active_user_count - 1)
 
 def manage_instance():
-    clean_link = "comfit-copilot.mch250095.projects.jetstream-cloud.org:3000"
-    full_url = f"http://{clean_link}"
-    instance_id = os.getenv("OS_INSTANCE_ID")
+    global is_comfit_alive
     try:
         conn = get_conn()
+        instance_id = os.getenv("OS_INSTANCE_ID")
         server = conn.compute.find_server(instance_id)
+        
         if not server:
             yield ("❌ Error: Instance Not Found", gr.update(visible=False))
             return
 
-        if server.status == "ACTIVE":
-            yield ("ℹ️ Already ACTIVE. Checking link...", gr.update(visible=False))
-        else:
+        if server.status != "ACTIVE":
             yield ("⏳ Milestone 1/3: Un-shelving...", gr.update(visible=False))
             if server.status == "SHELVED_OFFLOADED":
                 conn.compute.unshelve_server(server)
             conn.compute.wait_for_server(server, status='ACTIVE', wait=600)
             yield ("✅ Milestone 2/3: VM is ACTIVE.", gr.update(visible=False))
 
-        yield ("🔄 Milestone 3/3: Waiting for software...", gr.update(visible=False))
+        # Check software
+        yield ("🔄 Milestone 3/3: Connecting to ComFit...", gr.update(visible=False))
         for i in range(24): 
             try:
                 if requests.get(full_url, timeout=3).status_code < 500:
+                    is_comfit_alive = True
                     yield ("🎉 Success! ComFit is ready.", gr.update(value=f"### [Visit: {clean_link}]({full_url})", visible=True))
                     return
             except: pass
@@ -81,13 +101,12 @@ def manage_instance():
     except Exception as e:
         yield (f"❌ Error: {str(e)}", gr.update(visible=False))
 
-# JavaScript for sound + Inactivity Timeout (10 minutes)
+# Combined JS for Sound + Inactivity Timeout (Change MAX_IDLE here)
 combined_js = """
 <script>
 let idleTime = 0;
-const MAX_IDLE = 10; // Minutes until auto-disconnect
+const MAX_IDLE = 10; // <--- CHANGE TIMEOUT HERE (in minutes)
 
-// Sound function
 function playNotification(status) {
     if (status.includes("RUNNING")) {
         var audio = new Audio('https://cdn.freesound.org');
@@ -95,29 +114,24 @@ function playNotification(status) {
     }
 }
 
-// Inactivity Monitor
 function resetTimer() { idleTime = 0; }
-window.onload = resetTimer;
 window.onmousemove = resetTimer;
 window.onmousedown = resetTimer;
-window.ontouchstart = resetTimer;
 window.onclick = resetTimer;
 window.onkeypress = resetTimer;
 
 setInterval(function() {
     idleTime++;
     if (idleTime >= MAX_IDLE) {
-        alert("Session Expired due to 10 minutes of inactivity. Please refresh the page.");
-        window.location.reload(); // Or stop timers by redirecting
+        alert("Session Expired due to inactivity. Please refresh.");
+        window.location.reload();
     }
-}, 60000); // Check every minute
+}, 60000); 
 </script>
 """
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.HTML(combined_js)
-    
-    # Session state for cleanup
     session_tracker = gr.State(value="active", delete_callback=user_left)
 
     gr.Markdown("# 🚀 ComFit Copilot Shared Control Panel")
@@ -133,11 +147,13 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     live_status.change(None, inputs=live_status, js="(s) => playNotification(s)")
     activate_btn.click(fn=manage_instance, outputs=[status_label, link_box])
 
+    # Timer refreshes status, user count, and global link visibility every 5s
     timer = gr.Timer(5)
-    timer.tick(fn=get_current_status, outputs=[live_status, user_display])
+    timer.tick(fn=get_current_status, outputs=[live_status, user_display, link_box])
     
-    demo.load(fn=user_joined, outputs=user_display)
-    demo.load(fn=get_current_status, outputs=[live_status, user_display])
+    # Load initial shared state for new users
+    demo.load(fn=user_joined, outputs=[user_display, link_box])
+    demo.load(fn=get_current_status, outputs=[live_status, user_display, link_box])
 
 if __name__ == "__main__":
     demo.launch()