Update main.py

main.py

@@ -778,6 +778,73 @@ def get_exam_questions(paper_id):
                 return jsonify({"paperId": paper_id, "meta": paper.get("meta"), "questions": paper.get("questions", [])})
     return jsonify({"error": "Not found"}), 404
 
+
+@app.route('/v1/reset', methods=['GET', 'POST'])
+def reset_and_rebuild():
+    """
+    One-time cache reset endpoint.
+    Wipes data_api/syllabi and data_api/vectors from Firebase,
+    then triggers a full fresh rebuild from the PDF files.
+
+    Protected by REBUILD_SECRET env var. If that var is not set,
+    protected by a hardcoded one-time token so it cannot be called accidentally.
+
+    Call from browser:  GET /v1/reset?token=marka-reset-2026
+    Or as POST with Authorization: Bearer marka-reset-2026
+    """
+    # Token check — use REBUILD_SECRET if set, otherwise the hardcoded token
+    expected = os.environ.get("REBUILD_SECRET", "marka-reset-2026")
+    token = (
+        request.args.get("token")
+        or request.headers.get("Authorization", "").replace("Bearer ", "").strip()
+    )
+    if token != expected:
+        return jsonify({
+            "error": "Unauthorized",
+            "hint":  "Pass ?token=<REBUILD_SECRET> in the URL"
+        }), 401
+
+    def _reset_bg():
+        global SYLLABUS_MAP, VECTOR_DB, VECTOR_MATRIX, EXAM_MAP
+        logger.info("=== CACHE RESET STARTED ===")
+
+        # 1. Wipe Firebase cache nodes
+        if FIREBASE_AVAILABLE:
+            try:
+                firebase_db_ref.child("data_api/syllabi").delete()
+                logger.info("  ✓ Deleted data_api/syllabi from Firebase")
+            except Exception as e:
+                logger.error(f"  ✗ Failed to delete syllabi: {e}")
+            try:
+                firebase_db_ref.child("data_api/vectors").delete()
+                logger.info("  ✓ Deleted data_api/vectors from Firebase")
+            except Exception as e:
+                logger.error(f"  ✗ Failed to delete vectors: {e}")
+        else:
+            logger.warning("  Firebase not available — skipping Firebase wipe")
+
+        # 2. Clear in-memory state
+        SYLLABUS_MAP  = {}
+        VECTOR_DB     = []
+        VECTOR_MATRIX = None
+        EXAM_MAP      = {}
+        logger.info("  ✓ In-memory state cleared")
+
+        # 3. Full rebuild from PDFs with vision classifier
+        logger.info("  Starting full rebuild with vision-based parser ...")
+        build_index()
+        logger.info("=== CACHE RESET COMPLETE ===")
+
+    threading.Thread(target=_reset_bg, daemon=True).start()
+
+    return jsonify({
+        "status":  "reset started",
+        "message": "Firebase cache wiped. Full rebuild running in background. "
+                   "Check /health in 3-5 minutes to see subject and vector counts update. "
+                   "All 24 PDFs will be re-parsed with the vision classifier.",
+        "watch":   "/health"
+    }), 202
+
 @app.route('/v1/rebuild', methods=['POST'])
 def trigger_rebuild():
     secret = os.environ.get("REBUILD_SECRET", "")