Update app.py

app.py

@@ -11,19 +11,24 @@ from huggingface_hub import HfApi
 
 # ---------------- Config ----------------
 IMAGE_DIR   = os.getenv("IMAGE_DIR", "images")
-
-# ✅ 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")
 
+# pick a writable place for labels.csv
+DATA_DIR = "/data"
+DEFAULT_LABELS = (
+    f"{DATA_DIR}/labels.csv"
+    if os.path.isdir(DATA_DIR) and os.access(DATA_DIR, os.W_OK)
+    else "labels.csv"
+)
+LABELS_CSV  = os.getenv("LABELS_CSV", DEFAULT_LABELS)
+
 # Label names
 LABEL_NONBLAST  = os.getenv("LABEL_NONBLAST", "NON-BLAST")
 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 (manual button below)
+# (Optional) Hugging Face sync
 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()
@@ -53,8 +58,12 @@ 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."""
-    os.makedirs(os.path.dirname(LABELS_CSV) or ".", exist_ok=True)
+    """Append one row to labels.csv and (optionally) push to the Space repo."""
+    # only create a directory if LABELS_CSV actually has one, and it's not already there
+    dirpath = os.path.dirname(LABELS_CSV)
+    if dirpath and not os.path.isdir(dirpath):
+        os.makedirs(dirpath, exist_ok=True)
+
     record = {
         "image": rel_to_image_dir(image_path),
         "label": label,
@@ -65,8 +74,18 @@ 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)
 
-    # ❌ removed auto upload to the Space (this caused rebuilds)
-    # If you want to sync, use the manual button in the sidebar.
+    # 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)
 
 # --------------- UI ---------------------
 st.title("🏷️ Blast Cell Labeling App")
@@ -93,24 +112,6 @@ 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)
@@ -196,4 +197,4 @@ else:
                 st.rerun()
 
 st.divider()
-st.caption("Labels are written to `/data/labels.csv`. Use **Sync** in the sidebar to push to the Space Files.")
+st.caption(f"Labels are saved to `{LABELS_CSV}` (and synced to this Space if HF_TOKEN is set).")