Update app.py

app.py

@@ -1,55 +1,78 @@
 import os
+import time
+import requests
 import gradio as gr
 import openstack
 
-# 1. Connect to OpenStack using HF Secrets
 def get_conn():
     return openstack.connect(
         auth_url=os.getenv("OS_AUTH_URL"),
         application_credential_id=os.getenv("OS_APPLICATION_CREDENTIAL_ID"),
         application_credential_secret=os.getenv("OS_APPLICATION_CREDENTIAL_SECRET"),
-        # ADD THIS LINE:
-        auth_type="v3applicationcredential", 
+        auth_type="v3applicationcredential",
         region_name="IU",
         interface="public"
     )
 
-
-def activate_copilot():
+def activate_and_monitor():
+    target_url = "https://chat.mch250095.projects.jetstream-cloud.org/"
+    instance_id = os.getenv("OS_INSTANCE_ID")
+    
     try:
         conn = get_conn()
+        server = conn.compute.find_server(instance_id)
         
-        # Pull the ID from Hugging Face Secrets
-        instance_id = os.getenv("OS_INSTANCE_ID")
+        if not server:
+            yield "❌ Error: Instance not found.", gr.update(visible=False)
+            return
+
+        # Milestone 1: Un-shelving Begin
+        yield "⏳ Milestone 1/3: Un-shelving Begin...", gr.update(visible=False)
         
-        if not instance_id:
-            return "❌ Error: OS_INSTANCE_ID secret is not set in HF Settings."
+        if server.status == "SHELVED_OFFLOADED":
+            conn.compute.unshelve_server(server)
+        elif server.status == "ACTIVE":
+            pass # Skip to next milestone
+        else:
+            yield f"⚠️ Instance is in {server.status} state. Attempting to proceed...", gr.update(visible=False)
 
-        # Find the server using the secret ID
-        server = conn.compute.find_server(instance_id)
+        # Milestone 2: Unshelved (Wait for ACTIVE status)
+        # wait_for_server waits until the status is 'ACTIVE'
+        conn.compute.wait_for_server(server, status='ACTIVE', wait=120)
+        yield "✅ Milestone 2/3: Unshelved (Instance is ACTIVE).", gr.update(visible=False)
+
+        # Milestone 3: URL Check (Wait for web service to respond)
+        yield "🔄 Milestone 3/3: Waiting for software to start at URL...", gr.update(visible=False)
         
-        if not server:
-            return f"❌ Error: Instance with ID {instance_id} not found."
-            
-        # Check current status before unshelving
-        if server.status == "ACTIVE":
-            return "ℹ️ Info: ComFit Copilot is already ACTIVE."
+        # Poll the URL for up to 2 minutes
+        for _ in range(24): 
+            try:
+                response = requests.get(target_url, timeout=5)
+                if response.status_code < 400:
+                    yield "🎉 Success! ComFit Copilot is alive.", gr.update(value=f"Visit: {target_url}", visible=True)
+                    return
+            except:
+                pass
+            time.sleep(5)
             
-        conn.compute.unshelve_server(server)
-        return f"✅ Success: Activation command sent for {server.name}."
-        
+        yield "⚠️ Instance is up, but the web link is not responding yet. Please try the link manually.", gr.update(value=f"Visit: {target_url}", visible=True)
+
     except Exception as e:
-        return f"❌ OpenStack Error: {str(e)}"
+        yield f"❌ Error: {str(e)}", gr.update(visible=False)
 
-# 2. Build the UI
+# UI Layout
 with gr.Blocks() as demo:
     gr.Markdown("# ComFit Copilot Controller")
-    gr.Markdown("Click the button below to unshelve the instance on Jetstream2.")
     
-    status_output = gr.Textbox(label="Status")
-    activate_btn = gr.Button("Activate ComFit Copilot", variant="primary")
-    
-    # Link button to function
-    activate_btn.click(fn=activate_copilot, outputs=status_output)
+    with gr.Column():
+        activate_btn = gr.Button("🚀 Activate ComFit Copilot", variant="primary")
+        status_text = gr.Label(value="Ready")
+        # Hidden link component that appears when ready
+        link_output = gr.Markdown(visible=False)
+
+    activate_btn.click(
+        fn=activate_and_monitor, 
+        outputs=[status_text, link_output]
+    )
 
-demo.launch()
+demo.launch()