Update app.py

app.py

@@ -14,18 +14,24 @@ import gradio as gr
 # -------------------------
 sandboxes = {}
 SANDBOX_TTL = 2 * 3600  # 2 hours
+MAX_RAM = 4 * 1024**3  # 4GB
 
 # -------------------------
 # Sandbox worker function
 # -------------------------
-def sandbox_worker(code: str, sandbox_id: str):
+def sandbox_worker(code: str, requirements: str, sandbox_id: str):
     user_dir = f"/tmp/sandbox_{sandbox_id}"
     os.makedirs(user_dir, exist_ok=True)
     os.chdir(user_dir)
 
+    # Save requirements.txt
+    req_file = os.path.join(user_dir, "requirements.txt")
+    with open(req_file, "w") as f:
+        f.write(requirements)
+
     # Resource limits
-    resource.setrlimit(resource.RLIMIT_AS, (4 * 1024**3, 4 * 1024**3))  # 4GB RAM
-    resource.setrlimit(resource.RLIMIT_CPU, (7200, 7200))  # 2h CPU
+    resource.setrlimit(resource.RLIMIT_AS, (MAX_RAM, MAX_RAM))
+    resource.setrlimit(resource.RLIMIT_CPU, (SANDBOX_TTL, SANDBOX_TTL))
 
     # Save user code
     code_file = os.path.join(user_dir, "user_code.py")
@@ -65,16 +71,16 @@ def cleanup_sandboxes():
                 if os.path.exists(folder):
                     shutil.rmtree(folder)
                 del sandboxes[sid]
-        time.sleep(60)  # check every minute
+        time.sleep(60)
 
 threading.Thread(target=cleanup_sandboxes, daemon=True).start()
 
 # -------------------------
 # Gradio interface functions
 # -------------------------
-def launch_sandbox(code):
+def launch_sandbox(code, requirements):
     sandbox_id = str(uuid.uuid4())[:8]
-    p = multiprocessing.Process(target=sandbox_worker, args=(code, sandbox_id))
+    p = multiprocessing.Process(target=sandbox_worker, args=(code, requirements, sandbox_id))
     p.start()
     sandboxes[sandbox_id] = {"pid": p.pid, "start_time": time.time()}
     return f"Sandbox launched! ID: {sandbox_id}", sandbox_id
@@ -113,9 +119,10 @@ def status_sandbox(sandbox_id):
 # Gradio UI
 # -------------------------
 with gr.Blocks() as demo:
-    gr.Markdown("# Sandbox Backend with Auto-Cleanup")
+    gr.Markdown("# Sandbox Backend with requirements.txt support")
 
     code_input = gr.Textbox(label="Python Code", lines=15, placeholder="Write your code here...")
+    req_input = gr.Textbox(label="requirements.txt", lines=5, placeholder="Add dependencies here...")
     launch_btn = gr.Button("Launch Sandbox")
     sandbox_id_output = gr.Textbox(label="Sandbox ID")
 
@@ -132,7 +139,7 @@ with gr.Blocks() as demo:
 
     launch_btn.click(
         launch_sandbox,
-        inputs=code_input,
+        inputs=[code_input, req_input],
         outputs=[sandbox_id_output, sandbox_id_storage]
     )
     logs_btn.click(