updated app.py

app.py

@@ -16,7 +16,8 @@ from huggingface_hub import HfApi, hf_hub_download
 DATASET_REPO_ID = os.environ.get("DATASET_REPO_ID", "fast-stager/property-labels") 
 HF_TOKEN = os.environ.get("HF_TOKEN") 
 
-CACHE_DIR = "/tmp"
+CACHE_DIR = os.path.join(os.getcwd(), "data")
+os.makedirs(CACHE_DIR, exist_ok=True)
 URL_FILE = "new_urls.json" 
 LABEL_FILE = os.path.join(CACHE_DIR, "annotations.csv")
 VERIFY_FILE = os.path.join(CACHE_DIR, "verifications.csv")
@@ -34,8 +35,7 @@ def sync_pull():
                 filename=filename,
                 repo_type="dataset",
                 local_dir=CACHE_DIR,
-                token=token,
-                local_dir_use_symlinks=False
+                token=token
             )
             print(f"✅ Loaded {filename}")
         except Exception as e:
@@ -102,17 +102,15 @@ def load_all_urls():
             data = json.load(f)
             if "groups" in data:
                 for group in data["groups"]:
-                    if "images" in group:
-                        urls.extend(group["images"])
+                    urls.extend(group.get("images", []))
             elif isinstance(data, dict):
-                for query_key, rows in data.items():
+                for rows in data.values():
                     if isinstance(rows, list):
                         for row in rows:
                             if "unstaged_images" in row:
                                 urls.append(row["unstaged_images"])
     except Exception as e:
         print(f"Error loading URLs: {e}")
-        
     return urls
 
 def get_ordered_groups():