Update app.py

app.py

@@ -1,5 +1,3 @@
-
-
 import os
 import glob
 from pathlib import Path
@@ -13,7 +11,10 @@ from huggingface_hub import HfApi
 
 # ---------------- Config ----------------
 IMAGE_DIR   = os.getenv("IMAGE_DIR", "images")
-LABELS_CSV  = os.getenv("LABELS_CSV", "labels.csv")
+
+# ✅ write to the Space's persistent volume (no rebuilds)
+LABELS_CSV  = os.getenv("LABELS_CSV", "/data/labels.csv")
+
 SUPPORTED_EXTS = (".png", ".jpg", ".jpeg", ".bmp", ".gif", ".tif", ".tiff", ".webp")
 
 # Label names
@@ -22,7 +23,7 @@ LABEL_BLAST     = os.getenv("LABEL_BLAST", "BLAST")
 LABEL_UNCERTAIN = os.getenv("LABEL_UNCERTAIN", "UNCERTAIN")
 LABEL_TRASH     = os.getenv("LABEL_TRASH", "LOW_QUALITY")
 
-# (Optional) Hugging Face sync
+# (Optional) Hugging Face sync (manual button below)
 HF_TOKEN = os.getenv("HF_TOKEN")  # set as a Secret in Space settings
 SPACE_ID = os.getenv("SPACE_ID", "MichaelDeutges/LabelingTest")  # set to your Space id
 api = HfApi()
@@ -52,7 +53,7 @@ def rel_to_image_dir(p: str):
         return p
 
 def write_label(image_path: str, label: str, annotator: str):
-    """Append one row to labels.csv and (optionally) push to the Space repo."""
+    """Append one row to labels.csv."""
     os.makedirs(os.path.dirname(LABELS_CSV) or ".", exist_ok=True)
     record = {
         "image": rel_to_image_dir(image_path),
@@ -64,18 +65,8 @@ def write_label(image_path: str, label: str, annotator: str):
         exists = os.path.exists(LABELS_CSV)
         pd.DataFrame([record]).to_csv(LABELS_CSV, mode="a", header=not exists, index=False)
 
-    # best-effort sync to Hugging Face repo (safe to skip if no token/space id)
-    if HF_TOKEN and SPACE_ID:
-        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)
+    # ❌ removed auto upload to the Space (this caused rebuilds)
+    # If you want to sync, use the manual button in the sidebar.
 
 # --------------- UI ---------------------
 st.title("🏷️ Blast Cell Labeling App")
@@ -102,6 +93,24 @@ with st.sidebar:
         help="Continue where you left off, based on your name in labels.csv",
     )
 
+    st.markdown("---")
+    # ✅ Optional manual sync to Space Files tab
+    if st.button("⬆️ Sync labels.csv to Space Files"):
+        if HF_TOKEN and SPACE_ID and os.path.exists(LABELS_CSV):
+            try:
+                api.upload_file(
+                    path_or_fileobj=LABELS_CSV,
+                    path_in_repo="labels.csv",
+                    repo_id=SPACE_ID,
+                    repo_type="space",
+                    token=HF_TOKEN
+                )
+                st.success("Synced labels.csv to Space Files.")
+            except Exception as e:
+                st.error(f"Sync failed: {e}")
+        else:
+            st.warning("Set HF_TOKEN & SPACE_ID and ensure labels.csv exists.")
+
 # session state
 st.session_state.setdefault("order", [])
 st.session_state.setdefault("idx", 0)
@@ -187,5 +196,4 @@ else:
                 st.rerun()
 
 st.divider()
-st.caption("Labels are saved to `labels.csv` (and synced to this Space if HF_TOKEN is set).")
-
+st.caption("Labels are written to `/data/labels.csv`. Use **Sync** in the sidebar to push to the Space Files.")