Deploy Forager's Field Station

README.md

@@ -8,6 +8,7 @@ sdk_version: 6.16.0
 python_version: "3.11"
 app_file: app.py
 pinned: false
+hf_oauth: true
 license: apache-2.0
 short_description: Pocket-sized intelligence for identifying edible wild foods
 ---

app.py

@@ -17,7 +17,7 @@ import gradio as gr
 
 from pipeline.convergence import build_result
 from pipeline.infer import Pipeline
-from game.ui import build_game_tab
+from game.ui import build_game_tab, build_stump_tab
 
 HERE = os.path.dirname(os.path.abspath(__file__))
 EXAMPLES_DIR = os.path.join(HERE, "examples")
@@ -349,6 +349,8 @@ with gr.Blocks(title="Forager's Field Station") as demo:
         "      <span>REFUSES BY DEFAULT</span></div>"
         "</div>"
     )
+    with gr.Row(elem_id="loginbar"):
+        gr.LoginButton(value="▸ Sign in with Hugging Face to post & contribute")
     with gr.Tabs():
         with gr.Tab("◧ FIELD STATION"):
             gr.HTML(EXPERTS_PANEL)
@@ -375,6 +377,9 @@ with gr.Blocks(title="Forager's Field Station") as demo:
         with gr.Tab("⚔ BEAT THE MACHINE"):
             build_game_tab(PIPE)
 
+        with gr.Tab("⚑ STUMP THE MACHINE"):
+            build_stump_tab(PIPE)
+
 if __name__ == "__main__":
     # theme/css belong in launch() in Gradio 6. ssr_mode=False is also enforced via
     # the GRADIO_SSR_MODE Space variable — Gradio 6's Node SSR proxy can't bind port

game/datastore.py

