suport for repo

add repo , previous veriosn works well

app.py

@@ -1,6 +1,5 @@
 import os
 import math
-import time
 import random
 import tempfile
 import gradio as gr
@@ -9,7 +8,7 @@ from huggingface_hub import HfApi, hf_hub_download
 # ======================================================
 # CONFIG
 # ======================================================
-REPO_ID = "rahul7star/WAN22-Mar-2026"
+DEFAULT_REPO = "rahul7star/WAN22-Mar-2026"
 FREE_REPO = "rahul7star/ohamlab"
 FREE_IMAGE_DIR = "showcase/image"
 
@@ -22,10 +21,10 @@ if not APP_PASSWORD:
     raise RuntimeError("APP_PASSWORD not set")
 
 api = HfApi()
-
 PAGE_SIZE = 6
+
 tmp_dir = tempfile.mkdtemp()
-images_cache = {}
+images_cache = {}   # (repo, date) -> images
 
 # ======================================================
 # FREE PREVIEW
@@ -56,10 +55,15 @@ def load_free_images():
 FREE_IMAGES = load_free_images()
 
 # ======================================================
-# DATE + CACHE HELPERS
+# HELPERS
 # ======================================================
-def list_date_folders(last_n=20):
-    files = api.list_repo_files(REPO_ID, token=HF_TOKEN)
+def clear_cache():
+    images_cache.clear()
+    print("[CACHE] cleared all")
+
+
+def list_date_folders(repo, last_n=20):
+    files = api.list_repo_files(repo, token=HF_TOKEN)
     dates = sorted(
         {f.split("/")[0] for f in files if "/" in f},
         reverse=True
@@ -67,17 +71,12 @@ def list_date_folders(last_n=20):
     return dates[:last_n]
 
 
-def clear_date_cache(date):
-    if date in images_cache:
-        del images_cache[date]
-        print(f"[CACHE] cleared {date}")
-
-
-def download_images_for_date(date):
-    if date in images_cache:
-        return images_cache[date]
+def download_images(repo, date):
+    key = (repo, date)
+    if key in images_cache:
+        return images_cache[key]
 
-    files = api.list_repo_files(REPO_ID, token=HF_TOKEN)
+    files = api.list_repo_files(repo, token=HF_TOKEN)
     local = []
 
     for f in files:
@@ -86,51 +85,59 @@ def download_images_for_date(date):
         if not f.lower().endswith((".png", ".jpg", ".jpeg", ".webp")):
             continue
 
-        dst_dir = os.path.join(tmp_dir, os.path.dirname(f))
-        os.makedirs(dst_dir, exist_ok=True)
+        dst = os.path.join(tmp_dir, repo.replace("/", "_"), f)
+        os.makedirs(os.path.dirname(dst), exist_ok=True)
 
         try:
             path = hf_hub_download(
-                REPO_ID,
+                repo,
                 f,
                 token=HF_TOKEN,
-                local_dir=dst_dir
+                local_dir=os.path.dirname(dst),
             )
             local.append(path)
-            print("[DOWNLOAD]", f, "->", path)
+            print("[DOWNLOAD]", repo, f)
         except Exception as e:
             print("[DOWNLOAD ERROR]", f, e)
 
-    images_cache[date] = local
-    print(f"[INFO] {date} images:", len(local))
+    images_cache[key] = local
+    print(f"[INFO] {repo} {date}: {len(local)} images")
     return local
 
 
 # ======================================================
 # PAGINATION
 # ======================================================
-def render_page(date, page):
-    images = download_images_for_date(date)
-
+def render_page(repo, date, page):
+    images = download_images(repo, date)
     total_pages = max(1, math.ceil(len(images) / PAGE_SIZE))
-    page = max(0, min(int(page), total_pages - 1))
 
+    page = max(0, min(int(page), total_pages - 1))
     start = page * PAGE_SIZE
     end = start + PAGE_SIZE
-    subset = images[start:end]
 
-    print(f"[RENDER] {date} page {page+1}/{total_pages}")
+    return (
+        images[start:end],
+        page,
+        f"Page {page+1} / {total_pages}",
+    )
+
 
-    return subset, page, f"Page {page+1} / {total_pages}"
+def nav(delta, repo, date, page):
+    return render_page(repo, date, page + delta)
 
 
-def nav(delta, date, page):
-    return render_page(date, page + delta)
+def change_date(repo, date):
+    clear_cache()
+    return render_page(repo, date, 0)
 
 
-def change_date(date):
-    clear_date_cache(date)
-    return render_page(date, 0)
+def change_repo(repo):
+    clear_cache()
+    dates = list_date_folders(repo)
+    first = dates[0] if dates else ""
+    imgs, page, info = render_page(repo, first, 0)
+    return dates, imgs, page, first, info
 
 
 # ======================================================
@@ -141,27 +148,24 @@ def check_password(pwd):
         return (
             gr.update(visible=True),
             gr.update(visible=False),
-            gr.update(visible=True),
             [],
             0,
             "",
             "❌ Wrong password",
         )
 
-    dates = list_date_folders()
+    dates = list_date_folders(DEFAULT_REPO)
     latest = dates[0]
 
-    imgs, page, info = render_page(latest, 0)
+    imgs, page, info = render_page(DEFAULT_REPO, latest, 0)
 
     return (
         gr.update(visible=False),
         gr.update(visible=True),
-        gr.update(visible=False),
         imgs,
         page,
         latest,
         info,
-        "✅ Access granted",
     )
 
 
@@ -170,7 +174,6 @@ def check_password(pwd):
 # ======================================================
 css = """
 footer, .gradio-footer { display:none!important; }
-
 #ohamlab-footer {
   position:fixed;
   bottom:0;
@@ -189,89 +192,100 @@ footer, .gradio-footer { display:none!important; }
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
     gr.Markdown("# 🔐 OhamLab Image Showcase (PRO)")
 
-    # -------- LOGIN --------
+    # ---------- LOGIN ----------
     with gr.Column(visible=True) as login_box:
         pwd = gr.Textbox(type="password", label="Password")
         login_btn = gr.Button("Unlock")
-        status = gr.Markdown()
 
         gr.Markdown("### 🎁 Free Preview")
         with gr.Row():
             for img in FREE_IMAGES:
                 gr.Image(img, show_label=False)
 
-    # -------- APP --------
+    # ---------- APP ----------
     with gr.Column(visible=False) as app:
-        current_date = gr.State("")
+        repo_state = gr.State(DEFAULT_REPO)
+        date_state = gr.State("")
         page_state = gr.State(0)
 
-        date_dropdown = gr.Dropdown(
-            choices=list_date_folders(),
-            label="Select date"
+        repo_input = gr.Textbox(
+            value=DEFAULT_REPO,
+            label="Base HuggingFace Repo",
         )
 
+        date_dropdown = gr.Dropdown(label="Select Date")
+
         manual_date = gr.Textbox(
             label="Or enter date manually (YYYY-MM-DD)"
         )
 
-        gallery = gr.Gallery(
-            label="Images",
-            columns=3,
-            height="auto"
-        )
-
+        gallery = gr.Gallery(columns=3, height="auto")
         page_info = gr.Markdown()
 
         with gr.Row():
             prev_btn = gr.Button("⬅ Previous")
             next_btn = gr.Button("Next ➡")
 
-    # -------- EVENTS --------
+    # ---------- EVENTS ----------
     login_btn.click(
         check_password,
         inputs=pwd,
         outputs=[
             login_box,
             app,
-            login_box,
             gallery,
             page_state,
-            current_date,
+            date_state,
             page_info,
-            status,
         ],
     )
 
-    prev_btn.click(
-        lambda d, p: nav(-1, d, p),
-        inputs=[current_date, page_state],
-        outputs=[gallery, page_state, page_info],
-    )
-
-    next_btn.click(
-        lambda d, p: nav(1, d, p),
-        inputs=[current_date, page_state],
-        outputs=[gallery, page_state, page_info],
+    repo_input.change(
+        change_repo,
+        inputs=repo_input,
+        outputs=[
+            date_dropdown,
+            gallery,
+            page_state,
+            date_state,
+            page_info,
+        ],
+    ).then(
+        lambda r: r,
+        inputs=repo_input,
+        outputs=repo_state,
     )
 
     date_dropdown.change(
-        change_date,
-        inputs=date_dropdown,
+        lambda r, d: change_date(r, d),
+        inputs=[repo_state, date_dropdown],
         outputs=[gallery, page_state, page_info],
     ).then(
         lambda d: d,
         inputs=date_dropdown,
-        outputs=current_date,
+        outputs=date_state,
     )
 
     manual_date.submit(
-        change_date,
-        inputs=manual_date,
+        lambda r, d: change_date(r, d),
+        inputs=[repo_state, manual_date],
         outputs=[gallery, page_state, page_info],
     ).then(
         lambda d: d,
         inputs=manual_date,
-        outputs=current_date,
+        outputs=date_state,
+    )
+
+    prev_btn.click(
+        lambda r, d, p: nav(-1, r, d, p),
+        inputs=[repo_state, date_state, page_state],
+        outputs=[gallery, page_state, page_info],
+    )
+
+    next_btn.click(
+        lambda r, d, p: nav(1, r, d, p),
+        inputs=[repo_state, date_state, page_state],
+        outputs=[gallery, page_state, page_info],
     )
 
     gr.HTML("<div id='ohamlab-footer'>© OhamLab Copyright</div>")