Deploy Rachana Data Studio

data_studio/api.py

@@ -167,8 +167,12 @@ def overview(user: dict[str, Any] = Depends(current_user)) -> dict[str, Any]:
 
 
 @app.post("/api/review/claim/{queue_name}")
-def claim(queue_name: str, user: dict[str, Any] = Depends(require_permission("review"))) -> dict[str, Any]:
-    return {"sample": claim_next_sample_with_enrichment(db, settings, queue_name, user["username"])}
+def claim(
+    queue_name: str,
+    dataset_key: str | None = None,
+    user: dict[str, Any] = Depends(require_permission("review")),
+) -> dict[str, Any]:
+    return {"sample": claim_next_sample_with_enrichment(db, settings, queue_name, user["username"], dataset_key)}
 
 
 @app.post("/api/review/{sample_id}/release")

data_studio/review.py

@@ -56,7 +56,12 @@ def queue_overview(db: Database, settings: StudioSettings) -> dict[str, Any]:
     }
 
 
-def claim_next_sample(db: Database, queue_name: str, reviewer: str) -> dict[str, Any] | None:
+def claim_next_sample(
+    db: Database,
+    queue_name: str,
+    reviewer: str,
+    dataset_key: str | None = None,
+) -> dict[str, Any] | None:
     now = utc_now()
     expires_at = now + timedelta(minutes=LEASE_MINUTES)
     lease_filter = {
@@ -68,6 +73,8 @@ def claim_next_sample(db: Database, queue_name: str, reviewer: str) -> dict[str,
             {"lease.reviewer": reviewer},
         ],
     }
