Update app.py

app.py

@@ -1028,14 +1028,14 @@ FAQ = [  # order follows the review: what it is, free access, how to, API, then
 ]
 
 # ---------- example grid ("Compare facial expression edits") ----------
-# Clips live in the model repo under drama_examples/<n>/:
+# Clips live in the model repo under drama-examples/<n>/ (drama_examples/ also works):
 #   <n>_original.<ext>                 the untouched clip
 #   <n>_<expression>[_<intensity>].<ext>  a DRAMA output, e.g. 2_happy_0.6.mp4 or 6_sad.mp4
 # They are copied into the Space at startup (needs HF_TOKEN with read access while the repo is private),
 # remuxed to MP4 with the index at the front (no re-encoding, so quality is unchanged), and given a
 # still frame, so the page only downloads a clip when someone presses play.
 EXAMPLES_REPO = os.environ.get("EXAMPLES_REPO", "trydrama-neuralgarage/DRAMA")
-EXAMPLES_FOLDER = "drama_examples"
+EXAMPLES_FOLDERS = ["drama-examples", "drama_examples"]  # either spelling works
 EXAMPLES_VISIBLE = 3                                  # rows shown before "Show more examples"
 EX_DOWNLOAD = os.path.abspath("example_cache")        # copy of the repo folder
 EX_WEB = os.path.abspath("example_web")               # browser-ready MP4s and still frames
@@ -1076,15 +1076,20 @@ def _web_ready(src, n):
 
 def load_examples():
     """Returns [(number, [(label, intensity, video_path, poster_path), ...]), ...]."""
-    base = os.path.abspath(EXAMPLES_FOLDER)          # a copy uploaded into the Space itself wins
-    if not os.path.isdir(base):
+    # a copy uploaded into the Space itself wins; otherwise copy it from the model repo
+    base = next((os.path.abspath(f) for f in EXAMPLES_FOLDERS if os.path.isdir(f)), None)
+    if base is None:
         try:
             from huggingface_hub import snapshot_download
             snapshot_download(EXAMPLES_REPO, repo_type="model", token=HF_TOKEN or None, local_dir=EX_DOWNLOAD,
-                              allow_patterns=[f"{EXAMPLES_FOLDER}/*/*"])
-            base = os.path.join(EX_DOWNLOAD, EXAMPLES_FOLDER)
+                              allow_patterns=[f"{f}/*/*" for f in EXAMPLES_FOLDERS])
+            base = next((os.path.join(EX_DOWNLOAD, f) for f in EXAMPLES_FOLDERS
+                         if os.path.isdir(os.path.join(EX_DOWNLOAD, f))), "")
+            if not base:
+                log.warning(f"[examples] no drama-examples/ folder found in {EXAMPLES_REPO}")
+                return []
         except Exception as e:
-            log.warning(f"[examples] couldn't copy {EXAMPLES_FOLDER}/ from {EXAMPLES_REPO} "
+            log.warning(f"[examples] couldn't copy drama-examples/ from {EXAMPLES_REPO} "
                         f"({type(e).__name__}: {short(str(e), 200)}). HF_TOKEN needs read access to that repo.")
             return []
     rows = []
@@ -1852,5 +1857,5 @@ def favicon_file():
 
 startup_check()
 demo.queue(max_size=20).launch(theme=THEME, css=CSS, js=FORCE_DARK, head=HEAD,
-                               allowed_paths=[EX_WEB, EX_DOWNLOAD, os.path.abspath(EXAMPLES_FOLDER)],
+                               allowed_paths=[EX_WEB, EX_DOWNLOAD, *(os.path.abspath(f) for f in EXAMPLES_FOLDERS)],
                                favicon_path=favicon_file(), footer_links=[])