Update app.py

app.py

@@ -7,15 +7,20 @@ import streamlit as st
 import pandas as pd
 from PIL import Image
 from filelock import FileLock
+from huggingface_hub import HfApi
 
 # ---------------- Config ----------------
-BASE_DIR = Path(".").resolve()   # Space root
 IMAGE_DIR = os.getenv("IMAGE_DIR", "images")
-LABELS_CSV = os.getenv("LABELS_CSV", str(BASE_DIR / "labels.csv"))
+LABELS_CSV = os.getenv("LABELS_CSV", "labels.csv")
 LABEL_LEFT = os.getenv("LABEL_A", "Class A")
 LABEL_RIGHT = os.getenv("LABEL_B", "Class B")
 SUPPORTED_EXTS = (".png", ".jpg", ".jpeg", ".bmp", ".gif", ".tif", ".tiff", ".webp")
 
+HF_TOKEN = os.getenv("HF_TOKEN")  # stored in Space secrets
+SPACE_ID = os.getenv("SPACE_ID", "MichaelDeutges/LabelingTest")  # change if repo name differs
+
+api = HfApi()
+
 st.set_page_config(page_title="Two-Button Image Labeler", layout="centered")
 
 # --------------- Helpers ----------------
@@ -41,7 +46,7 @@ def rel_to_image_dir(p: str):
         return p
 
 def write_label(image_path: str, label: str, annotator: str):
-    os.makedirs(Path(LABELS_CSV).parent, exist_ok=True)
+    os.makedirs(os.path.dirname(LABELS_CSV) or ".", exist_ok=True)
     record = {
         "image": rel_to_image_dir(image_path),
         "label": label,
@@ -53,9 +58,21 @@ def write_label(image_path: str, label: str, annotator: str):
         df = pd.DataFrame([record])
         df.to_csv(LABELS_CSV, mode="a", header=not exists, index=False)
 
+    # Upload to Hugging Face repo
+    try:
+        api.upload_file(
+            path_or_fileobj=LABELS_CSV,
+            path_in_repo="labels.csv",
+            repo_id=SPACE_ID,
+            repo_type="space",
+            token=HF_TOKEN
+        )
+    except Exception as e:
+        print("Upload failed:", e)
+
 # --------------- UI ---------------------
 st.title("🏷️ Two-Button Image Labeler")
-st.write("Enter your name, click **Start**, then label each image using the big buttons. Use **Skip** to bypass an image.")
+st.write("Enter your name, click **Start**, then label each image. Use **Skip** to bypass an image.")
 
 with st.sidebar:
     default_name = st.session_state.get("annotator", "")
@@ -65,7 +82,7 @@ with st.sidebar:
     with c1:
         start_btn = st.button("Start", type="primary")
     with c2:
-        restart_req = st.button("Restart")
+        reset_btn = st.button("🔄 Reset Session")
 
     continue_by_name = st.toggle(
         "Only show images I haven't labeled yet",
@@ -82,16 +99,13 @@ if "total" not in st.session_state:
     st.session_state.total = 0
 if "started" not in st.session_state:
     st.session_state.started = False
-if "annotator" not in st.session_state:
-    st.session_state.annotator = ""
 
-# restart clears state
-if restart_req:
+# reset
+if reset_btn:
+    st.session_state.started = False
     st.session_state.order = []
     st.session_state.idx = 0
     st.session_state.total = 0
-    st.session_state.started = False
-    st.session_state.annotator = ""
     st.rerun()
 
 # start
@@ -99,18 +113,19 @@ if start_btn:
     if not annotator.strip():
         st.sidebar.error("Please enter your name.")
     else:
-        st.session_state.annotator = annotator.strip()
+        st.session_state["annotator"] = annotator.strip()
         imgs = list_images()
         labels_df = read_labels()
 
-        # Already labeled by this annotator?
         if continue_by_name and len(labels_df) > 0:
-            labeled = set(labels_df.query("annotator == @annotator")["image"].astype(str).tolist())
+            already = set(
+                labels_df.query("annotator == @annotator")["image"].astype(str).tolist()
+            )
             rel_imgs = [rel_to_image_dir(p) for p in imgs]
-            imgs = [p for p, r in zip(imgs, rel_imgs) if r not in labeled]
+            imgs = [p for p, r in zip(imgs, rel_imgs) if r not in already]
 
         if not imgs:
-            st.warning("No images found (or all labeled by you). Upload PNG/JPG/etc. to the `images/` folder.")
+            st.warning("No images found (or all labeled).")
         else:
             st.session_state.order = imgs
             st.session_state.idx = 0
@@ -127,13 +142,11 @@ else:
         st.success("All done 🎉 Thank you!")
     else:
         current_image = st.session_state.order[idx]
-        st.caption(f"{idx} / {total}")
+        st.caption(f"{idx+1} / {total}")
         try:
             img = Image.open(current_image)
-            # If multipage TIFF, show first page
             if getattr(img, "n_frames", 1) > 1:
                 img.seek(0)
-            # Convert to RGB for browsers
             if img.mode not in ("RGB", "RGBA"):
                 img = img.convert("RGB")
             st.image(img, use_container_width=True)
@@ -143,20 +156,19 @@ else:
         col1, col2, col3 = st.columns([1,1,1])
         with col1:
             if st.button(f"⬅️ {LABEL_LEFT}", use_container_width=True):
-                write_label(current_image, LABEL_LEFT, st.session_state.annotator)
+                write_label(current_image, LABEL_LEFT, annotator.strip())
                 st.session_state.idx += 1
                 st.rerun()
         with col2:
             if st.button("Skip", use_container_width=True):
-                write_label(current_image, "Skip", st.session_state.annotator)
+                write_label(current_image, "Skip", annotator.strip())
                 st.session_state.idx += 1
                 st.rerun()
         with col3:
             if st.button(f"{LABEL_RIGHT} ➡️", use_container_width=True):
-                write_label(current_image, LABEL_RIGHT, st.session_state.annotator)
+                write_label(current_image, LABEL_RIGHT, annotator.strip())
                 st.session_state.idx += 1
                 st.rerun()
 
-# Footer / admin
 st.divider()
-st.caption("Labels are saved to `labels.csv` in this Space. You can download it from the Files tab.")
+st.caption("Labels are saved to `labels.csv` and synced to this Space.")