Update app.py

app.py

@@ -16,6 +16,9 @@ STRANGE_ROOT = ROOT / "Strange"
 DOWN_GRAPHICS = DOWN_ROOT / "graphics"
 STRANGE_GRAPHICS = STRANGE_ROOT / "graphics"
 
+DOWN_RESULTS = DOWN_ROOT / "results"
+STRANGE_RESULTS = STRANGE_ROOT / "results"
+
 
 # ============================================================
 # Helpers
@@ -23,7 +26,6 @@ STRANGE_GRAPHICS = STRANGE_ROOT / "graphics"
 
 def _clean_pngs(folder: Path):
     folder.mkdir(parents=True, exist_ok=True)
-    # Clean recursively to handle "graphics/graphics"
     for f in folder.rglob("*.png"):
         try:
             f.unlink()
@@ -34,10 +36,32 @@ def _clean_pngs(folder: Path):
 def _collect_pngs(folder: Path):
     if not folder.exists():
         return []
-    # Collect recursively to handle nested folders
     return sorted(str(p) for p in folder.rglob("*.png"))
 
 
+def _run_capture(cmd, cwd: Path, env: dict):
+    """
+    Run a subprocess and capture stdout/stderr.
+    If it fails, raise a RuntimeError containing the logs so Gradio shows it.
+    """
+    result = subprocess.run(
+        cmd,
+        cwd=str(cwd),
+        env=env,
+        capture_output=True,
+        text=True,
+    )
+    if result.returncode != 0:
+        raise RuntimeError(
+            "Subprocess failed\n"
+            f"CMD: {' '.join(map(str, cmd))}\n"
+            f"CWD: {cwd}\n\n"
+            f"STDOUT:\n{result.stdout}\n"
+            f"STDERR:\n{result.stderr}"
+        )
+    return result.stdout, result.stderr
+
+
 # ============================================================
 # DOWN (Unified: Down + AntiDown)
 # ============================================================
@@ -47,24 +71,31 @@ def run_down_all(episodes: int):
     if not run_all.exists():
         raise RuntimeError(f"{run_all} not found (check folder name: must be 'Down/')")
 
+    DOWN_GRAPHICS.mkdir(parents=True, exist_ok=True)
+    DOWN_RESULTS.mkdir(parents=True, exist_ok=True)
+
     _clean_pngs(DOWN_GRAPHICS)
 
+    env = os.environ.copy()
+    env["TWOQUARKS_RESULTS_DIR"] = str(DOWN_RESULTS)
+    env["TWOQUARKS_GRAPHICS_DIR"] = str(DOWN_GRAPHICS)
+
     cmd = [sys.executable, str(run_all), "--episodes", str(int(episodes))]
 
-    subprocess.run(
-        cmd,
-        cwd=str(DOWN_ROOT),
-        env=os.environ.copy(),
-        check=True,
-    )
+    # capture logs so the UI shows the real failure reason if any
+    _run_capture(cmd, cwd=DOWN_ROOT, env=env)
 
     images = _collect_pngs(DOWN_GRAPHICS)
     meta = {
         "status": "completed",
         "pipeline": "DOWN + AntiDown",
-        "episodes": int(episodes),
+        "episodes_per_phase": int(episodes),
         "graphics_count": len(images),
         "graphics_dir": str(DOWN_GRAPHICS),
+        "results_dir": str(DOWN_RESULTS),
+        # canonical CSV names the dual plot expects
+        "csv_down": str(DOWN_RESULTS / "down_paradox_tabular_results.csv"),
+        "csv_antidown": str(DOWN_RESULTS / "antidown_corrupted_valley_tabular.csv"),
     }
     return meta, images
 
@@ -78,16 +109,18 @@ def run_strange_all():
     if not run_all.exists():
         raise RuntimeError(f"{run_all} not found (check folder name: must be 'Strange/')")
 
+    STRANGE_GRAPHICS.mkdir(parents=True, exist_ok=True)
+    STRANGE_RESULTS.mkdir(parents=True, exist_ok=True)
+
     _clean_pngs(STRANGE_GRAPHICS)
 
+    env = os.environ.copy()
+    env["TWOQUARKS_RESULTS_DIR"] = str(STRANGE_RESULTS)
+    env["TWOQUARKS_GRAPHICS_DIR"] = str(STRANGE_GRAPHICS)
+
     cmd = [sys.executable, str(run_all)]
 
-    subprocess.run(
-        cmd,
-        cwd=str(STRANGE_ROOT),
-        env=os.environ.copy(),
-        check=True,
-    )
+    _run_capture(cmd, cwd=STRANGE_ROOT, env=env)
 
     images = _collect_pngs(STRANGE_GRAPHICS)
     meta = {
@@ -95,6 +128,7 @@ def run_strange_all():
         "pipeline": "STRANGE + AntiStrange",
         "graphics_count": len(images),
         "graphics_dir": str(STRANGE_GRAPHICS),
+        "results_dir": str(STRANGE_RESULTS),
     }
     return meta, images