Update app.py

app.py

@@ -1,47 +1,63 @@
 import gradio as gr
-from e2b import Sandbox
+import subprocess
 import os
 
-# Session persistence across functions
-sandbox = None
+# Where uploads land inside the container
+WORKDIR = "/workspace"
+os.makedirs(WORKDIR, exist_ok=True)
 
-def start_sandbox():
-    global sandbox
-    sandbox = Sandbox(template="base")  # No api_key needed
-    return sandbox.url or "Failed to launch sandbox"
+# Embed noVNC iframe HTML
+NO_VNC_IFRAME = """
+<iframe src="http://localhost:6080/vnc.html?autoconnect=true&resize=remote"
+        style="width:100%; height:600px; border:none;">
+</iframe>
+"""
 
-def run_shell(cmd):
-    global sandbox
-    if not sandbox:
-        return "Sandbox not launched"
-    output = sandbox.run_shell(cmd)
-    return output.stdout if output.stdout else output.stderr
+def start_sandbox():
+    # The noVNC server is already running; just return iframe
+    return NO_VNC_IFRAME
+
+def run_shell(cmd: str):
+    # Run command inside container; keep DISPLAY so GUI apps can launch
+    result = subprocess.run(
+        cmd,
+        shell=True,
+        capture_output=True,
+        text=True,
+        env={**os.environ, "DISPLAY": ":1"}
+    )
+    return result.stdout + result.stderr
 
 def upload_file(file):
-    global sandbox
-    if not sandbox:
-        return "Sandbox not launched"
-    file_path = sandbox.upload_file(file.name, open(file.name, "rb"))
-    return f"Uploaded to sandbox: {file_path}"
-
-with gr.Blocks() as demo:
-    gr.Markdown("# 🖥️ Sandbox Desktop with Shell & Upload")
-    
-    with gr.Row():
-        launch_btn = gr.Button("Start Sandbox")
-        url_output = gr.Textbox(label="Desktop URL", interactive=False)
-
-    launch_btn.click(start_sandbox, outputs=url_output)
-
-    shell_input = gr.Textbox(label="Shell Command")
-    shell_btn = gr.Button("Run Command")
-    shell_output = gr.Textbox(label="Command Output", lines=6)
-
-    shell_btn.click(run_shell, inputs=shell_input, outputs=shell_output)
-
-    file_input = gr.File(label="Upload File to Sandbox")
-    file_output = gr.Textbox(label="Upload Status")
-
-    file_input.change(upload_file, inputs=file_input, outputs=file_output)
+    # Save uploaded file into WORKDIR
+    dest = os.path.join(WORKDIR, os.path.basename(file.name))
+    with open(dest, "wb") as f:
+        f.write(file.read())
+    return f"Saved to sandbox: {dest}"
+
+with gr.Blocks(css="@import url('https://cdn.simplecss.org/simple.min.css');") as demo:
+    gr.Markdown("# 🔒 Offline Sandbox Desktop")
+
+    with gr.Tab("Desktop"):
+        btn = gr.Button("Show Desktop")
+        desktop_view = gr.HTML()
+        btn.click(start_sandbox, outputs=desktop_view)
+
+    with gr.Tab("Shell"):
+        cmd = gr.Textbox(label="Shell Command", placeholder="e.g. ls -al /workspace")
+        run = gr.Button("Run")
+        out = gr.Textbox(label="Output", lines=10)
+        run.click(run_shell, inputs=cmd, outputs=out)
+
+    with gr.Tab("Upload"):
+        uploader = gr.File(label="Upload to Sandbox")
+        status = gr.Textbox(label="Upload Status")
+        uploader.change(upload_file, inputs=uploader, outputs=status)
+
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        ssr_mode=False
+    )
 
-demo.launch()