stable base

stable version to show images

app.py

@@ -1,192 +1,293 @@
 import os
+import math
+import random
 import tempfile
-from functools import lru_cache
-from huggingface_hub import HfApi, hf_hub_download
 import gradio as gr
+from huggingface_hub import HfApi, hf_hub_download
+
+# ======================================================
+# CONFIG
+# ======================================================
+DEFAULT_REPO = "rahul7star/WAN22-Mar-2026"
+FREE_REPO = "rahul7star/ohamlab"
+FREE_IMAGE_DIR = "showcase/image"
 
 HF_TOKEN = os.getenv("HF_TOKEN")
+APP_PASSWORD = os.getenv("APP_PASSWORD")
 
-DEFAULT_REPO = "rahul7star/WAN22-Feb-2026"
-IMAGES_PER_PAGE = 6
+if not HF_TOKEN:
+    raise RuntimeError("HF_TOKEN not set")
+if not APP_PASSWORD:
+    raise RuntimeError("APP_PASSWORD not set")
 
-api = HfApi(token=HF_TOKEN)
+api = HfApi()
+PAGE_SIZE = 6
 
+tmp_dir = tempfile.mkdtemp()
+images_cache = {}   # (repo, date) -> images
 
-# -------------------------------------------------
-# Helpers
-# -------------------------------------------------
+# ======================================================
+# FREE PREVIEW
+# ======================================================
+def load_free_images():
+    tmp = tempfile.mkdtemp()
+    files = api.list_repo_files(FREE_REPO, token=HF_TOKEN)
 
-def list_date_folders(repo_id):
-    files = api.list_repo_files(repo_id=repo_id)
-    return sorted(
-        {f.split("/")[0] for f in files if "/" in f},
-        reverse=True
-    )
+    imgs = [
+        f for f in files
+        if f.startswith(FREE_IMAGE_DIR)
+        and f.lower().endswith((".png", ".jpg", ".jpeg", ".webp"))
+    ]
 
+    random.shuffle(imgs)
+    imgs = imgs[:5]
 
-def list_image_folders(repo_id, date):
-    files = api.list_repo_files(repo_id=repo_id)
-    return sorted(
-        {
-            "/".join(f.split("/")[:2])
-            for f in files
-            if f.startswith(date + "/") and f.count("/") >= 1
-        }
-    )
+    out = []
+    for f in imgs:
+        try:
+            out.append(hf_hub_download(FREE_REPO, f, token=HF_TOKEN, local_dir=tmp))
+        except Exception as e:
+            print("[FREE IMG ERROR]", e)
 
+    return out
 
-def find_image_file(repo_id, folder):
-    files = api.list_repo_files(repo_id=repo_id)
-    for f in files:
-        if f.startswith(folder) and f.lower().endswith((".png", ".jpg", ".jpeg", ".webp")):
-            return f
-    return None
 
+FREE_IMAGES = load_free_images()
 
-def find_summary_file(repo_id, folder):
-    files = api.list_repo_files(repo_id=repo_id)
-    for f in files:
-        if f.startswith(folder) and f.endswith("summary.txt"):
-            return f
-    return None
+# ======================================================
+# HELPERS
+# ======================================================
+def clear_cache():
+    images_cache.clear()
+    print("[CACHE] cleared all")
 
 
-def download_file(repo_id, path):
-    return hf_hub_download(
-        repo_id=repo_id,
-        filename=path,
-        token=HF_TOKEN,
-        cache_dir=tempfile.mkdtemp()
+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
     )
+    return dates[:last_n]
 
 
-# -------------------------------------------------
-# Cache (refreshed per repo + date)
-# -------------------------------------------------
+def download_images(repo, date):
+    key = (repo, date)
+    if key in images_cache:
+        return images_cache[key]
 
-@lru_cache(maxsize=32)
-def load_date(repo_id, date):
-    items = []
+    files = api.list_repo_files(repo, token=HF_TOKEN)
+    local = []
 
-    for folder in list_image_folders(repo_id, date):
-        img_path = find_image_file(repo_id, folder)
-        if not img_path:
+    for f in files:
+        if not f.startswith(date + "/"):
+            continue
+        if not f.lower().endswith((".png", ".jpg", ".jpeg", ".webp")):
             continue
 
-        local_img = download_file(repo_id, img_path)
+        dst = os.path.join(tmp_dir, repo.replace("/", "_"), f)
+        os.makedirs(os.path.dirname(dst), exist_ok=True)
+
+        try:
+            path = hf_hub_download(
+                repo,
+                f,
+                token=HF_TOKEN,
+                local_dir=os.path.dirname(dst),
+            )
+            local.append(path)
+            print("[DOWNLOAD]", repo, f)
+        except Exception as e:
+            print("[DOWNLOAD ERROR]", f, e)
+
+    images_cache[key] = local
+    print(f"[INFO] {repo} {date}: {len(local)} images")
+    return local
+
+
+# ======================================================
+# PAGINATION
+# ======================================================
+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))
+    start = page * PAGE_SIZE
+    end = start + PAGE_SIZE
+
+    return (
+        images[start:end],
+        page,
+        f"Page {page+1} / {total_pages}",
+    )
 
-        summary_path = find_summary_file(repo_id, folder)
-        summary_text = ""
-        if summary_path:
-            local_summary = download_file(repo_id, summary_path)
-            with open(local_summary, "r", encoding="utf-8") as f:
-                summary_text = f.read().strip()
 
