Deploy Rachana Data Studio

data_studio/app.py

@@ -283,24 +283,29 @@ def render_transliteration_editor(sample: dict[str, Any]) -> dict[str, Any]:
     return {"current_transliteration": {"native_text": native_text, "latin_text": latin_text}}
 
 
-def render_review(db, user: dict[str, Any]) -> None:
-    st.subheader("Review queue")
-    counts = queue_status_counts(db)
-    queue_name = st.selectbox(
-        "Queue",
-        ["pure_telugu", "translation_tel_eng", "transliteration"],
-    )
-    queue_stats = counts.get(queue_name, {})
-    stats_cols = st.columns(4)
-    stats_cols[0].metric("Pending", queue_stats.get("pending", 0))
-    stats_cols[1].metric("Accepted", queue_stats.get("accepted", 0))
-    stats_cols[2].metric("Edited", queue_stats.get("edited", 0))
-    stats_cols[3].metric("Rejected", queue_stats.get("rejected", 0))
-
-    sample = claim_next_sample(db, queue_name, user["username"])
-    if sample is None:
-        st.info("No pending samples for this queue.")
-        return
+def render_review(db, user: dict[str, Any]) -> None:
+    st.subheader("Review queue")
+    counts = queue_status_counts(db)
+    sources = [source for source in list_sources(db) if source.get("enabled", True)]
+    dataset_options = ["All active datasets"] + [source["dataset_key"] for source in sources]
+    selected_dataset = st.selectbox("Dataset", dataset_options)
+    queue_name = "all"
+    queue_stats = counts.get("pure_telugu", {})
+    stats_cols = st.columns(4)
+    stats_cols[0].metric("Pending", queue_stats.get("pending", 0))
+    stats_cols[1].metric("Accepted", queue_stats.get("accepted", 0))
+    stats_cols[2].metric("Edited", queue_stats.get("edited", 0))
+    stats_cols[3].metric("Rejected", queue_stats.get("rejected", 0))
+
+    sample = claim_next_sample(
+        db,
+        queue_name,
+        user["username"],
+        None if selected_dataset == "All active datasets" else selected_dataset,
+    )
+    if sample is None:
+        st.info("No pending samples for the selected pool.")
+        return
 
     st.caption(f"Sample ID: {sample['sample_id']}")
     meta_cols = st.columns(4)

data_studio/auth.py

@@ -27,6 +27,17 @@ def single_user_username() -> str:
     return normalize_username(os.getenv("SINGLE_USER_USERNAME", ""))
 
 
+def admin_usernames() -> set[str]:
+    raw = os.getenv("ADMIN_USERNAMES", "prashanth").strip()
+    if not raw:
+        return set()
+    return {normalize_username(item) for item in raw.split(",") if item.strip()}
+
+
+def is_admin_username(username: str) -> bool:
+    return normalize_username(username) in admin_usernames()
+
+
 def is_single_user_owner(username: str) -> bool:
     return single_user_mode_enabled() and normalize_username(username) == single_user_username()
 
@@ -63,7 +74,7 @@ def serialize_user(user: dict[str, Any] | None) -> dict[str, Any] | None:
     if user is None:
         return None
     normalized_username = normalize_username(user["username"])
