Update app.py

app.py

@@ -1,38 +1,47 @@
 import gradio as gr
 import requests
+import threading
+import time
 
 # -------------------------
 # Backend Space URLs
 # -------------------------
 BACKENDS = {
-    "Sandbox Backend 1": "https://chocolaboratory-sandbox-backend.hf.space/",
-    "Sandbox Backend 2": "https://chocolaboratory-sandboxbackend2.hf.space/"
+    "Sandbox Backend 1": "https://chocolaboratory-sandboxbackend2.hf.space",
+    "Sandbox Backend 2": "https://chocolaboratory-sandbox-backend.hf.space"
 }
 
-# Track running sandboxes per backend for automatic selection
+# Track running sandboxes per backend
 backend_load = {name: 0 for name in BACKENDS.keys()}
+# Store user sandboxes: sandbox_id -> {"backend": ..., "start_time": ..., "card": ...}
+user_sandboxes = {}
+SANDBOX_TTL = 2*3600  # 2 hours
 
 # -------------------------
-# Frontend functions
+# API calls
 # -------------------------
 def choose_backend():
-    # Select backend with fewest sandboxes
     return min(backend_load, key=lambda k: backend_load[k])
 
-def launch_sandbox(code):
+def launch_sandbox_api(code):
     backend_name = choose_backend()
-    url = BACKENDS[backend_name] + "/launch_sandbox"
+    url = f"{BACKENDS[backend_name]}/launch_sandbox"
     try:
         r = requests.post(url, json={"code": code})
         r.raise_for_status()
         sandbox_id = r.json()["sandbox_id"]
         backend_load[backend_name] += 1
-        return f"Sandbox {sandbox_id} launched on {backend_name}", sandbox_id, backend_name
+        user_sandboxes[sandbox_id] = {"backend": backend_name, "start_time": time.time(), "card": None}
+        return sandbox_id, backend_name
     except Exception as e:
-        return f"Error: {e}", None, None
-
-def fetch_logs(sandbox_id, backend_name):
-    url = f"{BACKENDS[backend_name]}/fetch_logs/{sandbox_id}"
+        return None, str(e)
+
+def fetch_logs_api(sandbox_id):
+    info = user_sandboxes.get(sandbox_id)
+    if not info:
+        return "Sandbox not found"
+    backend = info["backend"]
+    url = f"{BACKENDS[backend]}/fetch_logs/{sandbox_id}"
     try:
         r = requests.get(url)
         r.raise_for_status()
@@ -41,18 +50,27 @@ def fetch_logs(sandbox_id, backend_name):
     except Exception as e:
         return str(e)
 
-def kill_sandbox(sandbox_id, backend_name):
-    url = f"{BACKENDS[backend_name]}/kill_sandbox/{sandbox_id}"
+def kill_sandbox_api(sandbox_id):
+    info = user_sandboxes.get(sandbox_id)
+    if not info:
+        return "Sandbox not found"
+    backend = info["backend"]
+    url = f"{BACKENDS[backend]}/kill_sandbox/{sandbox_id}"
     try:
         r = requests.post(url)
         r.raise_for_status()
-        backend_load[backend_name] -= 1
-        return r.json().get("status", "Killed")
+        backend_load[backend] -= 1
+        del user_sandboxes[sandbox_id]
+        return f"Sandbox {sandbox_id} killed"
     except Exception as e:
         return str(e)
 
-def status_sandbox(sandbox_id, backend_name):
-    url = f"{BACKENDS[backend_name]}/status_sandbox/{sandbox_id}"
+def status_sandbox_api(sandbox_id):
+    info = user_sandboxes.get(sandbox_id)
+    if not info:
+        return "Sandbox not found"
+    backend = info["backend"]
+    url = f"{BACKENDS[backend]}/status_sandbox/{sandbox_id}"
     try:
         r = requests.get(url)
         r.raise_for_status()
@@ -61,65 +79,71 @@ def status_sandbox(sandbox_id, backend_name):
         return str(e)
 
 # -------------------------
