Update app.py

app.py

@@ -5,12 +5,12 @@ from pathlib import Path
 import gradio as gr
 
 # ============================================================
-# Paths
+# Paths (Linux is case-sensitive!)
 # ============================================================
 
 ROOT = Path(__file__).resolve().parent
 
-DOWN_ROOT = ROOT / "down"
+DOWN_ROOT = ROOT / "Down"
 STRANGE_ROOT = ROOT / "Strange"
 
 DOWN_GRAPHICS = DOWN_ROOT / "graphics"
@@ -22,29 +22,34 @@ STRANGE_GRAPHICS = STRANGE_ROOT / "graphics"
 # ============================================================
 
 def _clean_pngs(folder: Path):
-    if folder.exists():
-        for f in folder.glob("*.png"):
+    folder.mkdir(parents=True, exist_ok=True)
+    # Clean recursively to handle "graphics/graphics"
+    for f in folder.rglob("*.png"):
+        try:
             f.unlink()
-    else:
-        folder.mkdir(parents=True, exist_ok=True)
+        except Exception:
+            pass
 
 
 def _collect_pngs(folder: Path):
-    return sorted(str(p) for p in folder.glob("*.png"))
+    if not folder.exists():
+        return []
+    # Collect recursively to handle nested folders
+    return sorted(str(p) for p in folder.rglob("*.png"))
 
 
 # ============================================================
 # DOWN (Unified: Down + AntiDown)
 # ============================================================
 
-def run_down_all():
+def run_down_all(episodes: int):
     run_all = DOWN_ROOT / "run_all.py"
     if not run_all.exists():
-        raise RuntimeError("Down/run_all.py not found")
+        raise RuntimeError(f"{run_all} not found (check folder name: must be 'Down/')")
 
     _clean_pngs(DOWN_GRAPHICS)
 
-    cmd = [sys.executable, "run_all.py"]
+    cmd = [sys.executable, str(run_all), "--episodes", str(int(episodes))]
 
     subprocess.run(
         cmd,
@@ -54,26 +59,28 @@ def run_down_all():
     )
 
     images = _collect_pngs(DOWN_GRAPHICS)
-
     meta = {
         "status": "completed",
         "pipeline": "DOWN + AntiDown",
+        "episodes": int(episodes),
         "graphics_count": len(images),
         "graphics_dir": str(DOWN_GRAPHICS),
     }
-
     return meta, images
 
+
+# ============================================================
 # STRANGE (Unified: Strange + AntiStrange)
+# ============================================================
 
 def run_strange_all():
     run_all = STRANGE_ROOT / "run_all.py"
     if not run_all.exists():
-        raise RuntimeError("Strange/run_all.py not found")
+        raise RuntimeError(f"{run_all} not found (check folder name: must be 'Strange/')")
 
     _clean_pngs(STRANGE_GRAPHICS)
 
-    cmd = [sys.executable, "run_all.py"]
+    cmd = [sys.executable, str(run_all)]
 
     subprocess.run(
         cmd,
@@ -83,14 +90,12 @@ def run_strange_all():
     )
 
     images = _collect_pngs(STRANGE_GRAPHICS)
-
     meta = {
         "status": "completed",
         "pipeline": "STRANGE + AntiStrange",
         "graphics_count": len(images),
         "graphics_dir": str(STRANGE_GRAPHICS),
     }
-
     return meta, images
 
 
@@ -99,14 +104,10 @@ def run_strange_all():
 # ============================================================
 
 def build_ui():
-    with gr.Blocks(
-        title="TwoQuarks — QuarksLab",
-    ) as demo:
-
+    with gr.Blocks(title="TwoQuarks — QuarksLab") as demo:
         gr.Markdown(
             """
-# TwoQuarks • QuarksLab 
-
+# TwoQuarks • QuarksLab
 **Run Models. Real artifacts. Real Models.**
 """
         )
@@ -117,8 +118,9 @@ def build_ui():
         gr.Markdown("## DOWN / AntiDown")
         gr.Markdown(
             """
-The Down / AntiDown models are reinforcement learning agents 
-designed to expose and analyze failure modes under deceptive reward conditions."""
+The Down / AntiDown models are reinforcement learning agents
+designed to expose and analyze failure modes under deceptive reward conditions.
+"""
         )
 
         down_eps = gr.Slider(
@@ -130,13 +132,8 @@ designed to expose and analyze failure modes under deceptive reward conditions."
         )
 
         run_down_btn = gr.Button("Run DOWN Model")
-
         down_meta = gr.JSON(label="Run status")
-        down_gallery = gr.Gallery(
-            label="Generated plots",
-            columns=2,
-            height="auto",
-        )
+        down_gallery = gr.Gallery(label="Generated plots", columns=2, height="auto")
 
         run_down_btn.click(
             fn=run_down_all,
@@ -152,19 +149,14 @@ designed to expose and analyze failure modes under deceptive reward conditions."
         gr.Markdown("## STRANGE / AntiStrange")
         gr.Markdown(
             """
-Strange / AntiStrange are reinforcement learning agents 
-designed to study failure under hidden regime shifts and phase transitions. 
+Strange / AntiStrange are reinforcement learning agents
+designed to study failure under hidden regime shifts and phase transitions.
 """
         )
 
         run_strange_btn = gr.Button("Run STRANGE Model")
-
         strange_meta = gr.JSON(label="Run status")
-        strange_gallery = gr.Gallery(
-            label="Generated plots",
-            columns=2,
-            height="auto",
-        )
+        strange_gallery = gr.Gallery(label="Generated plots", columns=2, height="auto")
 
         run_strange_btn.click(
             fn=run_strange_all,
@@ -176,7 +168,7 @@ designed to study failure under hidden regime shifts and phase transitions.
             """
 ---
 ### Notes:
-- This lab, its models & the TwoQuarks website, 
+- This lab, its models & the TwoQuarks website,
   are developed end to end by Jaime Ledesma.
 - Gradio does not import or orchestrate internal model logic.
 - All visualizations are the result produced by the model.