2nd page fail error

app.py

@@ -10,41 +10,63 @@ import secrets, time, string
 
 from functools import lru_cache
 
+def _infer_relpath_from_filename(url_or_name: str) -> str:
+    """
+    url_or_name: videos.json에 들어있는 v["url"] (전체 URL이든 파일명만이든)
+    return: HF dataset 내부에서의 상대경로
+            ex) "video_examples/Wan2.2/Wan2.2/Wan2.2_BodyWeightSquats_01_g01_c01.mp4"
+    규칙:
+      1) basename만 뽑는다.
+      2) prefix = basename.split('_')[0]  (예: "Wan2.2", "Hunyuan_videos", "wan21_videos", "RunwayGen4")
+      3) relpath = f"video_examples/{prefix}/{prefix}/{basename}"
+    """
+    # 1) 혹시 전체 URL이면 맨 마지막 파일명만 추출
+    base = os.path.basename(url_or_name)
+
+    # 2) prefix 는 첫 '_' 이전까지
+    #    예: "Wan2.2_BodyWeightSquats_01_g01_c01.mp4" -> "Wan2.2"
+    #         "Hunyuan_videos_BodyWeightSquats_01_08d40ea1.mp4" -> "Hunyuan_videos"
+    #    만약 '_'가 없다면 전체 이름 자체를 prefix로 쓴다.
+    if "_" in base:
+        prefix = base.split("_")[0]
+    else:
+        prefix = os.path.splitext(base)[0]
+
+    # 3) 최종 상대경로
+    rel = f"video_examples/{prefix}/{prefix}/{base}"
+    return rel
+
+
 @lru_cache(maxsize=2048)
 def get_local_video_path(hf_url_or_relpath: str) -> str:
     """
-    hf_url_or_relpath:
-      - 만약 이미 로컬 경로(/tmp/...mp4 등)이면 그대로 리턴
-      - 만약 'videos/...mp4' 같은 repo 내부 경로면 hf_hub_download로 받아서 로컬 경로 리턴
-      - 만약 https://huggingface.co/... 형태면 파일명만 떼와서 받아서 로컬 리턴
+    1) 이미 로컬이면 그대로 쓴다.
+    2) 아니면 repo에서 hf_hub_download로 받아서 /tmp에 저장하고 음소거본 경로 리턴.
     """
-    # 1) 이미 로컬 파일이면 그대로
+    # 로컬 경로면 바로 사용
     if os.path.exists(hf_url_or_relpath):
         return ensure_muted_copy(hf_url_or_relpath)
 
-    # 2) 허깅페이스 프라이빗 URL일 가능성 -> 파일 basename 뽑기
-    basename = os.path.basename(hf_url_or_relpath)
+    if not HF_TOKEN:
+        raise RuntimeError("HF_TOKEN is not set or empty in this Space. Please add it as a secret.")
 
-    # 우리가 실제 dataset repo에 올려둔 mp4가 예를 들면 "videos/Wan2.2_...mp4" 이런 구조라고 가정
-    # videos.json이 url 대신 repo 내부 상대경로를 담도록 되어 있으면 제일 깔끔함.
-    # 만약 진짜로 "videos/....mp4" 이런 상대경로를 가지고 있다면 그대로 쓰고:
-    possible_rel = hf_url_or_relpath
-    # 혹시 전체 URL이었다면 basename만 쓰는 식으로 단순화:
-    if possible_rel.startswith("http"):
-        possible_rel = f"videos/{basename}"
+    # 우리가 가진 문자열로부터 HF repo 내부 경로를 추론
+    rel = _infer_relpath_from_filename(hf_url_or_relpath)
 
     local_path = hf_hub_download(
         repo_id=REPO_ID,
-        filename=possible_rel,
-        repo_type="dataset",
+        filename=rel,
+        repo_type=REPO_TYPE,
         token=HF_TOKEN,
         local_dir="/tmp",
         local_dir_use_symlinks=False,
     )
-    # 음소거본까지 만들기(네가 이미 ensure_muted_copy() 준비해둠)
+
+    # 무음 버전으로 변환해서 리턴
     return ensure_muted_copy(local_path)
 
 
+
 # 사용자가 '선택'을 완료한 값 (초기엔 None)
 selected_action = gr.State(None)
 selected_phys   = gr.State(None)
@@ -191,7 +213,9 @@ def push_final_feedback(participant_id: str, comment: str):
 
 
 # -------------------- Config --------------------
-REPO_ID = os.getenv("RESULTS_REPO", "dghadiya/video_eval")  # 업로드한 리포와 일치
+# REPO_ID = os.getenv("RESULTS_REPO", "dghadiya/video_eval")  # 업로드한 리포와 일치
+REPO_ID = os.getenv("RESULTS_REPO", "dghadiya/video_examples")
+
 HF_TOKEN = os.getenv("HF_TOKEN")
 RESULTS_FILE = "results.csv"
 TOTAL_PER_PARTICIPANT = 30  # 목표 평가 개수(세션 기준)