Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from gradio_client import Client
-import time
 import random
+import time
 
 # -------------------------
 # Backend HF clients
@@ -15,7 +15,7 @@ backend_engines = [
 # Helper functions
 # -------------------------
 def pick_engine():
-    # Simple load balancing (random for now)
+    # Random backend selection (simple load balancing)
     return random.choice(backend_engines)
 
 def launch_sandbox(code):
@@ -25,32 +25,57 @@ def launch_sandbox(code):
     return f"Sandbox launched! ID: {sandbox_id}", sandbox_id, "120:00"  # 2h TTL
 
 def run_command(cmd, sandbox_id):
+    if not sandbox_id:
+        return "No sandbox ID. Launch a sandbox first!"
     engine = pick_engine()
-    # Here we assume the backend runs the code immediately in the sandbox
-    # For demo, we send each command as a mini script
-    code = cmd
-    engine.predict(code=code, api_name="/launch_sandbox")  # you could create a /run_command endpoint
-    logs = engine.predict(api_name="/fetch_logs")
-    return logs
+    # For simplicity, run command as a mini Python script
+    try:
+        engine.predict(code=cmd, api_name="/launch_sandbox")  # ideally /run_command endpoint
+        logs = engine.predict(api_name="/fetch_logs")
+        return logs
+    except Exception as e:
+        return f"Error: {e}"
 
 def kill_sandbox(sandbox_id):
+    if not sandbox_id:
+        return "No sandbox ID to kill."
     engine = pick_engine()
-    result = engine.predict(api_name="/kill_sandbox")
-    return result
+    try:
+        result = engine.predict(api_name="/kill_sandbox")
+        return result
+    except Exception as e:
+        return f"Error: {e}"
 
 def get_status(sandbox_id):
+    if not sandbox_id:
+        return "No sandbox ID."
     engine = pick_engine()
-    return engine.predict(api_name="/status_sandbox")
+    try:
+        return engine.predict(api_name="/status_sandbox")
+    except Exception as e:
+        return f"Error: {e}"
 
 # -------------------------
 # Gradio UI
 # -------------------------
 with gr.Blocks() as demo:
+    # Global CSS for terminal style (white-on-black)
+    gr.HTML("""
+    <style>
+    .terminal-box textarea {
+        background-color: #000000;
+        color: #ffffff;
+        font-family: monospace;
+        font-size: 14px;
+    }
+    </style>
+    """)
+
     with gr.Row():
         with gr.Column(scale=3):
-            gr.Markdown("## 🖥 Sandbox Cloud")
+            gr.Markdown("## 🖥 Sandbox Cloud Console")
             terminal_output = gr.Textbox(label="Terminal Output", lines=20, interactive=False, elem_classes=["terminal-box"])
-            command_input = gr.Textbox(label="Enter command", placeholder="ls -la", interactive=True)
+            command_input = gr.Textbox(label="Enter command", placeholder="print('Hello!')", interactive=True)
             run_btn = gr.Button("Run Command")
         with gr.Column(scale=1):
             launch_btn = gr.Button("Launch Sandbox")
@@ -58,7 +83,7 @@ with gr.Blocks() as demo:
             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 container!")
+            gr.Markdown("💡 Note: Running on a Debian-based HF container!")
 
     # Callbacks
     launch_btn.click(launch_sandbox, inputs=command_input, outputs=[terminal_output, sandbox_id_box, time_remaining])
@@ -66,14 +91,4 @@ with gr.Blocks() as demo:
     kill_btn.click(kill_sandbox, inputs=sandbox_id_box, outputs=terminal_output)
     status_btn.click(get_status, inputs=sandbox_id_box, outputs=terminal_output)
 
-# CSS for terminal look
-css = """
-.terminal-box textarea {
-    background-color: #1e1e1e;
-    color: #00ff00;
-    font-family: monospace;
-    font-size: 14px;
-}
-"""
-
-demo.launch(css=css)
+demo.launch()