+    if dataset_key:
+        lease_filter["source.dataset_key"] = dataset_key
     candidate_ids = [
         row["_id"]
         for row in db["samples"].aggregate(
@@ -106,8 +113,9 @@ def claim_next_sample_with_enrichment(
     settings: StudioSettings,
     queue_name: str,
     reviewer: str,
+    dataset_key: str | None = None,
 ) -> dict[str, Any] | None:
-    return claim_next_sample(db, queue_name, reviewer)
+    return claim_next_sample(db, queue_name, reviewer, dataset_key=dataset_key)
 
 
 def next_sample(db: Database, queue_name: str) -> dict[str, Any] | None:

frontend/app/globals.css

@@ -13,7 +13,11 @@
 .reviewStack{display:grid;gap:18px}
 .decisionCardBottom{width:100%}
 .connectedCard p{color:var(--muted);margin:0 0 12px}
+.connectedHeader{display:flex;justify-content:space-between;gap:14px;align-items:end;margin-bottom:14px}
+.connectedHeader label{display:grid;gap:6px;font-size:12px;font-weight:700;min-width:240px}
+.connectedHeader select{border:1px solid var(--line);border-radius:10px;padding:10px 12px;background:#fff;color:var(--ink)}
 .connectedList{display:grid;grid-template-columns:repeat(2,minmax(0,1fr));gap:10px}
-.connectedChip{border:1px solid var(--line);border-radius:12px;padding:12px;background:#f7f8f4;display:grid;gap:4px}
+.connectedChip{border:1px solid var(--line);border-radius:12px;padding:12px;background:#f7f8f4;display:grid;gap:4px;text-align:left}
+.connectedChip.active{border-color:var(--green);background:var(--green-soft)}
 .connectedChip small{color:var(--muted)}
 @media(max-width:760px){.shellCollapsed{grid-template-columns:1fr}.sidebarToggle{width:100%}}

frontend/app/page.tsx

@@ -130,7 +130,7 @@ export default function Home() {
   const [message, setMessage] = useState("");
   const [authChecked, setAuthChecked] = useState(false);
   const [sidebarCollapsed, setSidebarCollapsed] = useState(false);
-  const canAccessAdmin = user ? user.permissions.some((p) => ["import", "export", "manage_sources", "manage_users"].includes(p)) : false;
+  const canAccessAdmin = user ? user.role === "admin" || user.permissions.some((p) => ["import", "export", "manage_sources", "manage_users"].includes(p)) : false;
 
   const refreshOverview = useCallback(async (authToken: string) => {
     const data = await api<{ queues: QueueCounts; sources: Source[]; user: User; token_progress: TokenProgress; collaboration: Collaboration }>("/api/overview", authToken);
@@ -160,11 +160,12 @@ export default function Home() {
     }).finally(() => setAuthChecked(true));
   }, [refreshOverview]);
 
-  const claim = async (selectedQueue = queue) => {
+  const claim = async (selectedQueue = queue, selectedDatasetKey?: string) => {
     setBusy(true);
     setMessage("");
     try {
-      const data = await api<{ sample: Sample | null }>(`/api/review/claim/${selectedQueue}`, token, { method: "POST" });
+      const datasetQuery = selectedDatasetKey ? `?dataset_key=${encodeURIComponent(selectedDatasetKey)}` : "";
+      const data = await api<{ sample: Sample | null }>(`/api/review/claim/${selectedQueue}${datasetQuery}`, token, { method: "POST" });
       setSample(data.sample);
       if (!data.sample) setMessage("This queue has no available pending samples.");
     } catch (error) {
@@ -253,7 +254,7 @@ export default function Home() {
       </aside>
       <main>
         {message && <div className="notice" role="status">{message}<button onClick={() => setMessage("")}>Dismiss</button></div>}
-        {section === "review" && <ReviewWorkspace queue={queue} queues={queues} sources={sources} sample={sample} busy={busy} collaboration={collaboration} onQueue={changeQueue} onClaim={() => claim()} onSubmit={submitReview} />}
+        {section === "review" && <ReviewWorkspace queue={queue} queues={queues} sources={sources} sample={sample} busy={busy} collaboration={collaboration} onQueue={changeQueue} onClaim={(datasetKey) => claim(queue, datasetKey)} onSubmit={submitReview} />}
         {section === "overview" && <Overview queues={queues} sources={sources} tokenProgress={tokenProgress} collaboration={collaboration} />}
         {section === "admin" && <Admin token={token} sources={sources} permissions={user.permissions} onDone={() => refreshOverview(token)} />}
       </main>
@@ -286,8 +287,16 @@ type EditorState = { cleaned: Record<string, unknown>; tags: string; notes: stri
 
 type ReviewLine = { id: string; left: string; right: string };
 
-function ReviewWorkspace({ queue, queues, sources, sample, busy, collaboration, onQueue, onClaim, onSubmit }: { queue: string; queues: QueueCounts; sources: Source[]; sample: Sample | null; busy: boolean; collaboration: Collaboration | null; onQueue: (q: string) => void; onClaim: () => void; onSubmit: (action: string, editor: EditorState) => void }) {
-  return <><header className="pageHeader"><div><span className="eyebrow">Review workspace</span><h1>Make one clear decision at a time.</h1><p>Each reviewer gets a random available sample. Token progress stays global across the whole team.</p></div><div className="headerStats">{collaboration && <div className="metricInline"><span>Total cleaned</span><strong>{collaboration.total_cleaned_samples.toLocaleString()}</strong></div>}{sample?.lease && <div className="lease">Reserved until {new Date(sample.lease.expires_at).toLocaleTimeString([], { hour: "2-digit", minute: "2-digit" })}</div>}</div></header><section className="tableCard connectedCard"><h2>Connected datasets</h2><p>All currently registered sources are listed here. If one is missing, reseed sources from Data operations.</p><div className="connectedList">{sources.map((source) => <div key={source.dataset_key} className="connectedChip"><strong>{source.name}</strong><small>{source.dataset_key}</small></div>)}</div></section><div className="queueTabs">{QUEUES.map((item) => <button key={item.id} className={queue === item.id ? "active" : ""} onClick={() => onQueue(item.id)}><strong>{item.label}</strong><span>{queues[item.id]?.pending || 0} pending</span></button>)}</div>{sample ? <ReviewEditor key={sample.sample_id} sample={sample} busy={busy} onSubmit={onSubmit} /> : <div className="emptyState"><div>✓</div><h2>Ready for the next item?</h2><p>{QUEUES.find((q) => q.id === queue)?.description}</p><button className="primary" onClick={onClaim} disabled={busy}>{busy ? "Finding an item..." : "Claim random sample"}</button></div>}</>;
+function ReviewWorkspace({ queue, queues, sources, sample, busy, collaboration, onQueue, onClaim, onSubmit }: { queue: string; queues: QueueCounts; sources: Source[]; sample: Sample | null; busy: boolean; collaboration: Collaboration | null; onQueue: (q: string) => void; onClaim: (datasetKey?: string) => void; onSubmit: (action: string, editor: EditorState) => void }) {
+  const [datasetKey, setDatasetKey] = useState(sources[0]?.dataset_key || "");
+  useEffect(() => {
+    if (sources.length && !sources.some((source) => source.dataset_key === datasetKey)) {
+      setDatasetKey(sources[0].dataset_key);
+    }
+  }, [datasetKey, sources]);
+  const selectedSource = sources.find((source) => source.dataset_key === datasetKey);
+  const selectedQueue = selectedSource?.queue_name || queue;
+  return <><header className="pageHeader"><div><span className="eyebrow">Review workspace</span><h1>Make one clear decision at a time.</h1><p>Each reviewer gets a random available sample. Token progress stays global across the whole team.</p></div><div className="headerStats">{collaboration && <div className="metricInline"><span>Total cleaned</span><strong>{collaboration.total_cleaned_samples.toLocaleString()}</strong></div>}{sample?.lease && <div className="lease">Reserved until {new Date(sample.lease.expires_at).toLocaleTimeString([], { hour: "2-digit", minute: "2-digit" })}</div>}</div></header><section className="tableCard connectedCard"><div className="connectedHeader"><div><h2>Connected datasets</h2><p>Pick a dataset, then claim from its queue. This keeps review focused without hiding other sources.</p></div><label>Active dataset<select value={datasetKey} onChange={(e) => { const next = e.target.value; setDatasetKey(next); const nextSource = sources.find((source) => source.dataset_key === next); if (nextSource) onQueue(nextSource.queue_name); }}>{sources.length ? null : <option value="">No connected datasets</option>}<option value="">Select a dataset</option>{sources.map((source) => <option key={source.dataset_key} value={source.dataset_key}>{source.name}</option>)}</select></label></div><div className="connectedList">{sources.map((source) => <button key={source.dataset_key} type="button" className={`connectedChip${source.dataset_key === datasetKey ? " active" : ""}`} onClick={() => { setDatasetKey(source.dataset_key); onQueue(source.queue_name); }}><strong>{source.name}</strong><small>{source.dataset_key}</small></button>)}</div></section><div className="queueTabs">{QUEUES.map((item) => <button key={item.id} className={queue === item.id ? "active" : ""} onClick={() => onQueue(item.id)}><strong>{item.label}</strong><span>{queues[item.id]?.pending || 0} pending</span></button>)}</div>{sample ? <ReviewEditor key={sample.sample_id} sample={sample} busy={busy} onSubmit={onSubmit} /> : <div className="emptyState"><div>✓</div><h2>Ready for the next item?</h2><p>{selectedSource ? `${selectedSource.name} is mapped to ${selectedQueue}.` : QUEUES.find((q) => q.id === queue)?.description}</p><button className="primary" onClick={() => onClaim(datasetKey || undefined)} disabled={busy}>{busy ? "Finding an item..." : "Claim random sample"}</button></div>}</>;
 }
 
 function ReviewEditor({ sample, busy, onSubmit }: { sample: Sample; busy: boolean; onSubmit: (action: string, editor: EditorState) => void }) {