-# AWS-style frontend UI
+# Sandbox card creation
+# -------------------------
+def create_sandbox_card(sandbox_id, sandbox_container):
+    info = user_sandboxes[sandbox_id]
+    backend = info["backend"]
+
+    with gr.Column(elem_classes="sandbox-card") as card:
+        gr.Markdown(f"**Sandbox ID:** {sandbox_id}")
+        status_badge = gr.Markdown(f"<span style='color:green'>Running</span>")
+        gr.Markdown(f"**Backend:** {backend}")
+        runtime_box = gr.Textbox(label="Runtime (sec)", value="0", interactive=False)
+        logs_box = gr.Textbox(label="Logs", lines=10, interactive=False)
+
+        # Buttons inside card
+        with gr.Row():
+            logs_btn = gr.Button("Fetch Logs")
+            kill_btn = gr.Button("Kill Sandbox")
+            status_btn = gr.Button("Check Status")
+
+        # Button actions
+        logs_btn.click(lambda sid=sandbox_id: fetch_logs_api(sid), inputs=[], outputs=logs_box)
+        kill_btn.click(lambda sid=sandbox_id: kill_sandbox_api(sid), inputs=[], outputs=status_badge)
+        status_btn.click(lambda sid=sandbox_id: status_sandbox_api(sid), inputs=[], outputs=runtime_box)
+
+    # Background updater
+    def updater():
+        while sandbox_id in user_sandboxes:
+            elapsed = int(time.time() - info["start_time"])
+            runtime_box.value = str(elapsed)
+            logs_box.value = fetch_logs_api(sandbox_id)
+            time.sleep(2)
+        # Remove card if sandbox killed/expired
+        sandbox_container.remove(card)
+
+    threading.Thread(target=updater, daemon=True).start()
+    info["card"] = card
+    return card
+
+# -------------------------
+# Main Gradio UI
 # -------------------------
 with gr.Blocks(css="""
     .sandbox-card {
         border-radius: 12px;
         padding: 15px;
-        margin: 5px;
-        background-color: #ffffff;
-        box-shadow: 0 3px 6px rgba(0,0,0,0.1);
+        margin: 10px;
+        background-color: #fefefe;
+        box-shadow: 0 4px 12px rgba(0,0,0,0.15);
     }
-    .header {font-size: 24px; font-weight: bold; margin-bottom: 15px;}
 """) as demo:
+    gr.Markdown("# 🛡 Sandbox Dashboard (AWS-style)")
 
-    gr.Markdown("<div class='header'>🛡 Sandbox Service Dashboard</div>", elem_classes="header")
-
-    # Code editor & launch
-    code_input = gr.Textbox(label="Python Code", lines=15, placeholder="Write your code here...")
+    code_input = gr.Textbox(label="Python Code", lines=10)
     launch_btn = gr.Button("Launch Sandbox")
-    
-    # Status and storage
     launch_status = gr.Textbox(label="Launch Status")
-    sandbox_id_storage = gr.State(value=None)
-    backend_storage = gr.State(value=None)
-
-    # Sandbox cards container
-    with gr.Column():
-        logs_btn = gr.Button("Fetch Logs")
-        kill_btn = gr.Button("Kill Sandbox")
-        status_btn = gr.Button("Check Status")
-
-    logs_output = gr.Textbox(label="Logs", lines=12)
-    kill_output = gr.Textbox(label="Kill Status")
-    status_output = gr.Textbox(label="Sandbox Status")
-
-    # -------------------------
-    # Button actions
-    # -------------------------
-    launch_btn.click(
-        launch_sandbox,
-        inputs=[code_input],
-        outputs=[launch_status, sandbox_id_storage, backend_storage]
-    )
-
-    logs_btn.click(
-        fetch_logs,
-        inputs=[sandbox_id_storage, backend_storage],
-        outputs=logs_output
-    )
-
-    kill_btn.click(
-        kill_sandbox,
-        inputs=[sandbox_id_storage, backend_storage],
-        outputs=kill_output
-    )
-
-    status_btn.click(
-        status_sandbox,
-        inputs=[sandbox_id_storage, backend_storage],
-        outputs=status_output
-    )
+
+    sandbox_container = gr.Column()
+
+    def on_launch(code):
+        sandbox_id, backend_or_error = launch_sandbox_api(code)
+        if not sandbox_id:
+            return f"Error: {backend_or_error}"
+        create_sandbox_card(sandbox_id, sandbox_container)
+        return f"Sandbox {sandbox_id} launched on {user_sandboxes[sandbox_id]['backend']}"
+
+    launch_btn.click(on_launch, inputs=code_input, outputs=launch_status)
 
 demo.launch()