Update app.py

app.py

@@ -2,11 +2,8 @@ import gradio as gr
 from gradio_client import Client
 import random
 import time
-import threading
 
-# -------------------------
-# Backend HF clients
-# -------------------------
+# Backend clients
 backend_engines = [
     Client("ChocoLaboratory/SANDBOX_BACKEND"),
     Client("ChocoLaboratory/SANDBOXBACKEND2")
@@ -14,9 +11,10 @@ backend_engines = [
 
 SANDBOX_TTL_SEC = 2 * 3600  # 2 hours
 
-# -------------------------
-# Helper functions
-# -------------------------
+# States
+sandbox_backend_state = gr.State(value=None)
+start_time_state = gr.State(value=0)
+
 def pick_engine():
     return random.choice(backend_engines)
 
@@ -27,17 +25,6 @@ def launch_sandbox(code):
     start_time = time.time()
     return f"Sandbox launched! ID: {sandbox_id}\n", sandbox_id, SANDBOX_TTL_SEC, start_time, engine
 
-def run_command(cmd, sandbox_id, start_time, engine):
-    if not sandbox_id:
-        return "No sandbox ID. Launch a sandbox first!", "0:00"
-    try:
-        # Run the command on the backend sandbox
-        engine.predict(code=cmd, api_name="/launch_sandbox")  # ideally /run_command endpoint
-        logs, remaining = fetch_logs_and_timer(sandbox_id, start_time, engine)
-        return logs, remaining
-    except Exception as e:
-        return f"Error: {e}", "0:00"
-
 def fetch_logs_and_timer(sandbox_id, start_time, engine):
     try:
         logs = engine.predict(api_name="/fetch_logs")
@@ -49,6 +36,16 @@ def fetch_logs_and_timer(sandbox_id, start_time, engine):
     time_str = f"{minutes:02d}:{seconds:02d}"
     return logs, time_str
 
+def run_command(cmd, sandbox_id, start_time, engine):
+    if not sandbox_id:
+        return "No sandbox ID. Launch a sandbox first!", "0:00"
+    try:
+        engine.predict(code=cmd, api_name="/launch_sandbox")  # ideally /run_command
+        logs, remaining = fetch_logs_and_timer(sandbox_id, start_time, engine)
+        return logs, remaining
+    except Exception as e:
+        return f"Error: {e}", "0:00"
+
 def kill_sandbox(sandbox_id, engine):
     if not sandbox_id:
         return "No sandbox ID to kill."
@@ -72,60 +69,54 @@ def get_status(sandbox_id, start_time, engine):
         return f"Error: {e}", "0:00"
 
 # -------------------------
-# Gradio UI
+# UI
 # -------------------------
 with gr.Blocks() as demo:
-    # CSS for terminal
+    # Dark theme CSS
     gr.HTML("""
     <style>
-    .terminal-box textarea {
-        background-color: #000000;
-        color: #ffffff;
-        font-family: monospace;
-        font-size: 14px;
-    }
+    body {background-color: #1e1e1e; color: #ffffff; font-family: monospace;}
+    .terminal-box textarea {background-color: #000000; color: #ffffff; font-family: monospace; font-size:14px; border-radius:6px;}
+    .side-panel {background-color: #2e2e2e; padding:10px; border-radius:6px;}
+    .side-panel button {margin-bottom:10px; width:100%;}
     </style>
     """)
 
     with gr.Row():
+        # Terminal Column
         with gr.Column(scale=3):
-            gr.Markdown("## 🖥 Sandbox Terminal (Debian)")
-            terminal_output = gr.Textbox(label="Terminal Output", lines=20, interactive=False, elem_classes=["terminal-box"])
-            command_input = gr.Textbox(label="Enter command", placeholder="print('Hello!')", interactive=True)
+            gr.Markdown("### 🖥 Sandbox Cloud Console")
+            terminal_output = gr.Textbox(label="", lines=20, interactive=False, elem_classes=["terminal-box"])
+            command_input = gr.Textbox(label="Command Input", placeholder="print('Hello!')", interactive=True)
             run_btn = gr.Button("Run Command")
+        
+        # Side Panel
         with gr.Column(scale=1):
-            launch_btn = gr.Button("Launch Sandbox")
-            kill_btn = gr.Button("Kill Sandbox")
-            status_btn = gr.Button("Check Status")
-            sandbox_id_box = gr.Textbox(label="Sandbox ID", interactive=False)
-            time_remaining = gr.Textbox(label="Time Remaining", interactive=False)
-            gr.Markdown("💡 Note: Running on a Debian-based HF container!")
-
-    # State variables
-    start_time_state = gr.State(value=0)
-    sandbox_backend_state = gr.State(value=None)
+            with gr.Box(elem_classes=["side-panel"]):
+                gr.Markdown("### Sandbox Controls")
+                launch_btn = gr.Button("Launch Sandbox")
+                kill_btn = gr.Button("Kill Sandbox")
+                status_btn = gr.Button("Check Status")
+                sandbox_id_box = gr.Textbox(label="Sandbox ID", interactive=False)
+                time_remaining = gr.Textbox(label="Time Remaining", interactive=False)
+                gr.Markdown("💡 Running on a Debian-based container!")
 
-    # -------------------------
     # Callbacks
-    # -------------------------
     launch_btn.click(
         launch_sandbox,
         inputs=command_input,
         outputs=[terminal_output, sandbox_id_box, time_remaining, start_time_state, sandbox_backend_state]
     )
-
     run_btn.click(
         run_command,
         inputs=[command_input, sandbox_id_box, start_time_state, sandbox_backend_state],
         outputs=[terminal_output, time_remaining]
     )
-
     kill_btn.click(
         kill_sandbox,
         inputs=[sandbox_id_box, sandbox_backend_state],
         outputs=terminal_output
     )
-
     status_btn.click(
         get_status,
         inputs=[sandbox_id_box, start_time_state, sandbox_backend_state],