feat: implement admin endpoint and UI for database and vector store reset

backend/main.py

@@ -9,7 +9,7 @@ from qdrant_client.models import Distance, VectorParams, PayloadSchemaType
 
 from src.config import get_settings
 from src.models import JobDescription, Candidate, MatchResult, Session
-from src.routers import jds, candidates, matching, sessions
+from src.routers import jds, candidates, matching, sessions, admin
 
 logger = logging.getLogger(__name__)
 settings = get_settings()
@@ -81,6 +81,7 @@ app.include_router(sessions.router, prefix="/api/sessions", tags=["Sessions"])
 app.include_router(jds.router, prefix="/api/jds", tags=["Job Descriptions"])
 app.include_router(candidates.router, prefix="/api/candidates", tags=["Candidates"])
 app.include_router(matching.router, prefix="/api/match", tags=["Matching"])
+app.include_router(admin.router, prefix="/api/admin", tags=["Admin"])
 
 
 @app.get("/health")

backend/src/database.py

@@ -48,7 +48,12 @@ engine = create_async_engine(
     },
     execution_options={"prepared_statement_cache_size": 0}
 )
-AsyncSessionLocal = async_sessionmaker(engine, expire_on_commit=False)
+from sqlalchemy.orm import sessionmaker
+AsyncSessionLocal = sessionmaker(
+    bind=engine,
+    class_=AsyncSession,
+    expire_on_commit=False
+)
 
 
 class Base(DeclarativeBase):

backend/src/routers/admin.py

@@ -0,0 +1,76 @@
+import os
+from fastapi import APIRouter, HTTPException
+from sqlalchemy import text
+from qdrant_client import QdrantClient
+from qdrant_client.models import Distance, VectorParams, PayloadSchemaType
+from alembic.config import Config
+from alembic import command
+from fastapi.concurrency import run_in_threadpool
+
+from ..database import engine
+from ..config import get_settings
+
+router = APIRouter()
+
+@router.post("/reset-db")
+async def reset_db(password: str = None):
+    if password != "ketan@D048":
+        raise HTTPException(status_code=403, detail="Invalid reset password.")
+    
+    settings = get_settings()
+    
+    try:
+        # 1. Wipe Postgres natively (drop schema and recreate)
+        async with engine.begin() as conn:
+            # We use CASCADE to drop everything including tables, types, etc.
+            await conn.execute(text('DROP SCHEMA public CASCADE'))
+            await conn.execute(text('CREATE SCHEMA public'))
+            await conn.execute(text('GRANT ALL ON SCHEMA public TO public'))
+        
+        # 2. Wipe Qdrant Collection
+        q = QdrantClient(url=settings.qdrant_url, api_key=settings.qdrant_api_key)
+        try:
+            q.delete_collection(settings.collection_name)
+        except Exception:
+            # OK if collection didn't exist
+            pass
+            
+        q.create_collection(
+            collection_name=settings.collection_name, 
+            vectors_config=VectorParams(size=settings.vector_size, distance=Distance.COSINE)
+        )
+        
+        # Reinject indices required by the pipeline
+        q.create_payload_index(
+            collection_name=settings.collection_name,
+            field_name="session_id",
+            field_schema=PayloadSchemaType.KEYWORD
+        )
+        q.create_payload_index(
+            collection_name=settings.collection_name,
+            field_name="years_of_experience",
+            field_schema=PayloadSchemaType.FLOAT
+        )
+
+        # 3. Run Alembic Migrations to restore schema
+        # We assume the API is running from the 'backend' directory where alembic.ini resides.
+        # If not, we attempt to find it relative to this file.
+        base_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+        ini_path = os.path.join(base_path, "alembic.ini")
+        
+        if not os.path.exists(ini_path):
+            # Fallback for different envs
+            ini_path = "alembic.ini"
+
+        def _run_migrations():
+            alembic_cfg = Config(ini_path)
+            # Ensure the script location is absolute to avoid confusion
+            alembic_cfg.set_main_option("script_location", os.path.join(base_path, "alembic"))
+            command.upgrade(alembic_cfg, "head")
+
+        await run_in_threadpool(_run_migrations)
+
+        return {"status": "success", "message": "Database and Vector Store have been reset and migrations reapplied."}
+    except Exception as e:
+        print(f"RESET ERROR: {e}")
+        raise HTTPException(status_code=500, detail=str(e))