-    role = "owner" if is_single_user_owner(normalized_username) else user.get("role", "reviewer")
+    role = "owner" if is_single_user_owner(normalized_username) else "admin" if is_admin_username(normalized_username) else user.get("role", "reviewer")
     return {
         "username": normalized_username,
         "display_name": user.get("display_name") or normalized_username,
@@ -111,11 +122,11 @@ def create_or_update_user(
     payload = {
         "username": normalized_username,
         "display_name": (display_name or normalized_username).strip(),
-        "role": "admin" if is_single_user_owner(normalized_username) else (role.strip() or "reviewer"),
+        "role": "admin" if (is_single_user_owner(normalized_username) or is_admin_username(normalized_username)) else (role.strip() or "reviewer"),
         "is_active": bool(is_active),
         "password": password_record,
         "updated_at": now,
-    }
+    }
     update_doc = {
         "$set": payload,
         "$setOnInsert": {"created_at": now, "last_login_at": None},

data_studio/review.py

@@ -64,8 +64,7 @@ def claim_next_sample(
 ) -> dict[str, Any] | None:
     now = utc_now()
     expires_at = now + timedelta(minutes=LEASE_MINUTES)
-    lease_filter = {
-        "queue_name": queue_name,
+    lease_filter: dict[str, Any] = {
         "status": "pending",
         "$or": [
             {"lease": {"$exists": False}},
@@ -73,6 +72,8 @@ def claim_next_sample(
             {"lease.reviewer": reviewer},
         ],
     }
+    if queue_name not in {"", "all", "*"}:
+        lease_filter["queue_name"] = queue_name
     if dataset_key:
         lease_filter["source.dataset_key"] = dataset_key
     candidate_ids = [

data_studio/security.py

@@ -7,7 +7,7 @@ from typing import Any
 
 from pymongo.database import Database
 
-from data_studio.auth import authenticate_user, get_active_user, is_single_user_owner, serialize_user
+from data_studio.auth import authenticate_user, get_active_user, is_admin_username, is_single_user_owner, serialize_user
 from data_studio.utils import utc_now
 
 
@@ -53,7 +53,7 @@ def permissions_for_role(role: str) -> list[str]:
 
 
 def permissions_for_user(user: dict[str, Any]) -> list[str]:
-    if is_single_user_owner(user["username"]):
+    if is_single_user_owner(user["username"]) or is_admin_username(user["username"]):
         return ALL_PERMISSIONS
     return permissions_for_role(user["role"])
 
@@ -93,6 +93,6 @@ def delete_session(db: Database, token: str) -> None:
 
 
 def has_permission(user: dict[str, Any], permission: str) -> bool:
-    if is_single_user_owner(user["username"]):
+    if is_single_user_owner(user["username"]) or is_admin_username(user["username"]):
         return True
     return permission in ROLE_PERMISSIONS.get(user.get("role", "reviewer"), set())

frontend/app/page.tsx

@@ -124,7 +124,7 @@ export default function Home() {
   const [tokenProgress, setTokenProgress] = useState<TokenProgress | null>(null);
   const [collaboration, setCollaboration] = useState<Collaboration | null>(null);
   const [section, setSection] = useState<"review" | "overview" | "admin">("review");
-  const [queue, setQueue] = useState("pure_telugu");
+  const [queue, setQueue] = useState("all");
   const [sample, setSample] = useState<Sample | null>(null);
   const [busy, setBusy] = useState(false);
   const [message, setMessage] = useState("");
@@ -297,7 +297,7 @@ function ReviewWorkspace({ queue, queues, sources, sample, busy, collaboration,
   const selectedSource = sources.find((source) => source.dataset_key === datasetKey);
   const selectedQueue = selectedSource?.queue_name || queue;
   const claimLabel = selectedSource ? `Claim from ${selectedSource.name}` : "Claim from all active datasets";
-  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>Leave the selector empty to pool across every active dataset in this queue.</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); }}><option value="">All active datasets</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..." : claimLabel}</button></div>}</>;
+  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>Leave the selector empty to pool across every active dataset.</p></div><label>Active dataset<select value={datasetKey} onChange={(e) => setDatasetKey(e.target.value)}><option value="">All active datasets</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)}><strong>{source.name}</strong><small>{source.dataset_key}</small></button>)}</div></section>{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 active.` : "All active datasets are pooled together."}</p><button className="primary" onClick={() => onClaim(datasetKey || undefined)} disabled={busy}>{busy ? "Finding an item..." : claimLabel}</button></div>}</>;
 }
 
 function ReviewEditor({ sample, busy, onSubmit }: { sample: Sample; busy: boolean; onSubmit: (action: string, editor: EditorState) => void }) {