Update app.py

app.py

@@ -3,26 +3,23 @@
 # dependencies = [
 #     "marimo",
 #     "polars==1.23.0",
-#     "sentence-transformers==3.4.1",
-#     "umap-learn==0.5.7",
-#     "llvmlite==0.44.0",
-#     "altair==5.5.0",
 #     "scikit-learn==1.6.1",
 #     "numpy==2.1.3",
 #     "mohtml==0.1.2",
 #     "model2vec==0.4.0",
+#     "altair==5.5.0",
 # ]
 # ///
 
 import marimo
 
-__generated_with = "0.11.9"
+__generated_with = "0.11.14"
 app = marimo.App()
 
 
 @app.cell
 def _(mo):
-    mo.md("""### Bulk labelling demo""")
+    mo.md("""### Fast labelling demo""")
     return
 
 
@@ -42,8 +39,8 @@ def _(mo):
 
 @app.cell
 def _(mo):
-    pos_label = mo.ui.text("pos", placeholder="positive label name")
-    neg_label = mo.ui.text("neg", placeholder="negative label name")
+    pos_label = mo.ui.text("pos", placeholder="positive label name", label="positive class name")
+    neg_label = mo.ui.text("neg", placeholder="negative label name", label="negative class name")
     return neg_label, pos_label
 
 
@@ -55,7 +52,7 @@ def _(uploaded_file, use_default_switch):
 
 @app.cell
 def _(mo, pl, should_stop, uploaded_file, use_default_switch):
-    mo.stop(form is None, "Need a text input to fetch example")
+    mo.stop(should_stop , mo.md("**Submit a dataset or use default one to continue.**"))
 
     if use_default_switch.value:
         df = pl.read_csv("spam.csv")
@@ -73,6 +70,16 @@ def _(StaticModel, mo):
     return (tfm,)
 
 
+@app.cell
+def _(mo, should_stop):
+    mo.stop(should_stop)
+
+    text_input = mo.ui.text_area("you will win a free ringtone!", label="Reference sentences")
+    form = mo.md("""{text_input}""").batch(text_input=text_input).form()
+    form
+    return form, text_input
+
+
 @app.cell
 def _(mo, texts, tfm):
     with mo.status.spinner(subtitle="Creating embeddings ...") as _spinner:
@@ -81,7 +88,7 @@ def _(mo, texts, tfm):
 
 
 @app.cell
-def _(add_label, get_example, mo, neg_label, pos_label):
+def _(add_label, get_example, mo, neg_label, pos_label, undo):
     btn_spam = mo.ui.button(
         label=f"Annotate {neg_label.value}", 
         on_click=lambda d: add_label(get_example(), neg_label.value), 
@@ -92,7 +99,12 @@ def _(add_label, get_example, mo, neg_label, pos_label):
         on_click=lambda d: add_label(get_example(), pos_label.value),
         keyboard_shortcut="Ctrl-K"
     )
-    return btn_ham, btn_spam
+    btn_undo = mo.ui.button(
+        label="Undo", 
+        on_click=lambda d: undo(),
+        keyboard_shortcut="Ctrl-U"
+    )
+    return btn_ham, btn_spam, btn_undo
 
 
 @app.cell
@@ -101,7 +113,11 @@ def _(gen, get_label, set_example, set_label):
         current_labels = get_label()
         set_label(current_labels + [{"text": text, "label": lab}])
         set_example(next(gen))
-    return (add_label,)
+
+    def undo(): 
+        current_labels = get_label()
+        set_label(current_labels[:-2])
+    return add_label, undo
 
 
 @app.cell
@@ -110,6 +126,23 @@ def _():
     return (br,)
 
 
+@app.cell
+def _(br, btn_ham, btn_spam, btn_undo, example, mo, neg_label, p, pos_label):
+    mo.vstack([
+        mo.hstack([
+           pos_label, neg_label
+        ]),
+        br(),
+        mo.hstack([
+            btn_ham, btn_spam, btn_undo
+        ]),
+        br(),
+        p("Current example:", klass="font-bold"),
+        example
+    ])
+    return
+
+
 @app.cell
 def _(mo):
     get_label, set_label = mo.state([])
@@ -122,23 +155,6 @@ def _(gen, mo):
     return get_example, set_example
 
 
-@app.cell
-def _(div, get_example, p):
-    div(
-        p(get_example()), 
-        klass="bg-gray-100 p-4 rounded-lg"
-    )
-    return
-
-
-@app.cell
-def _(btn_ham, btn_spam, mo):
-    mo.hstack([
-        btn_ham, btn_spam
-    ])
-    return
-
-
 @app.cell
 def _():
     from mohtml import tailwind_css, div, p
@@ -147,16 +163,6 @@ def _():
     return div, p, tailwind_css
 
 
-@app.cell
-def _(mo, should_stop):
-    mo.stop(should_stop)
-
-    text_input = mo.ui.text_area("Query can go here", label="Reference sentences")
-    form = mo.md("""{text_input}""").batch(text_input=text_input).form()
-    form
-    return form, text_input
-
-
 @app.cell
 def _(get_label, mo):
     import json
@@ -173,9 +179,8 @@ def _(get_label, mo):
 
 
 @app.cell
-def _(X, cosine_similarity, form, mo, pl, texts, tfm):
-    mo.stop(form is None, "Need a query input to fetch example")
-    mo.stop(form.value is None, "Need a query input to fetch example")
+def _(X, cosine_similarity, form, get_label, mo, pl, texts, tfm):
+    mo.stop(not form.value.get("text_input", None), "Need a text input to fetch example")
 
     df_emb = (
         pl.DataFrame({
@@ -188,15 +193,25 @@ def _(X, cosine_similarity, form, mo, pl, texts, tfm):
     query = tfm.encode([form.value["text_input"]])
     similarity = cosine_similarity(query, X)[0]
     df_emb = df_emb.with_columns(sim=similarity).sort(pl.col("sim"), descending=True)
-    gen = (_["text"] for _ in df_emb.head(100).to_dicts())
-    return df_emb, gen, query, similarity
+    label_texts = [_["text"] for _ in get_label()]
+    gen = (_["text"] for _ in df_emb.head(100).to_dicts() if _["text"] not in label_texts)
+    return df_emb, gen, label_texts, query, similarity
+
+
+@app.cell
+def _(div, get_example, p):
+    example = div(
+        p(get_example()), 
+        klass="bg-gray-100 p-4 rounded-lg"
+    )
+    return (example,)
 
 
 @app.cell
 def _(get_label, mo, pl, should_stop):
     mo.stop(should_stop)
 
-    pl.DataFrame(get_label())
+    pl.DataFrame(get_label()).reverse()
     return
 
 
@@ -204,12 +219,9 @@ def _(get_label, mo, pl, should_stop):
 def _(mo):
     with mo.status.spinner(subtitle="Loading libraries ...") as _spinner:
         import polars as pl
-        import altair as alt
         import numpy as np
         from sklearn.metrics.pairwise import cosine_similarity
-        from sklearn.linear_model import LogisticRegression
-        from sklearn.decomposition import PCA
-    return LogisticRegression, PCA, alt, cosine_similarity, np, pl
+    return cosine_similarity, np, pl
 
 
 @app.cell
@@ -231,4 +243,4 @@ def _():
 
 
 if __name__ == "__main__":
-    app.run()
+    app.run()