Initial commit

app.py

@@ -1,7 +1,17 @@
+import os
+
+import huggingface_hub
+if not hasattr(huggingface_hub, "HfFolder"):
+    class HfFolder:
+        @staticmethod
+        def get_token(): return os.environ.get("HF_TOKEN")
+        @staticmethod
+        def save_token(t): pass
+    huggingface_hub.HfFolder = HfFolder
+
 import gradio as gr
 import pandas as pd
 import requests
-import os
 import csv
 import threading
 from io import BytesIO
@@ -27,9 +37,8 @@ ROOM_CLASSES = ["living_room", "bedroom", "kitchen", "bathroom", "dining_room",
 
 def sync_pull():
     if not HF_TOKEN:
-        print("⚠️ HF_TOKEN not found. Data will not be saved to dataset.")
+        print("⚠️ HF_TOKEN not set. Persistence disabled.")
         return
-
     print(f"🔄 Syncing from {DATASET_REPO_ID}...")
     for filename in ["annotations.csv", "verifications.csv", "skipped.csv"]:
         try:
@@ -46,7 +55,6 @@ def sync_pull():
 
 def sync_push_background(local_path, remote_filename):
     if not HF_TOKEN: return
-    
     def _push():
         try:
             api = HfApi(token=HF_TOKEN)
@@ -60,13 +68,10 @@ def sync_push_background(local_path, remote_filename):
             print(f"☁️ Synced {remote_filename}")
         except Exception as e:
             print(f"❌ Sync failed: {e}")
-
-    thread = threading.Thread(target=_push)
-    thread.start()
+    threading.Thread(target=_push).start()
 
 def init_files():
     sync_pull()
-
     for f in [LABEL_FILE, VERIFY_FILE, SKIP_FILE]:
         if not os.path.exists(f):
             if f == LABEL_FILE: cols = ["timestamp", "user", "group_id", "url", "score", "label"]
@@ -75,7 +80,7 @@ def init_files():
             pd.DataFrame(columns=cols).to_csv(f, index=False)
             
     if not os.path.exists(URL_FILE):
-        print("⚠️ urls.txt not found! Please upload it.")
+        print("⚠️ urls.txt not found in root! Please upload it.")
 
 init_files()
 
@@ -167,7 +172,6 @@ def render_workspace(mode, history, specific_index=None, move_back=False):
         found = False
         for i, gid in enumerate(all_groups):
             if gid in s_done: continue
-            
             is_ready = False
             if mode == "label" and gid not in l_done: is_ready = True
             elif mode == "verify" and gid in l_done and gid not in v_done: is_ready = True
@@ -182,7 +186,6 @@ def render_workspace(mode, history, specific_index=None, move_back=False):
              return {screen_menu: gr.update(visible=True), screen_work: gr.update(visible=False), log_box: "No more tasks found."}
 
     urls = get_group_urls(target_gid)
-    
     if not history or history[-1] != target_gid:
         history.append(target_gid)
         
@@ -195,7 +198,7 @@ def render_workspace(mode, history, specific_index=None, move_back=False):
         for future in futures:
             processed_images[futures[future]] = future.result()
 
-    header = f"# Property #{target_idx + 1} <span style='font-size:14px;color:gray;'>(ID: {target_gid})</span>"
+    header = f"# Property #{target_idx + 1} (ID: {target_gid})"
     
     updates = {
         screen_menu: gr.update(visible=False),
@@ -216,7 +219,6 @@ def render_workspace(mode, history, specific_index=None, move_back=False):
         if i < len(urls):
             u = urls[i]
             img_data = processed_images[i]
-            
             updates[img_c] = gr.update(value=img_data, visible=True)
             v_sc = saved_vals.get(u, {}).get('score', 5)
             v_lbl = saved_vals.get(u, {}).get('label', ROOM_CLASSES[0])
@@ -239,7 +241,6 @@ def render_workspace(mode, history, specific_index=None, move_back=False):
             updates[c_drp] = gr.update(visible=False)
             updates[c_chk] = gr.update(visible=False)
             updates[c_lbl] = gr.update(visible=False)
-
     return updates
 
 def save_data(mode, history, urls, *args):
@@ -247,7 +248,6 @@ def save_data(mode, history, urls, *args):
     gid = history[-1]
     ts = datetime.now().isoformat()
     rows = []
-    
     for i, u in enumerate(urls):
         base = i * 4
         sc, lbl, chk = args[base], args[base+1], args[base+2]
@@ -255,14 +255,11 @@ def save_data(mode, history, urls, *args):
         else: rows.append([ts, "user", gid, u, chk, lbl])
         
     fname = LABEL_FILE if mode == "label" else VERIFY_FILE
-    
     with FileLock(LOCK_FILE):
         with open(fname, "a", newline="") as f:
             csv.writer(f).writerows(rows)
-    
     remote_filename = os.path.basename(fname)
     sync_push_background(fname, remote_filename)
-            
     return render_workspace(mode, history)
 
 def skip_group(idx, history, mode):
@@ -272,7 +269,6 @@ def skip_group(idx, history, mode):
             with open(SKIP_FILE, "a", newline="") as f:
                 csv.writer(f).writerow([datetime.now().isoformat(), "user", gid])
         sync_push_background(SKIP_FILE, "skipped.csv")
-    
     return render_workspace(mode, history, specific_index=idx + 1)
 
 def refresh_cat():
@@ -281,7 +277,6 @@ def refresh_cat():
     except: l_set = set()
     try: v_set = set(pd.read_csv(VERIFY_FILE)['group_id'].unique())
     except: v_set = set()
-    
     data = []
     for i, gid in enumerate(all_gids):
         s = "⚪ Pending"
@@ -291,7 +286,6 @@ def refresh_cat():
     return pd.DataFrame(data, columns=["#", "Status", "ID"])
 
 with gr.Blocks(theme=gr.themes.Soft(), title="Labeling Tool") as demo:
-    
     state_mode = gr.State("label")
     state_hist = gr.State([])
     state_urls = gr.State([])
@@ -305,7 +299,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Labeling Tool") as demo:
         with gr.Tab("Workspace", id=0):
             with gr.Group() as screen_menu:
                 gr.Markdown("# Welcome! 👋")
-                gr.Markdown("Connected to HF Dataset for free persistence.")
+                gr.Markdown("Persistent & Compatible Version (Gradio 3.x)")
                 with gr.Row():
                     b_start_l = gr.Button("Start Labeling", variant="primary")
                     b_start_v = gr.Button("Start Verification")
@@ -313,7 +307,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Labeling Tool") as demo:
             with gr.Group(visible=False) as screen_work:
                 header_md = gr.Markdown()
                 img_objs, input_objs = [], []
-                
                 with gr.Row():
                     for i in range(MAX_IMAGES):
                         with gr.Column(min_width=250):
@@ -325,7 +318,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Labeling Tool") as demo:
                                 chk = gr.Checkbox(label="Correct?", value=True, visible=False)
                             img_objs.append(img)
                             input_objs.extend([sld, drp, chk, lbl])
-
                 with gr.Row():
                     b_back = gr.Button("⬅ Back")
                     b_skip = gr.Button("Skip ➡")
@@ -344,16 +336,12 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Labeling Tool") as demo:
     
     b_start_l.click(lambda: "label", None, state_mode).then(render_workspace, [state_mode, state_hist], ALL_IO)
     b_start_v.click(lambda: "verify", None, state_mode).then(render_workspace, [state_mode, state_hist], ALL_IO)
-    
     b_save.click(save_data, [state_mode, state_hist, state_urls] + input_objs, ALL_IO)
     b_skip.click(skip_group, [state_idx, state_hist, state_mode], ALL_IO)
     b_back.click(lambda m, h: render_workspace(m, h, move_back=True), [state_mode, state_hist], ALL_IO)
-    
     btn_home.click(lambda: {screen_menu: gr.update(visible=True), screen_work: gr.update(visible=False), state_hist: []}, None, [screen_menu, screen_work, state_hist])
-    
     b_go_l.click(lambda: "label", None, state_mode).then(lambda n, m, h: render_workspace(m, h, specific_index=int(n)-1), [num_in, state_mode, state_hist], ALL_IO)
     b_go_v.click(lambda: "verify", None, state_mode).then(lambda n, m, h: render_workspace(m, h, specific_index=int(n)-1), [num_in, state_mode, state_hist], ALL_IO)
-
     b_ref_cat.click(refresh_cat, None, df_cat)
     demo.load(refresh_cat, None, df_cat).then(get_stats_text, None, top_stats)
 

requirements.txt

@@ -1,5 +1,6 @@
-gradio==4.44.1
-huggingface_hub==0.24.6
+gradio==3.50.2
+huggingface_hub==0.23.0
+pydantic<2.0
 pandas
 requests
 Pillow