update to argilla on main repo

src/argilla_utils.py

@@ -1,4 +1,4 @@
-import argilla_sdk as rg
+import argilla as rg
 from datasets import load_dataset
 
 from datasets import load_dataset
@@ -14,12 +14,11 @@ from src.dataset import (
     load_repo_id,
 )
 
-client = rg.Argilla(api_url="http://localhost:6900", api_key="owner.apikey")
-
 
 def define_dataset_setting(
-    dataset_name, field_columns, question_columns, metadata_columns
+    dataset_name, field_columns, question_columns, metadata_columns, argilla_space_url
 ):
+    client = rg.Argilla(api_url=argilla_space_url, api_key="owner.apikey")
     split = load_split()
 
     fields, questions, metadata, vectors = [], [], [], []
@@ -78,7 +77,8 @@ def define_dataset_setting(
     return str(settings.serialize()), mapping
 
 
-def add_records(argilla_dataset_name, mapping, n_records):
+def add_records(argilla_dataset_name, mapping, n_records, argilla_space_url):
+    client = rg.Argilla(api_url=argilla_space_url, api_key="owner.apikey")
     split = load_split()
     df = load_dataset(load_repo_id())[split].take(n_records).to_pandas()
     dataset = client.datasets(argilla_dataset_name)

src/spaces.py

@@ -2,10 +2,20 @@ from huggingface_hub import duplicate_space
 
 
 def create_argilla_space(target_argilla_space):
-    duplicate_space(
+    url = duplicate_space(
         from_id="argilla/argilla-template-space",
         to_id=target_argilla_space,
         private=False,
         exist_ok=True,
     )
-    return target_argilla_space
+    return repo_url_to_embed_url(url)
+
+
+def repo_url_to_embed_url(url):
+    if url.startswith("https://huggingface.co/spaces/"):
+        space_id = url.split("/")[-1].replace("_", "-")
+        user_id = url.split("/")[-2].replace("_", "-")
+        embed_url = f"https://{user_id}-{space_id}.hf.space"
+        return embed_url
+    else:
+        return url