update app

app.py

@@ -5,6 +5,7 @@ import re
 import threading
 from concurrent.futures import ThreadPoolExecutor, TimeoutError as FuturesTimeoutError
 from typing import Any, Dict, List, Optional, Tuple
+import requests
 
 import gradio as gr
 import logging
@@ -32,6 +33,8 @@ TRUNCATE_LIMIT = 400
 PLAN_MAX_TOKENS = 16384
 CODE_MAX_TOKENS = 8192
 DATASET_PREVIEW_TIMEOUT = 60  # seconds
+RUN_CODE_URL = os.getenv("DATACHEF_RUN_URL", "http://127.0.0.1:7001/run_code")
+RUN_CODE_TIMEOUT = int(os.getenv("DATACHEF_RUN_TIMEOUT", "300"))
 _DATASET_PREVIEW_CACHE: Dict[str, Dict] = {}
 # Preset library
 PRESETS = {
@@ -1119,6 +1122,40 @@ def _run_code(
     return _generate_code_from_plan(datasets, plan_text.strip())
 
 
+def _execute_generated_code_remote(code_text: str, timeout: int = RUN_CODE_TIMEOUT) -> str:
+    """Send generated code to remote runner via proxy and return aggregated output."""
+    if not code_text or not code_text.strip():
+        return "No code to run. Please generate code first."
+    try:
+        resp = requests.post(
+            RUN_CODE_URL,
+            json={"code": code_text, "timeout": timeout},
+            timeout=timeout + 10,
+        )
+        resp.raise_for_status()
+        data = resp.json() if resp.content else {}
+    except Exception as e:
+        return f"Run error: {e}"
+
+    status = data.get("status", "unknown")
+    retcode = data.get("returncode", "n/a")
+    summary = data.get("summary", "")
+    stdout = data.get("stdout", "")
+    stderr = data.get("stderr", "")
+    duration = data.get("duration_sec")
+
+    lines = [f"Status: {status} (returncode={retcode})"]
+    if duration is not None:
+        lines.append(f"Duration: {duration:.2f}s")
+    if summary:
+        lines.append(f"Summary: {summary}")
+    if stdout:
+        lines.append("\n--- stdout ---\n" + stdout)
+    if stderr:
+        lines.append("\n--- stderr ---\n" + stderr)
+    return "\n".join(lines).strip()
+
+
 def _run_plan_and_code(
     task_description: str,
     benchmark_name: str,
@@ -1492,6 +1529,14 @@ def create_demo() -> gr.Blocks:
                             interactive=False, 
                             lines=28
                         )
+                        run_code_btn = gr.Button("▶️ Run Code", variant="secondary", size="sm")
+                        run_output = gr.Code(
+                            value="",
+                            language="text",
+                            label=None,
+                            interactive=False,
+                            lines=16,
+                        )
                         
         # --- Logic Binding ---
         preset_dropdown.change(
@@ -1510,6 +1555,13 @@ def create_demo() -> gr.Blocks:
             show_progress="hidden" 
         )
         
+        run_code_btn.click(
+            _execute_generated_code_remote,
+            inputs=[code_out],
+            outputs=[run_output],
+            show_progress="hidden",
+        )
+        
         load_model_btn.click(_warmup_model, outputs=[]).then(get_status_text, outputs=[status_html])
         demo.load(_warmup_model, outputs=[]).then(get_status_text, outputs=[status_html])