Update app.py

app.py

@@ -14,7 +14,7 @@ from huggingface_hub import HfApi, CommitOperationAdd
 # ----------------------
 # Configuration
 # ----------------------
-# --- HF Repo (optional) ---
+# --- HF Repo ---
 HF_RESULTS_REPO = os.getenv("HF_RESULTS_REPO")
 HF_RESULTS_REPO_TYPE = "dataset"
 HF_TOKEN = os.getenv("HF_TOKEN")
@@ -27,18 +27,17 @@ CONTACT_EMAIL = "ffallah@asu.edu"
 
 # --- Paths ---
 # We still use the JSON file to get the list of filenames, 
-# even though we aren't using the caption text anymore.
 CAPTIONS_JSON_PATH = os.environ.get("CAPTIONS_JSON_PATH", "data/captions.json")
 
 # Folders with matching filenames across all FIVE folders:
-GT_MASKED_DIR   = os.environ.get("GT_MASKED_DIR", "data/gt_masked")    # Image 1
-GT_UNMASKED_DIR = os.environ.get("GT_UNMASKED_DIR", "data/gt")        # Image 2
-SR_DIR          = os.environ.get("SR_DIR", "data/sr")               # Image 3
-ORIGINAL_DIR    = os.environ.get("ORIGINAL_DIR", "data/lr")     # Image 4
-IMAGE_5_DIR     = os.environ.get("ORIGINAL_DIR", "data/lr")  # Image 5 (New)
+GT_MASKED_DIR   = "data/gt_b"   # Image 1
+GT_UNMASKED_DIR = "data/adc_b"  # Image 2
+SR_DIR          = "data/sr_b"   # Image 3
+ORIGINAL_DIR    = "data/lr_b"   # Image 4
+IMAGE_5_DIR     = "data/see_b"  # Image 5
 
 # --- Results ---
-RESULTS_DIR = os.environ.get("RESULTS_DIR", "results")
+RESULTS_DIR = "results"
 PROGRESS_PATH = os.path.join(RESULTS_DIR, "progress.json")
 ALL_RESULTS_JSONL = os.path.join(RESULTS_DIR, "all_results.jsonl")
 SAVE_PII = True # Set to False to anonymize saved name/email
@@ -363,7 +362,7 @@ def _save_record_and_progress(
     os.makedirs(RESULTS_DIR, exist_ok=True)
     append_jsonl(os.path.join(RESULTS_DIR, f"{uid}.jsonl"), record)
     append_jsonl(ALL_RESULTS_JSONL, record)
-    # push_results_to_private_repo(uid) # Optional: uncomment if using HF Hub
+    push_results_to_private_repo(uid) 
 
     seen.add(sample.sample_id)
     progress[uid]["seen"] = sorted(list(seen))
@@ -671,9 +670,26 @@ with gr.Blocks(title="RTS Human Evaluation", theme=gr.themes.Soft()) as demo:
     )
 
 if __name__ == "__main__":
+    # --- DYNAMIC READ FROM HF_RESULTS_REPO ---
+    if HF_RESULTS_REPO:
+        from huggingface_hub import snapshot_download
+        # print(f"Reading images and metadata from: {HF_RESULTS_REPO}...")
+        try:
+            # This pulls your repo's 'data' folder into the current workspace
+            snapshot_download(
+                repo_id=HF_RESULTS_REPO,
+                repo_type="dataset",
+                local_dir=".",            
+                allow_patterns="data/*",  
+                token=HF_TOKEN
+            )
+        except Exception as e:
+            print(f"Error reading from HF: {e}")
+
     ensure_paths()
-    # Pre-check dataset load
+    # Pre-check dataset load from the newly downloaded files
     _ = load_dataset(CAPTIONS_JSON_PATH, GT_MASKED_DIR, GT_UNMASKED_DIR, SR_DIR, ORIGINAL_DIR, IMAGE_5_DIR)
-    print("✅ Dataset loaded successfully. Launching app.")
+    
+    print("✅ Successfully synced with HF Repo. Launching app.")
     demo.queue()
     demo.launch()