removed weight endpts

Dockerfile

@@ -66,8 +66,10 @@ HEALTHCHECK --interval=15s --timeout=5s --start-period=30s --retries=3 \
     CMD curl -f http://localhost:${PORT}/health || exit 1
 
 # ── Add non-root user ───────────────────────────────────────────────
+# Save a pristine copy of repo weights so cached HF weights can be overridden
+RUN cp -r /app/weights /app/weights_pristine 2>/dev/null || true
 RUN useradd -m -u 1000 appuser \
-    && mkdir -p /app/weights /app/results \
+    && mkdir -p /app/weights /app/results /app/weights_pristine \
     && chown -R appuser:appuser /app
 USER appuser
 # ── Default command: start the FastAPI server ─────────────────────────────────

dashboard.html

@@ -296,6 +296,12 @@
             <button class="btn btn-primary" onclick="startTraining()" id="trainBtn">▶ Start Training</button>
         </div>
         <div class="terminal" id="trainTerminal">Ready to train. Run PPO to update weights.</div>
+        <div style="margin-top:12px; display:flex; gap:8px; align-items:center; flex-wrap:wrap;">
+            <span style="font-size:12px; color:var(--text-muted);">📥 Download Weights:</span>
+            <button class="btn" onclick="downloadWeights('easy')" style="font-size:11px; padding:4px 10px;">🟢 Easy</button>
+            <button class="btn" onclick="downloadWeights('medium')" style="font-size:11px; padding:4px 10px;">🟡 Medium</button>
+            <button class="btn" onclick="downloadWeights('hard')" style="font-size:11px; padding:4px 10px;">🔴 Hard</button>
+        </div>
     </div>
 
     <div class="testing-container active" id="testingView">
@@ -858,6 +864,20 @@ async function pollTraining() {
     } catch(e) {}
 }
 
+// ═══════════════════════════════════════════════════════════════════
+// Weight Downloads
+// ═══════════════════════════════════════════════════════════════════
+function downloadWeights(taskId) {
+    const url = `${SERVER}/agent/weights/${taskId}`;
+    const a = document.createElement('a');
+    a.href = url;
+    a.download = `ppo_${taskId}.json`;
+    document.body.appendChild(a);
+    a.click();
+    document.body.removeChild(a);
+    showToast(`Downloading ${taskId} weights…`, 'info');
+}
+
 // ═══════════════════════════════════════════════════════════════════
 // Init
 // ═══════════════════════════════════════════════════════════════════

src/adaptive_alert_triage/server.py

@@ -263,10 +263,37 @@ async def _episode_loop() -> None:
 
 # ── Startup / shutdown ────────────────────────────────────────────────────────
 
+def _restore_pristine_weights():
+    """
+    On HF Spaces, the filesystem cache persists across rebuilds.
+    Old trained weights survive and override repo weights.
+    Fix: copy the pristine repo weights (saved during Docker build)
+    back into the working weights/ directory on every startup.
+    """
+    import shutil
+    pristine_dir = os.path.join(_project_root if _project_root else os.getcwd(), "weights_pristine")
+    weights_dir  = os.path.join(_project_root if _project_root else os.getcwd(), "weights")
+
+    if not os.path.exists(pristine_dir):
+        print("   [STARTUP] No pristine weights found, skipping restore.")
+        return
+
+    os.makedirs(weights_dir, exist_ok=True)
+    for f in os.listdir(pristine_dir):
+        if f.startswith("ppo_") and f.endswith(".json"):
+            src = os.path.join(pristine_dir, f)
+            dst = os.path.join(weights_dir, f)
+            shutil.copy2(src, dst)
+            print(f"   [STARTUP] Restored pristine weights: {f}")
+
+
 @app.on_event("startup")
 async def startup():
     global env, _loop_task
 
+    # Restore repo-committed weights, overriding any stale HF cache
+    _restore_pristine_weights()
+
     env = AdaptiveAlertTriageEnv(task_id="hard")
     env.real_alerts_queue = deque(maxlen=50)
     env.reset()   # ← FIX 1: immediately populate env.alerts
@@ -536,12 +563,11 @@ async def recommend():
 
 @app.get("/agent/weights/{task_id}")
 async def download_weights(task_id: str):
-    """Allow users to manually download their trained weights."""
+    """Download trained weights for a task."""
+    from fastapi import HTTPException
     path = os.path.join(_project_root if _project_root else os.getcwd(), "weights", f"ppo_{task_id}.json")
     if not os.path.exists(path):
-        from fastapi import HTTPException
         raise HTTPException(status_code=404, detail=f"No trained weights found for {task_id}")
-    from fastapi.responses import FileResponse
     return FileResponse(path, media_type='application/json', filename=f"ppo_{task_id}.json")
 # ── WebSocket ─────────────────────────────────────────────────────────────────