Upload app.py

app.py

@@ -105,6 +105,19 @@ else:
     print("   2. Place this notebook inside the aegis-reason/ directory, OR")
     print("   3. Clone: !git clone <repo-url> aegis-reason")
 DDIR = 'aegis_dataset'
+# Robust search for aegis_dataset directory (handles HF Spaces path variations)
+if not os.path.isdir(DDIR):
+    _script_dir = os.path.dirname(os.path.abspath(__file__)) if '__file__' in dir() else os.getcwd()
+    _ddir_candidates = [
+        os.path.join(_script_dir, 'aegis_dataset'),
+        os.path.join(os.getcwd(), 'aegis_dataset'),
+        '/home/user/app/aegis_dataset',
+        os.path.join(REPO, 'aegis_dataset') if REPO else None,
+    ]
+    for _dc in _ddir_candidates:
+        if _dc and os.path.isdir(_dc):
+            DDIR = _dc
+            break
 
 # ═══════════════════════════ DESIGN SYSTEM ═══════════════════════════
 # (rcParams already set above — dark)
@@ -8899,6 +8912,18 @@ def impact_scenario(disaster_name, rescue_boost=0.0):
 
 
 def explore():
+    global DDIR
+    # Fallback search if DDIR doesn't resolve at runtime
+    if not os.path.isdir(DDIR):
+        _script_dir = os.path.dirname(os.path.abspath(__file__)) if '__file__' in dir() else os.getcwd()
+        for _candidate in [
+            os.path.join(_script_dir, 'aegis_dataset'),
+            os.path.join(os.getcwd(), 'aegis_dataset'),
+            '/home/user/app/aegis_dataset',
+        ]:
+            if os.path.isdir(_candidate):
+                DDIR = _candidate
+                break
     if not os.path.isdir(DDIR): return '❌',None,EMPTY_3D_HTML
     lines_e=[]
     for sub in ['topdown','egocentric','chains']:
@@ -8938,6 +8963,17 @@ def explore():
 
 def view_chain(idx):
     sft=os.path.join(DDIR,'sft_data.jsonl')
+    # Fallback search if primary DDIR doesn't contain sft_data.jsonl
+    if not os.path.exists(sft):
+        _script_dir = os.path.dirname(os.path.abspath(__file__)) if '__file__' in dir() else os.getcwd()
+        for _candidate in [
+            os.path.join(_script_dir, 'aegis_dataset', 'sft_data.jsonl'),
+            os.path.join(os.getcwd(), 'aegis_dataset', 'sft_data.jsonl'),
+            '/home/user/app/aegis_dataset/sft_data.jsonl',
+        ]:
+            if os.path.exists(_candidate):
+                sft = _candidate
+                break
     if not os.path.exists(sft): return 'No data.',None
     recs=[json.loads(l) for l in open(sft)]; idx=max(0,min(int(idx),len(recs)-1)); r=recs[idx]
     ch=r['conversations'][2]['content']
@@ -8965,17 +9001,24 @@ def view_chain(idx):
     return f"ID: {r['id']}\n{len(ch)} chars\n\n{ch}",img
 
 def run_validation():
+    # Resolve the repo root for all relative path lookups
+    _root = REPO or os.path.dirname(os.path.abspath(__file__)) if '__file__' in dir() else os.getcwd()
+    if not _root or not os.path.isdir(_root):
+        _root = os.getcwd()
     out=['═══ E2E ═══']
-    if os.path.exists('scripts/aegis_e2e_test.py'):
-        r=subprocess.run([sys.executable,'scripts/aegis_e2e_test.py'],capture_output=True,text=True,timeout=300); out.append(r.stdout[:2000])
+    e2e_path = os.path.join(_root, 'scripts', 'aegis_e2e_test.py')
+    if os.path.exists(e2e_path):
+        r=subprocess.run([sys.executable, e2e_path],capture_output=True,text=True,timeout=300, cwd=_root); out.append(r.stdout[:2000])
     else: out.append('❌ Not found')
     out.append('\n═══ AUDIT ═══'); iss=0
-    for pyf in glob.glob('*.py')+glob.glob('scripts/*.py'):
+    for pyf in glob.glob(os.path.join(_root, '*.py'))+glob.glob(os.path.join(_root, 'scripts', '*.py')):
         for i,ln in enumerate(open(pyf).readlines(),1):
-            if '/home/claude/' in ln and not ln.strip().startswith('#'): out.append(f'  ⚠️ {pyf}:{i}'); iss+=1
+            if '/home/claude/' in ln and not ln.strip().startswith('#'):
+                out.append(f'  ⚠️ {os.path.relpath(pyf, _root)}:{i}'); iss+=1
     if iss==0: out.append('  ✅ Clean')
     for f in ['README.md','configs/aegis_sft.toml','configs/aegis_grpo.toml','LICENSE','requirements.txt']:
-        out.append(f'  {"✅" if os.path.exists(f) else "❌"} {f}')
+        fp = os.path.join(_root, f)
+        out.append(f'  {"✅" if os.path.exists(fp) else "❌"} {f}')
     return '\n'.join(out)
 
 # ─── Interactive weight explorer ───
@@ -12759,15 +12802,15 @@ Physical AI agents must **ground decisions in domain knowledge**. CRAG closes th
     with gr.Tab('🔬 Ablation'):
         gr.Markdown("**Conditions**: A (Oracle) → B (Partial: spatial+decision only) → C (Degraded: shifted coords) → D (Vague)")
         with gr.Accordion("🔄 Interactive 3D: Ablation Conditions (rotate me!)", open=True):
-            gr.HTML(value=fig3d_html(plotly_ablation_3d()), label='3D Ablation')
-        ab_b=gr.Button('Generate 3D Ablation',variant='primary'); ab_p=gr.HTML(value=EMPTY_3D_HTML, label='3D Interactive Ablation')
+            ab_p=gr.HTML(value=EMPTY_3D_HTML, label='3D Interactive Ablation')
+        ab_b=gr.Button('Generate 3D Ablation',variant='primary')
         ab_b.click(fn=_safe(show_ablation, lambda e: EMPTY_3D_HTML),outputs=ab_p)
 
     # ──────────────── TAB 5: CALCULUS ────────────────
     with gr.Tab('📏 Calibration'):
         with gr.Accordion("🔄 Interactive 3D: Calibration (rotate me!)", open=True):
-            gr.HTML(value=fig3d_html(plotly_calibration_3d()), label='3D Calibration')
-        cr_b=gr.Button('Regenerate 3D Calibration',variant='primary'); cr_p=gr.HTML(value=EMPTY_3D_HTML, label='3D Interactive')
+            cr_p=gr.HTML(value=EMPTY_3D_HTML, label='3D Interactive')
+        cr_b=gr.Button('Generate 3D Calibration',variant='primary')
         cr_b.click(fn=_safe(show_calibration, lambda e: EMPTY_3D_HTML),outputs=cr_p)
 
     # ──────────────── TAB 10: DATASET ────────────────
@@ -12891,7 +12934,7 @@ Benchmarks: Baseten 48% P95↓, Moonshot AI 10x, DeepSeek-R1 30x on GB200
                 label='⚡ Inference Backend'
             )
         with gr.Row():
-            accel_scenario = gr.Dropdown(
+            accel_scenario = gr.Radio(
                 choices=[v['domain'] for v in CR2_HEALTHCARE_SCENARIOS.values()],
                 value=list(CR2_HEALTHCARE_SCENARIOS.values())[0]['domain'],
                 label='🏥 Healthcare Scenario'