Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from gradio_client import Client
 import random
 import time
+import threading
 
 # -------------------------
 # Backend HF clients
@@ -11,30 +12,47 @@ backend_engines = [
     Client("ChocoLaboratory/SANDBOXBACKEND2")
 ]
 
+SANDBOX_TTL_SEC = 2 * 3600  # 2 hours
+
 # -------------------------
 # Helper functions
 # -------------------------
 def pick_engine():
-    # Random backend selection (simple load balancing)
     return random.choice(backend_engines)
 
 def launch_sandbox(code):
     engine = pick_engine()
     result = engine.predict(code=code, api_name="/launch_sandbox")
     sandbox_id = result if isinstance(result, str) else result.get("sandbox_id", "unknown")
-    return f"Sandbox launched! ID: {sandbox_id}", sandbox_id, "120:00"  # 2h TTL
+    start_time = time.time()
+    return f"Sandbox launched! ID: {sandbox_id}\n", sandbox_id, SANDBOX_TTL_SEC, start_time
 
-def run_command(cmd, sandbox_id):
+def fetch_logs_and_timer(sandbox_id, start_time):
     if not sandbox_id:
-        return "No sandbox ID. Launch a sandbox first!"
+        return "No sandbox ID.", "0:00"
     engine = pick_engine()
-    # 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}"
+        logs = f"Error fetching logs: {e}"
+    elapsed = int(time.time() - start_time)
+    remaining = max(0, SANDBOX_TTL_SEC - elapsed)
+    minutes, seconds = divmod(remaining, 60)
+    time_str = f"{minutes:02d}:{seconds:02d}"
+    return logs, time_str
+
+def run_command(cmd, sandbox_id, start_time):
+    if not sandbox_id:
+        return "No sandbox ID. Launch a sandbox first!", "0:00"
+    engine = pick_engine()
+    try:
+        # send code as mini script
+        engine.predict(code=cmd, api_name="/launch_sandbox")  # ideally /run_command endpoint
+        # fetch logs after running
+        logs, remaining = fetch_logs_and_timer(sandbox_id, start_time)
+        return logs, remaining
+    except Exception as e:
+        return f"Error: {e}", "0:00"
 
 def kill_sandbox(sandbox_id):
     if not sandbox_id:
@@ -46,20 +64,24 @@ def kill_sandbox(sandbox_id):
     except Exception as e:
         return f"Error: {e}"
 
-def get_status(sandbox_id):
+def get_status(sandbox_id, start_time):
     if not sandbox_id:
-        return "No sandbox ID."
+        return "No sandbox ID.", "0:00"
     engine = pick_engine()
     try:
-        return engine.predict(api_name="/status_sandbox")
+        status = engine.predict(api_name="/status_sandbox")
+        elapsed = int(time.time() - start_time)
+        remaining = max(0, SANDBOX_TTL_SEC - elapsed)
+        minutes, seconds = divmod(remaining, 60)
+        time_str = f"{minutes:02d}:{seconds:02d}"
+        return status, time_str
     except Exception as e:
-        return f"Error: {e}"
+        return f"Error: {e}", "0:00"
 
 # -------------------------
 # Gradio UI
 # -------------------------
 with gr.Blocks() as demo:
-    # Global CSS for terminal style (white-on-black)
     gr.HTML("""
     <style>
     .terminal-box textarea {
@@ -83,12 +105,15 @@ 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 HF container!")
+            gr.Markdown("💡 Note: Running on a Debian-based container!")
+
+    # State for start time
+    start_time_state = gr.State(value=0)
 
     # Callbacks
-    launch_btn.click(launch_sandbox, inputs=command_input, outputs=[terminal_output, sandbox_id_box, time_remaining])
-    run_btn.click(run_command, inputs=[command_input, sandbox_id_box], outputs=terminal_output)
+    launch_btn.click(launch_sandbox, inputs=command_input, outputs=[terminal_output, sandbox_id_box, time_remaining, start_time_state])
+    run_btn.click(run_command, inputs=[command_input, sandbox_id_box, start_time_state], outputs=[terminal_output, time_remaining])
+    status_btn.click(get_status, inputs=[sandbox_id_box, start_time_state], outputs=[terminal_output, time_remaining])
     kill_btn.click(kill_sandbox, inputs=sandbox_id_box, outputs=terminal_output)
-    status_btn.click(get_status, inputs=sandbox_id_box, outputs=terminal_output)
 
 demo.launch()