Update app.py

app.py

@@ -5,22 +5,31 @@ import requests
 # Backend Space URLs
 # -------------------------
 BACKENDS = {
-    "Sandbox Backend 1": "https://your-backend1.hf.space",
-    "Sandbox Backend 2": "https://your-backend2.hf.space"
+    "Sandbox Backend 1": "https://chocolaboratory-sandbox-backend.hf.space/",
+    "Sandbox Backend 2": "https://chocolaboratory-sandboxbackend2.hf.space/"
 }
 
+# Track running sandboxes per backend for automatic selection
+backend_load = {name: 0 for name in BACKENDS.keys()}
+
 # -------------------------
 # Frontend functions
 # -------------------------
-def launch_sandbox(code, backend_name):
+def choose_backend():
+    # Select backend with fewest sandboxes
+    return min(backend_load, key=lambda k: backend_load[k])
+
+def launch_sandbox(code):
+    backend_name = choose_backend()
     url = BACKENDS[backend_name] + "/launch_sandbox"
     try:
         r = requests.post(url, json={"code": code})
         r.raise_for_status()
         sandbox_id = r.json()["sandbox_id"]
-        return f"Sandbox launched! ID: {sandbox_id}", sandbox_id
+        backend_load[backend_name] += 1
+        return f"Sandbox {sandbox_id} launched on {backend_name}", sandbox_id, backend_name
     except Exception as e:
-        return f"Error: {e}", None
+        return f"Error: {e}", None, None
 
 def fetch_logs(sandbox_id, backend_name):
     url = f"{BACKENDS[backend_name]}/fetch_logs/{sandbox_id}"
@@ -37,6 +46,7 @@ def kill_sandbox(sandbox_id, backend_name):
     try:
         r = requests.post(url)
         r.raise_for_status()
+        backend_load[backend_name] -= 1
         return r.json().get("status", "Killed")
     except Exception as e:
         return str(e)
@@ -51,51 +61,65 @@ def status_sandbox(sandbox_id, backend_name):
         return str(e)
 
 # -------------------------
-# Gradio AWS-style frontend UI
+# AWS-style frontend UI
 # -------------------------
-with gr.Blocks() as demo:
-    gr.Markdown("# Sandbox Service Frontend")
+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);
+    }
+    .header {font-size: 24px; font-weight: bold; margin-bottom: 15px;}
+""") as demo:
 
-    # Backend selector
-    backend_selector = gr.Dropdown(list(BACKENDS.keys()), label="Choose Backend")
+    gr.Markdown("<div class='header'>🛡 Sandbox Service Dashboard</div>", elem_classes="header")
 
-    # Code editor
+    # Code editor & launch
     code_input = gr.Textbox(label="Python Code", lines=15, placeholder="Write your code here...")
     launch_btn = gr.Button("Launch Sandbox")
-    status_output = gr.Textbox(label="Launch Status")
+    
+    # Status and storage
+    launch_status = gr.Textbox(label="Launch Status")
     sandbox_id_storage = gr.State(value=None)
+    backend_storage = gr.State(value=None)
 
-    # Logs and Kill buttons
-    with gr.Row():
+    # 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=15)
+
+    logs_output = gr.Textbox(label="Logs", lines=12)
     kill_output = gr.Textbox(label="Kill Status")
-    status_output_box = gr.Textbox(label="Sandbox Status")
+    status_output = gr.Textbox(label="Sandbox Status")
 
     # -------------------------
     # Button actions
     # -------------------------
     launch_btn.click(
         launch_sandbox,
-        inputs=[code_input, backend_selector],
-        outputs=[status_output, sandbox_id_storage]
+        inputs=[code_input],
+        outputs=[launch_status, sandbox_id_storage, backend_storage]
     )
+
     logs_btn.click(
         fetch_logs,
-        inputs=[sandbox_id_storage, backend_selector],
+        inputs=[sandbox_id_storage, backend_storage],
         outputs=logs_output
     )
+
     kill_btn.click(
         kill_sandbox,
-        inputs=[sandbox_id_storage, backend_selector],
+        inputs=[sandbox_id_storage, backend_storage],
         outputs=kill_output
     )
+
     status_btn.click(
         status_sandbox,
-        inputs=[sandbox_id_storage, backend_selector],
-        outputs=status_output_box
+        inputs=[sandbox_id_storage, backend_storage],
+        outputs=status_output
     )
 
 demo.launch()