import argilla as rg
from datasets import load_dataset

client = rg.Argilla(
    api_url="https://ihssane123-my-argilla.hf.space",
    api_key="UJ-3ANbta5mlYiUgJJxWPtp7nr4mXd6wvXEgl2AVi4uEk1ChOdLAjtu1n1lNSYJEB2WylAFRXGvJe6m4cynzk_U4o1twaBP6Kapk85XZIZY",
)

settings = rg.Settings(
    guidelines="Classify the reviews as positive or negative.",
    fields=[
        rg.TextField(
            name="review",
            title="Text from the review",
            use_markdown=False,
        ),
    ],
    questions=[
        rg.LabelQuestion(
            name="my_label",
            title="In which category does this article fit?",
            labels=["positive", "negative"],
        )
    ],
)

dataset = rg.Dataset(
    name=f"my_first_dataset",
    settings=settings,
    client=client,
    #workspace="argilla"
)
dataset.create()

data = load_dataset("imdb", split="train[:100]").to_list()
dataset.records.log(records=data, mapping={"text": "review"})