-        items.append((local_img, summary_text))
+def nav(delta, repo, date, page):
+    return render_page(repo, date, page + delta)
 
-    return items
 
+def change_date(repo, date):
+    clear_cache()
+    return render_page(repo, date, 0)
 
-def render_page(repo_id, date, page):
-    all_items = load_date(repo_id, date)
-    total_pages = max(1, (len(all_items) + IMAGES_PER_PAGE - 1) // IMAGES_PER_PAGE)
 
-    page = max(0, min(page, total_pages - 1))
-    start = page * IMAGES_PER_PAGE
-    end = start + IMAGES_PER_PAGE
+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
 
-    return all_items[start:end], page, total_pages
 
+# ======================================================
+# AUTH
+# ======================================================
+def check_password(pwd):
+    if pwd != APP_PASSWORD:
+        return (
+            gr.update(visible=True),
+            gr.update(visible=False),
+            [],
+            0,
+            "",
+            "❌ Wrong password",
+        )
 
-def nav(delta, repo_id, date, page):
-    return (*render_page(repo_id, date, page + delta),)
+    dates = list_date_folders(DEFAULT_REPO)
+    latest = dates[0]
 
+    imgs, page, info = render_page(DEFAULT_REPO, latest, 0)
 
-def refresh(repo_id, date):
-    load_date.cache_clear()
-    return render_page(repo_id, date, 0)
+    return (
+        gr.update(visible=False),
+        gr.update(visible=True),
+        imgs,
+        page,
+        latest,
+        info,
+    )
 
 
-# -------------------------------------------------
-# UI
-# -------------------------------------------------
-
-with gr.Blocks(css="""
-footer {visibility: hidden;}
-#copyright {
-  text-align: center;
-  font-size: 12px;
-  color: #888;
-  margin-top: 12px;
+# ======================================================
+# CSS
+# ======================================================
+css = """
+footer, .gradio-footer { display:none!important; }
+#ohamlab-footer {
+  position:fixed;
+  bottom:0;
+  width:100%;
+  background:#f8f9fb;
+  font-size:12px;
+  padding:8px;
+  border-top:1px solid #e5e7eb;
+  text-align:center;
 }
-""") as demo:
+"""
 
-    gr.Markdown("## 📸 Image Browser (Private HF Repo)")
+# ======================================================
+# UI
+# ======================================================
+with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
+    gr.Markdown("# 🔐 OhamLab Image Showcase (PRO)")
+
+    # ---------- LOGIN ----------
+    with gr.Column(visible=True) as login_box:
+        pwd = gr.Textbox(type="password", label="Password")
+        login_btn = gr.Button("Unlock")
+
+        gr.Markdown("### 🎁 Free Preview")
+        with gr.Row():
+            for img in FREE_IMAGES:
+                gr.Image(img, show_label=False)
+
+    # ---------- APP ----------
+    with gr.Column(visible=False) as app:
+        repo_state = gr.State(DEFAULT_REPO)
+        date_state = gr.State("")
+        page_state = gr.State(0)
 
-    with gr.Row():
         repo_input = gr.Textbox(
             value=DEFAULT_REPO,
-            label="Repository ID"
+            label="Base HuggingFace Repo",
         )
-        date_dd = gr.Dropdown(
-            choices=list_date_folders(DEFAULT_REPO),
-            label="Date"
+
+        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",
-        show_label=True
+        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 ----------
+    login_btn.click(
+        check_password,
+        inputs=pwd,
+        outputs=[
+            login_box,
+            app,
+            gallery,
+            page_state,
+            date_state,
+            page_info,
+        ],
     )
 
-    page_state = gr.State(0)
-    page_info = gr.Markdown()
-
-    with gr.Row():
-        prev_btn = gr.Button("⬅ Prev")
-        next_btn = gr.Button("Next ➡")
-        refresh_btn = gr.Button("🔄 Refresh")
+    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,
+    )
 
-    gr.HTML("<div id='copyright'>© OhamLab</div>")
+    date_dropdown.change(
+        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=date_state,
+    )
 
-    # Initial load
-    demo.load(
-        lambda r, d: render_page(r, d, 0),
-        inputs=[repo_input, date_dd],
-        outputs=[gallery, page_state, page_info]
+    manual_date.submit(
+        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=date_state,
     )
 
-    # Navigation
     prev_btn.click(
         lambda r, d, p: nav(-1, r, d, p),
-        inputs=[repo_input, date_dd, page_state],
-        outputs=[gallery, page_state, page_info]
+        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_input, date_dd, page_state],
-        outputs=[gallery, page_state, page_info]
-    )
-
-    refresh_btn.click(
-        refresh,
-        inputs=[repo_input, date_dd],
-        outputs=[gallery, page_state, page_info]
-    )
-
-    # Change date or repo → refresh cache
-    date_dd.change(
-        refresh,
-        inputs=[repo_input, date_dd],
-        outputs=[gallery, page_state, page_info]
+        inputs=[repo_state, date_state, page_state],
+        outputs=[gallery, page_state, page_info],
     )
 
-    repo_input.change(
-        lambda r: gr.Dropdown(choices=list_date_folders(r)),
-        inputs=repo_input,
-        outputs=date_dd
-    )
+    gr.HTML("<div id='ohamlab-footer'>© OhamLab Copyright</div>")
 
 demo.launch()