Update app.py

app.py

@@ -43,46 +43,40 @@ def manage_instance(action, confirmed=False):
         server = conn.compute.find_server(instance_id)
         
         if not server:
-            yield "❌ Error: Instance ID not found.", gr.update(visible=False)
+            yield ("❌ Error: Instance ID not found.", gr.update(visible=False))
             return
 
         if action == "activate":
-            yield "⏳ Milestone 1/3: Un-shelving Begin...", gr.update(visible=False)
+            yield ("⏳ Milestone 1/3: Un-shelving Begin...", gr.update(visible=False))
             if server.status == "SHELVED_OFFLOADED":
                 conn.compute.unshelve_server(server)
             
-            # Wait for VM to reach ACTIVE state
             conn.compute.wait_for_server(server, status='ACTIVE', wait=180)
-            yield "✅ Milestone 2/3: Unshelved! VM is now ACTIVE.", gr.update(visible=False)
+            yield ("✅ Milestone 2/3: Unshelved! VM is now ACTIVE.", gr.update(visible=False))
 
-            yield "🔄 Milestone 3/3: Waiting for software...", gr.update(visible=False)
-            
-            # Poll for the web service to boot (~30s)
+            yield ("🔄 Milestone 3/3: Waiting for software...", gr.update(visible=False))
             for i in range(24): 
                 try:
                     response = requests.get(full_url, timeout=3)
                     if response.status_code < 400:
-                        yield "🎉 Success! Ready.", gr.update(value=f"### [Visit: {clean_link}]({full_url})", visible=True)
+                        yield ("🎉 Success! Ready.", gr.update(value=f"### [Visit: {clean_link}]({full_url})", visible=True))
                         return
                 except:
                     pass
-                
-                # IMPORTANT: Yield 2 values every loop to keep Gradio happy
-                yield f"🔄 Milestone 3/3: Software booting (Attempt {i+1}/24)...", gr.update(visible=False)
+                yield (f"🔄 Milestone 3/3: Software booting (Attempt {i+1}/24)...", gr.update(visible=False))
                 time.sleep(5)
             
-            yield "⚠️ Service slow.", gr.update(value=f"### [Manual Link: {clean_link}]({full_url})", visible=True)
+            yield ("⚠️ Service slow.", gr.update(value=f"### [Manual Link: {clean_link}]({full_url})", visible=True))
 
         elif action == "shelve":
-            yield "💤 Shelving Begin... (Saving and Stopping)", gr.update(visible=False)
+            yield ("💤 Shelving Begin... (Saving and Stopping)", gr.update(visible=False))
             conn.compute.shelve_server(server)
             conn.compute.wait_for_server(server, status='SHELVED_OFFLOADED', wait=180)
-            yield "🛌 Instance is now SHELVED.", gr.update(visible=False)
+            yield ("🛌 Instance is now SHELVED.", gr.update(visible=False))
 
     except Exception as e:
-        yield f"❌ Error: {str(e)}", gr.update(visible=False)
+        yield (f"❌ Error: {str(e)}", gr.update(visible=False))
 
-# UI Layout
 with gr.Blocks() as demo:
     gr.Markdown("# 🚀 ComFit Copilot Control Panel")
     
@@ -91,7 +85,6 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         activate_btn = gr.Button("🚀 Activate (Unshelve)", variant="primary")
-        
         with gr.Column():
             confirm_check = gr.Checkbox(label="Confirm Shelve", value=False)
             shelve_btn = gr.Button("💤 Shelve (Stop)", variant="stop", interactive=False)
@@ -99,17 +92,24 @@ with gr.Blocks() as demo:
     status_label = gr.Label(value="Ready")
     link_box = gr.Markdown(visible=False)
 
-    # UI Logic
     confirm_check.change(fn=lambda x: gr.update(interactive=x), inputs=confirm_check, outputs=shelve_btn)
 
-    activate_btn.click(fn=lambda: manage_instance("activate"), outputs=[status_label, link_box])
-    shelve_btn.click(fn=lambda: manage_instance("shelve", True), outputs=[status_label, link_box])
+    # Added concurrency_limit to support the generator (yield) logic
+    activate_btn.click(
+        fn=lambda: manage_instance("activate"), 
+        outputs=[status_label, link_box],
+        concurrency_limit=5
+    )
+    
+    shelve_btn.click(
+        fn=lambda: manage_instance("shelve", True), 
+        outputs=[status_label, link_box],
+        concurrency_limit=5
+    )
 
-    # Timer for status refresh (Gradio 5/6 style)
     timer = gr.Timer(30)
     timer.tick(fn=get_current_status, outputs=live_status)
     demo.load(fn=get_current_status, outputs=live_status)
 
-# Start the app and apply the theme
 if __name__ == "__main__":
     demo.launch(theme=gr.themes.Soft())