backend/src/routers/matching.py

@@ -83,12 +83,16 @@ async def trigger_match(
     shortlist = await stage1_retrieve(jd_dict, db, qdrant, session_id=sid_str)
     final_ranked = await stage2_rerank(jd_dict, shortlist)
 
-    await db.execute(
-        delete(MatchResult).where(
-            MatchResult.jd_id == jd_id,
-            MatchResult.session_id == session_id if session_id else MatchResult.session_id.is_(None),
+    try:
+        await db.execute(
+            delete(MatchResult).where(
+                MatchResult.jd_id == jd_id,
+                MatchResult.session_id == session_id if session_id else MatchResult.session_id.is_(None),
+            )
         )
-    )
+    except Exception:
+        await db.rollback()
+        raise
 
     from ..workers.explain import generate_top_explanations
 

frontend/src/app/layout.tsx

@@ -16,7 +16,10 @@ export default function RootLayout({ children }: { children: React.ReactNode })
             <Link href="/" className="text-base font-bold tracking-tight bg-gradient-to-r from-[var(--color-brand-light)] to-purple-400 bg-clip-text text-transparent select-none">
               ⚡ TalentPulse
             </Link>
-            <div className="flex items-center gap-1">
+            <div className="flex items-center gap-2">
+              <Link href="/reset" className="px-3 py-1.5 rounded-lg text-xs font-medium text-slate-500 hover:text-red-400 transition-colors">
+                Reset
+              </Link>
               <Link href="/pipeline" className="ml-2 px-3 py-1.5 rounded-lg text-sm font-semibold text-[var(--color-brand-light)] bg-[var(--color-brand-dim)] border border-[var(--color-brand-glow)] hover:bg-[var(--color-brand)] hover:text-white transition-all">
                 ⚡ Auto Pipeline
               </Link>

frontend/src/app/reset/page.tsx

@@ -0,0 +1,137 @@
+"use client";
+
+import { useState } from "react";
+import { useRouter } from "next/navigation";
+
+export default function ResetPage() {
+  const [loading, setLoading] = useState(false);
+  const [result, setResult] = useState<{ status: string; message: string } | null>(null);
+  const [error, setError] = useState<string | null>(null);
+  const [password, setPassword] = useState("");
+  const router = useRouter();
+
+  const handleReset = async () => {
+    if (!password) {
+      setError("Please enter the system password.");
+      return;
+    }
+
+    if (!confirm("☢️ WARNING: This will PERMANENTLY WIPE all candidates, jobs, and vector embeddings. Are you sure?")) {
+      return;
+    }
+
+    setLoading(true);
+    setError(null);
+    setResult(null);
+
+    try {
+      const baseUrl = process.env.NEXT_PUBLIC_API_URL || "http://localhost:8000";
+      const resp = await fetch(`${baseUrl}/api/admin/reset-db?password=${encodeURIComponent(password)}`, {
+        method: "POST",
+      });
+
+      if (!resp.ok) {
+        const errData = await resp.json();
+        throw new Error(errData.detail || "Reset failed");
+      }
+
+      const data = await resp.json();
+      setResult(data);
+    } catch (err: any) {
+      setError(err.message);
+    } finally {
+      setLoading(false);
+    }
+  };
+
+  return (
+    <div className="flex flex-col items-center justify-center min-h-[calc(100vh-64px)] p-6 bg-slate-950 text-slate-100">
+      <div className="max-w-md w-full p-8 rounded-3xl bg-slate-900/50 border border-slate-800 shadow-2xl backdrop-blur-sm">
+        <div className="flex flex-col items-center text-center space-y-6">
+          <div className="w-20 h-20 rounded-2xl bg-red-500/10 flex items-center justify-center border border-red-500/20">
+            <span className="text-4xl">☢️</span>
+          </div>
+          
+          <div className="space-y-2">
+            <h1 className="text-3xl font-bold bg-gradient-to-r from-white to-slate-400 bg-clip-text text-transparent">
+              System Reset
+            </h1>
+            <p className="text-slate-400 text-sm">
+              Perform a clean wipe of the PostgreSQL database and Qdrant vector store.
+            </p>
+          </div>
+
+          <div className="w-full p-4 rounded-xl bg-slate-800/30 border border-slate-800 text-xs text-left space-y-2">
+            <p className="font-semibold text-slate-300">This action will:</p>
+            <ul className="list-disc list-inside text-slate-500 space-y-1">
+              <li>Drop and recreate the Public schema</li>
+              <li>Delete and recreate Qdrant collections</li>
+              <li>Re-run all Alembic migrations (Head)</li>
+              <li>Re-initialize vector indices</li>
+            </ul>
+          </div>
+
+          <div className="w-full space-y-2">
+             <input
+              type="password"
+              placeholder="System Password"
+              value={password}
+              onChange={(e) => setPassword(e.target.value)}
+              className="w-full px-4 py-3 rounded-xl bg-slate-950 border border-slate-800 focus:outline-none focus:border-red-500/50 text-sm transition-all"
+            />
+          </div>
+
+          {result && (
+            <div className="w-full p-4 rounded-xl bg-emerald-500/10 border border-emerald-500/20 text-emerald-400 text-sm">
+              <p className="font-bold">Success!</p>
+              <p className="opacity-80">{result.message}</p>
+              <button 
+                onClick={() => router.push("/")}
+                className="mt-3 text-xs underline underline-offset-4 hover:text-emerald-300"
+              >
+                Return to Dashboard
+              </button>
+            </div>
+          )}
+
+          {error && (
+            <div className="w-full p-4 rounded-xl bg-red-500/10 border border-red-500/20 text-red-400 text-sm">
+              <p className="font-bold">Error</p>
+              <p className="opacity-80 break-words">{error}</p>
+            </div>
+          )}
+
+          <button
+            onClick={handleReset}
+            disabled={loading}
+            className={`w-full py-4 rounded-2xl font-bold text-sm tracking-widest uppercase transition-all ${
+              loading 
+                ? "bg-slate-800 text-slate-500 cursor-not-allowed" 
+                : "bg-gradient-to-r from-red-600 to-red-500 hover:from-red-500 hover:to-red-400 active:scale-95 shadow-[0_0_30px_-5px_rgba(220,38,38,0.4)]"
+            }`}
+          >
+            {loading ? (
+              <span className="flex items-center justify-center gap-2">
+                <svg className="animate-spin h-5 w-5 text-white" xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 24 24">
+                  <circle className="opacity-25" cx="12" cy="12" r="10" stroke="currentColor" strokeWidth="4"></circle>
+                  <path className="opacity-75" fill="currentColor" d="M4 12a8 8 0 018-8V0C5.373 0 0 5.373 0 12h4zm2 5.291A7.962 7.962 0 014 12H0c0 3.042 1.135 5.824 3 7.938l3-2.647z"></path>
+                </svg>
+                Executing Wipe...
+              </span>
+            ) : (
+              "Nuke Everything"
+            )}
+          </button>
+          
+          <button 
+            onClick={() => router.back()}
+            disabled={loading}
+            className="text-slate-500 text-xs hover:text-slate-300 transition-colors"
+          >
+            Cancel and Go Back
+          </button>
+        </div>
+      </div>
+    </div>
+  );
+}