2nd page error

app.py

@@ -11,27 +11,71 @@ import secrets, time, string
 from functools import lru_cache
 
 
+TOP_DIRS = [
+    "Hunyuan_videos",
+    "Opensora_768",
+    "RunwayGen4",
+    "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로 채택.
+    (실제로는 '_' 패턴이면 충분할 가능성이 큼)
+    """
+    for d in TOP_DIRS:
+        # 일반적으로 "<dir>_" 로 시작함
+        if base.startswith(d + "_"):
+            return d
+        # 혹시 그냥 정확히 같은 이름만 있을 수도 있으니까(안전장치)
+        if base.startswith(d):
+            return d
+    # fallback: 이전 단순 로직 (한 번도 못 맞췄을 때 대비)
+    if "_" in base:
+        return base.split("_")[0]
+    return os.path.splitext(base)[0]
+
+
 def _candidate_relpaths(url_or_name: str):
-    # --- sanitize filename ---
+    # 파일명만 뽑고 공백/개행 제거
     base = os.path.basename(url_or_name).strip()
 
-    # prefix = "Wan2.2", "Hunyuan_videos", "RunwayGen4", "wan21_videos", ...
-    if "_" in base:
-        prefix = base.split("_")[0]
-    else:
-        prefix = os.path.splitext(base)[0]
-
-    # sometimes folders are <prefix>/<prefix>/..., sometimes <prefix>/...,
-    # sometimes <prefix>/<prefix_videos>/...
-    alt_prefix_videos = prefix if prefix.endswith("_videos") else (prefix + "_videos")
+    # 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 자체가 틀려서 죽은거라 둘 다 시도해도 안전해.
+
+    alt_prefix_videos = (
+        prefix if prefix.endswith("_videos") else (prefix + "_videos")
+    )
 
-    return [
+    # 최종적으로 가능한 모든 패턴을 다 시도
+    candidates = [
         f"video_examples/{prefix}/{prefix}/{base}",
         f"video_examples/{prefix}/{alt_prefix_videos}/{base}",
         f"video_examples/{prefix}/{base}",
         f"video_examples/{base}",
     ]
 
+    # 중복 제거
+    uniq = []
+    for c in candidates:
+        if c not in uniq:
+            uniq.append(c)
+    return uniq
+
 
 @lru_cache(maxsize=2048)
 def get_local_video_path(hf_url_or_relpath: str) -> str: