Update app.py

app.py

@@ -58,82 +58,84 @@ def status_sandbox_api(sandbox_id):
     info = user_sandboxes.get(sandbox_id)
     if not info:
         return "Sandbox not found"
-    try:
-        return info["client"].predict(api_name="/status_sandbox", sandbox_id=sandbox_id)
-    except Exception as e:
-        return str(e)
+    elapsed = int(time.time() - info["start_time"])
+    return f"Running for {elapsed} seconds"
 
 # -------------------------
-# Sandbox card creation
+# Dark-themed CSS
 # -------------------------
-def create_sandbox_card(sandbox_id, sandbox_container, code, requirements):
-    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)
-
-        # Side-by-side panels: Main Code | requirements.txt
-        with gr.Row():
-            code_box = gr.Textbox(label="Main Code", value=code, lines=10, interactive=False)
-            req_box = gr.Textbox(label="requirements.txt", value=requirements, lines=10, interactive=False)
-
-        # Buttons
-        with gr.Row():
-            logs_btn = gr.Button("Fetch Logs")
-            kill_btn = gr.Button("Kill Sandbox")
-            status_btn = gr.Button("Check Status")
-
-        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)
-            if elapsed >= SANDBOX_TTL:
-                kill_sandbox_api(sandbox_id)
-                break
-            time.sleep(2)
-        sandbox_container.remove(card)
-
-    threading.Thread(target=updater, daemon=True).start()
-    return card
+DARK_CSS = """
+body {
+    background-color: #1e1e1e;
+    color: #f0f0f0;
+}
+.sandbox-card {
+    border-radius: 12px;
+    padding: 15px;
+    margin: 10px;
+    background-color: #2b2b2b;
+    box-shadow: 0 4px 12px rgba(0,0,0,0.4);
+}
+.gr-button {
+    background-color: #3a3a3a;
+    color: #f0f0f0;
+    border: 1px solid #555;
+}
+.gr-button:hover {
+    background-color: #505050;
+}
+.gr-textbox, .gr-markdown {
+    background-color: #1e1e1e;
+    color: #f0f0f0;
+    border: 1px solid #444;
+}
+"""
 
 # -------------------------
 # Main Gradio UI
 # -------------------------
-with gr.Blocks(css="""
-    .sandbox-card {
-        border-radius: 12px;
-        padding: 15px;
-        margin: 10px;
-        background-color: #fefefe;
-        box-shadow: 0 4px 12px rgba(0,0,0,0.15);
-    }
-""") as demo:
-    gr.Markdown("# Sandbox Dashboard")
-
-    code_input = gr.Textbox(label="Main Code", lines=10)
-    req_input = gr.Textbox(label="requirements.txt", lines=5)
+with gr.Blocks(css=DARK_CSS) as demo:
+    gr.Markdown("# 🛡 Sandbox Dashboard")
+
+    with gr.Row():
+        code_input = gr.Textbox(label="Main Code", lines=10)
+        req_input = gr.Textbox(label="requirements.txt", lines=5)
     launch_btn = gr.Button("Launch Sandbox")
     launch_status = gr.Textbox(label="Launch Status")
 
     sandbox_container = gr.Column()
 
+    # -------------------------
+    # Handlers for dynamic sandbox cards
+    # -------------------------
     def on_launch(code, requirements):
         sandbox_id, backend_or_error = launch_sandbox_api(code, requirements)
         if not sandbox_id:
             return f"Error: {backend_or_error}"
-        create_sandbox_card(sandbox_id, sandbox_container, code, requirements)
-        return f"Sandbox {sandbox_id} launched on {user_sandboxes[sandbox_id]['backend']}"
+
+        # Create a new card in the container
+        with sandbox_container:
+            with gr.Column(elem_classes="sandbox-card") as card:
+                gr.Markdown(f"**Sandbox ID:** {sandbox_id}")
+                gr.Markdown(f"**Backend:** {backend_or_error}")
+                runtime_box = gr.Textbox(label="Runtime (sec)", value="0", interactive=False)
+                logs_box = gr.Textbox(label="Logs", lines=10, interactive=False)
+
+                with gr.Row():
+                    code_box = gr.Textbox(label="Main Code", value=code, lines=10, interactive=False)
+                    req_box = gr.Textbox(label="requirements.txt", value=requirements, lines=10, interactive=False)
+
+                # Buttons
+                logs_btn = gr.Button("Fetch Logs")
+                kill_btn = gr.Button("Kill Sandbox")
+                status_btn = gr.Button("Check Status")
+
+                # Assign clicks using lambdas
+                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=logs_box)
+                status_btn.click(lambda sid=sandbox_id: status_sandbox_api(sid), inputs=[], outputs=runtime_box)
+
+        return f"Sandbox {sandbox_id} launched on {backend_or_error}"
 
     launch_btn.click(on_launch, inputs=[code_input, req_input], outputs=launch_status)