@@ -79,6 +79,23 @@ def post_score(contributor: str, you_correct: int, total: int, machine_correct:
     return list(rows.values())
 
 
+def load_contributors() -> list[dict]:
+    """Count accepted sightings per contributor from metadata.jsonl (the Contributor board)."""
+    if not _TOKEN:
+        return []
+    try:
+        from collections import Counter
+        from huggingface_hub import hf_hub_download
+        path = hf_hub_download(DATASET_REPO, "metadata.jsonl", repo_type="dataset",
+                               token=_TOKEN, force_download=True)
+        with open(path) as f:
+            rows = [json.loads(line) for line in f if line.strip()]
+        counts = Counter(r.get("contributor", "?") for r in rows)
+        return [{"contributor": k, "count": v} for k, v in counts.most_common()]
+    except Exception:
+        return []
+
+
 def append_sighting(image, user_label: str, machine: dict, contributor: str) -> bool:
     """
     Phase 2: commit one contributed photo + metadata row to the dataset.

game/ui.py

@@ -126,10 +126,7 @@ def build_game_tab(pipe) -> None:
     score_html = gr.HTML(_scoreboard({"you": 0, "machine": 0, "total": 0}))
 
     gr.HTML("<div class='gm-divider'>▌ LEADERBOARD</div>")
-    with gr.Row():
-        nick = gr.Textbox(label="Name for the leaderboard", placeholder="handle",
-                          scale=2, elem_classes="eink-input")
-        post_btn = gr.Button("POST MY SCORE", elem_classes="eink-scan", scale=1)
+    post_btn = gr.Button("POST MY SCORE", elem_classes="eink-scan")
     lb_status = gr.HTML("")
     lb = gr.Dataframe(value=_lb_df(), headers=_LB_COLS, interactive=False,
                       elem_classes="gm-lb", wrap=True)
@@ -156,20 +153,116 @@ def build_game_tab(pipe) -> None:
         card = {**card, "answered": True}
         return _reveal_html(card, tier, you_right, mtier, mach_right), card, score
 
-    def _post(nickname, score):
-        name = (nickname or "").strip()
-        if not name:
-            return "<div class='gm-warn'>Enter a name first.</div>", gr.update()
+    def _post(score, profile: gr.OAuthProfile | None = None):
+        if profile is None:
+            return "<div class='gm-warn'>Log in (top of the page) to post your score.</div>", gr.update()
         if score["total"] == 0:
             return "<div class='gm-warn'>Play at least one round first.</div>", gr.update()
-        datastore.post_score(name, score["you"], score["total"], score["machine"])
+        datastore.post_score(profile.username, score["you"], score["total"], score["machine"])
         note = "" if datastore.persistence_enabled() else \
             " <span class='gm-note'>(session-only until the dataset token is set)</span>"
-        return f"<div class='gm-ok'>Posted, {name}.{note}</div>", _lb_df()
+        return f"<div class='gm-ok'>Posted as {profile.username}.{note}</div>", _lb_df()
 
     deal_btn.click(_deal, [score_state, card_state], [card_img, card_state, reveal, score_html])
     for b, t in [(safe_b, "SAFE"), (caut_b, "CAUTION"), (dead_b, "DEADLY")]:
         b.click(partial(_guess, t), [card_state, score_state], [reveal, card_state, score_state]) \
          .then(_scoreboard, [score_state], [score_html])
-    post_btn.click(_post, [nick, score_state], [lb_status, lb])
+    post_btn.click(_post, [score_state], [lb_status, lb])
     refresh_btn.click(lambda: _lb_df(), None, [lb])
+
+
+# ── Stump the Machine: upload your own find -> the data flywheel ──────────────
+_CONTRIB_COLS = ["Contributor", "Sightings"]
+
+
+def _contrib_df() -> pd.DataFrame:
+    rows = datastore.load_contributors()
+    return pd.DataFrame([[r["contributor"], r["count"]] for r in rows], columns=_CONTRIB_COLS)
+
+
+def _stump_reject_html() -> str:
+    return (
+        "<div class='gm-reveal'>"
+        "<div class='gm-truth-h'>OUT OF DOMAIN</div>"
+        "<div class='gm-truth' style='color:#57544c'>NOT STORED</div>"
+        "<div class='gm-species'>The router didn't place this in berry, mushroom, or plant — "
+        "so it wasn't added to the dataset. Try a clear photo of a wild plant, mushroom, or berry.</div>"
+        "</div>"
+    )
+
+
+def _stump_result_html(label: str, res, stored: bool) -> str:
+    mtier = "UNKNOWN" if res.abstained else res.safety
+    c = TIER_COLOR.get(mtier, "#57544c")
+    mcall = "refused to commit" if res.abstained else f"{_pretty(res.species)} · {mtier}"
+    head = "🏆 You stumped the machine!" if res.abstained else "The machine made its call."
+    store_line = (
+        "<div class='gm-ok'>✓ Added to the open dataset — thank you. This is exactly the "
+        "data that makes the next model better.</div>" if stored else
+        "<div class='gm-warn'>Couldn't persist (dataset token not set on the Space).</div>"
+    )
+    return (
+        f"<div class='gm-reveal'>"
+        f"<div class='gm-truth-h'>YOUR FIND</div>"
+        f"<div class='gm-flav'>{head}</div>"
+        f"<div class='gm-species'>you said: <b>{label}</b></div>"
+        f"<div class='gm-chips'><div class='gm-chip' style='border-color:{c}'>"
+        f"<span class='gm-chip-h' style='color:{c} !important'>MACHINE</span>"
+        f"<span class='gm-chip-t' style='color:{c} !important'>{mcall}</span></div></div>"
+        f"{store_line}"
+        f"<div class='gm-abst'>An ID here is never permission to eat — verify with an expert.</div>"
+        f"</div>"
+    )
+
+
+def build_stump_tab(pipe) -> None:
+    """Upload a real find: the router gates it, the model calls it, and (consented) it
+    joins the public dataset. The flywheel — every submission trains the next model."""
+    gr.HTML(
+        "<div class='gm-intro'>"
+        "<div class='gm-intro-h'>STUMP THE MACHINE</div>"
+        "<div class='gm-intro-b'>Upload your own find. The machine calls it on the spot — and "
+        "when it can't, you've stumped it. Every accepted photo joins the open "
+        "<b>CC-BY-4.0</b> dataset that trains the next model. You're not just playing; you're "
+        "building the thing.</div></div>"
+    )
+
+    with gr.Row():
+        with gr.Column(scale=1):
+            up = gr.Image(type="pil", sources=["upload", "webcam"], label="YOUR FIND",
+                          elem_classes="eink-input", height=300)
+            guess = gr.Textbox(label="What do you think it is?", placeholder="e.g. chanterelle",
+                               elem_classes="eink-input")
+            consent = gr.Checkbox(
+                value=False,
+                label="Contribute this photo to the public CC-BY-4.0 dataset (HomesteaderLabs/forager-sightings).")
+            submit = gr.Button("▸ SUBMIT FIND", variant="primary", elem_classes="eink-scan")
+        with gr.Column(scale=1, elem_classes="eink-screen"):
+            sresult = gr.HTML("<div class='gm-idle'>Upload a find, add your guess, and submit.</div>")
+
+    gr.HTML("<div class='gm-divider'>▌ TOP CONTRIBUTORS</div>")
+    contrib = gr.Dataframe(value=_contrib_df(), headers=_CONTRIB_COLS, interactive=False,
+                           elem_classes="gm-lb", wrap=True)
+    crefresh = gr.Button("⟳ refresh", elem_classes="gm-refresh")
+
+    def _submit(image, user_label, consented, profile: gr.OAuthProfile | None = None):
+        if profile is None:
+            return "<div class='gm-warn'>Log in (top of the page) to contribute.</div>", gr.update()
+        if image is None:
+            return "<div class='gm-warn'>Upload a photo first.</div>", gr.update()
+        if not (user_label or "").strip():
+            return "<div class='gm-warn'>Add your guess first.</div>", gr.update()
+        if not consented:
+            return "<div class='gm-warn'>Check the consent box to contribute.</div>", gr.update()
+        call = pipe.identify(image)
+        res = build_result(call)
+        in_domain = (not res.abstained) or call.get("reason", "") == "low_confidence"
+        if not in_domain:
+            return _stump_reject_html(), gr.update()
+        machine = {"species": res.species, "confidence": res.confidence,
+                   "abstained": res.abstained, "safety": res.safety, "domain": res.domain}
+        stored = datastore.append_sighting(image, user_label.strip(), machine, profile.username)
+        return _stump_result_html(user_label.strip(), res, stored), _contrib_df()
+
+    submit.click(_submit, [up, guess, consent], [sresult, contrib])
+    crefresh.click(lambda: _contrib_df(), None, [contrib])

requirements.txt

@@ -1,4 +1,4 @@
-gradio==6.16.0
+gradio[oauth]==6.16.0
 onnxruntime==1.18.0
 numpy<2
 pillow