Deploy Rachana Data Studio

data_studio/app.py

@@ -134,9 +134,9 @@ def render_overview(db) -> None:
         st.warning("No source configs seeded yet.")
 
 
-def render_sources(db) -> None:
-    st.subheader("Source setup")
-    st.caption("Seed the default source configurations for the first three review queues.")
+def render_sources(db) -> None:
+    st.subheader("Source setup")
+    st.caption("Seed the default source configurations for the active review queues.")
     if st.button("Seed default sources", type="primary"):
         inserted = seed_default_sources(db)
         st.success(f"Source config seed completed. New records inserted: {inserted}.")

data_studio/sources.py

@@ -108,41 +108,17 @@ def default_source_configs() -> list[dict[str, Any]]:
                 "updated_at": now,
             }
         )
-    sources.append(
-        {
-            "dataset_key": "wikipedia_telugu",
-            "name": "Wikipedia Telugu",
-            "description": "Pure Telugu document review queue from Wikimedia Wikipedia Telugu.",
-            "queue_name": "pure_telugu",
-            "sample_type": "document",
-            "hf_dataset": "wikimedia/wikipedia",
-            "hf_config": "20231101.te",
-            "hf_split": "train",
-            "hf_data_dir": None,
-            "field_mapping": {
-                "source_record_id": "id",
-                "text": "text",
-                "doc_type": None,
-                "title": "title",
-                "url": "url",
-            },
-            "filters": {"min_chars": 30},
-            "enabled": True,
-            "created_at": now,
-            "updated_at": now,
-        }
-    )
 
     return sources
-
-
-def seed_default_sources(db: Database) -> int:
-    inserted = 0
-    for source in default_source_configs():
-        result = db["source_configs"].update_one(
-            {"dataset_key": source["dataset_key"]},
-            {
-                "$set": {
+
+
+def seed_default_sources(db: Database) -> int:
+    inserted = 0
+    for source in default_source_configs():
+        result = db["source_configs"].update_one(
+            {"dataset_key": source["dataset_key"]},
+            {
+                "$set": {
                     "name": source["name"],
                     "description": source["description"],
                     "queue_name": source["queue_name"],
@@ -159,10 +135,11 @@ def seed_default_sources(db: Database) -> int:
                 "$setOnInsert": {"created_at": source["created_at"]},
             },
             upsert=True,
-        )
-        if result.upserted_id is not None:
-            inserted += 1
-    return inserted
+        )
+        if result.upserted_id is not None:
+            inserted += 1
+    db["source_configs"].delete_many({"dataset_key": "wikipedia_telugu"})
+    return inserted
 
 
 def _serialize_source(source: dict[str, Any]) -> dict[str, Any]: