2nd page error

app.py

@@ -18,22 +18,28 @@ TOP_DIRS = [
     "Wan2.2",
     "wan21_videos",
 ]
-
 def _choose_prefix_from_filename(base: str) -> str:
     """
-    base = 'Opensora_768_PushUps_03_124bba2d.mp4' 같은 실제 파일 이름.
-    위에서 정의한 top-level 디렉토리 후보들 중
-    base가 "<dir>_" 혹은 "<dir>-" 혹은 "<dir>." 로 시작하면 그걸 prefix로 채택.
-    (실제로는 '_' 패턴이면 충분할 가능성이 큼)
+    base 예:
+      'wan21_videos_BodyWeightSquats_06_10c02b51.mp4'
+      'Wan2.2_BodyWeightSquats_03_g09_c02.mp4'
+      'Opensora_768_PushUps_03_124bba2d.mp4'
+
+    리포의 실제 상위 폴더명(TOP_DIRS) 중
+    파일명이 그 폴더명으로 시작하면 그걸 prefix로 쓴다.
+    가장 긴 매칭을 우선으로 한다.
     """
+    best = None
     for d in TOP_DIRS:
-        # 일반적으로 "<dir>_" 로 시작함
-        if base.startswith(d + "_"):
-            return d
-        # 혹시 그냥 정확히 같은 이름만 있을 수도 있으니까(안전장치)
-        if base.startswith(d):
-            return d
-    # fallback: 이전 단순 로직 (한 번도 못 맞췄을 때 대비)
+        # 제일 흔한 패턴은 "<dir>_" 로 시작하는 거라서 이걸 먼저 본다
+        if base.startswith(d + "_") or base.startswith(d + "/") or base.startswith(d + "-") or base.startswith(d):
+            if best is None or len(d) > len(best):
+                best = d
+
+    if best is not None:
+        return best
+
+    # fallback (거의 안 타야 정상)
     if "_" in base:
         return base.split("_")[0]
     return os.path.splitext(base)[0]
@@ -43,25 +49,19 @@ def _candidate_relpaths(url_or_name: str):
     # 파일명만 뽑고 공백/개행 제거
     base = os.path.basename(url_or_name).strip()
 
-    # prefix 결정: 실제 repo 상위 폴더 이름과 맞춰주기
     prefix = _choose_prefix_from_filename(base)
 
-    # 일부 모델(Hunyuan_videos, Wan2.2 등)은
-    #   video_examples/<prefix>/<prefix>/<file.mp4>
-    # 이런 구조(2단계 중첩)
-    # Opensora_768도 실제로 보면?
-    #   video_examples/Opensora_768/Opensora_768/파일.mp4 인지
-    #   video_examples/Opensora_768/파일.mp4 인지 확인 필요.
-    # 방금 Wan2.2 스샷은 2단계 중첩 (Wan2.2/Wan2.2/파일).
-    # 너가 Opensora_768 쪽 스샷은 아직 안보여줬지만,
-    # 지금 에러에서 우리는 <prefix>/<prefix>/<file>은 시도 안 해서 죽은 게 아니고
-    # prefix 자체가 틀려서 죽은거라 둘 다 시도해도 안전해.
-
+    # 예:
+    # video_examples/Wan2.2/Wan2.2/Wan2.2_BodyWeightSquats_03_....mp4
+    # video_examples/Hunyuan_videos/Hunyuan_videos/Hunyuan_videos_....mp4
+    # video_examples/Opensora_768/Opensora_768/...mp4  또는
+    # video_examples/Opensora_768/...mp4 (둘 다 가능성)
+    #
+    # 그래서 여러 패턴을 다 시도해본다.
     alt_prefix_videos = (
         prefix if prefix.endswith("_videos") else (prefix + "_videos")
     )
 
-    # 최종적으로 가능한 모든 패턴을 다 시도
     candidates = [
         f"video_examples/{prefix}/{prefix}/{base}",
         f"video_examples/{prefix}/{alt_prefix_videos}/{base}",
@@ -77,6 +77,7 @@ def _candidate_relpaths(url_or_name: str):
     return uniq
 
 
+
 @lru_cache(maxsize=2048)
 def get_local_video_path(hf_url_or_relpath: str) -> str:
     # if already local, just ensure it's muted