Update app.py

app.py

@@ -6,39 +6,13 @@ import tempfile
 import gradio as gr
 from huggingface_hub import HfApi, hf_hub_download
 
-# ======================================================
-# QUOTA CONFIG
-# ======================================================
-LIMIT = int(os.getenv("APP_DAILY_LIMIT", "3"))
-WINDOW = 86400
-usage = {}
-
-def quota_guard(fn):
-    def wrapper(*args, **kwargs):
-        request: gr.Request | None = kwargs.get("request")
-        session = request.session_hash if request else "global"
-        now = time.time()
-
-        count, start = usage.get(session, (0, now))
-        if now - start > WINDOW:
-            count, start = 0, now
-
-        if count >= LIMIT:
-            raise gr.Error(
-                f"🚫 Daily demo limit reached.\n\n"
-                f"Max {LIMIT} unlocks per day."
-            )
-
-        usage[session] = (count + 1, start)
-        return fn(*args)
-    return wrapper
-
 # ======================================================
 # CONFIG
 # ======================================================
 REPO_ID = "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")
 
@@ -48,7 +22,9 @@ if not APP_PASSWORD:
     raise RuntimeError("APP_PASSWORD not set")
 
 api = HfApi()
-TMP_ROOT = tempfile.mkdtemp()
+
+PAGE_SIZE = 6
+tmp_dir = tempfile.mkdtemp()
 images_cache = {}
 
 # ======================================================
@@ -57,33 +33,45 @@ images_cache = {}
 def load_free_images():
     tmp = tempfile.mkdtemp()
     files = api.list_repo_files(FREE_REPO, token=HF_TOKEN)
+
     imgs = [
         f for f in files
-        if f.startswith(FREE_IMAGE_DIR + "/")
+        if f.startswith(FREE_IMAGE_DIR)
         and f.lower().endswith((".png", ".jpg", ".jpeg", ".webp"))
     ]
+
     random.shuffle(imgs)
     imgs = imgs[:5]
 
     out = []
     for f in imgs:
-        out.append(
-            hf_hub_download(FREE_REPO, f, local_dir=tmp, token=HF_TOKEN)
-        )
+        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
 
+
 FREE_IMAGES = load_free_images()
 
 # ======================================================
-# HELPERS
+# DATE + CACHE HELPERS
 # ======================================================
-def list_date_folders(last_n_days=20):
+def list_date_folders(last_n=20):
     files = api.list_repo_files(REPO_ID, token=HF_TOKEN)
     dates = sorted(
         {f.split("/")[0] for f in files if "/" in f},
         reverse=True
     )
-    return dates[:last_n_days]
+    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:
@@ -93,42 +81,77 @@ def download_images_for_date(date):
     local = []
 
     for f in files:
-        if f.startswith(date + "/") and f.lower().endswith((".png", ".jpg", ".jpeg", ".webp")):
-            local_path = hf_hub_download(
+        if not f.startswith(date + "/"):
+            continue
+        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)
+
+        try:
+            path = hf_hub_download(
                 REPO_ID,
                 f,
-                local_dir=TMP_ROOT,
-                token=HF_TOKEN
+                token=HF_TOKEN,
+                local_dir=dst_dir
             )
-            local.append(local_path)
+            local.append(path)
+            print("[DOWNLOAD]", f, "->", path)
+        except Exception as e:
+            print("[DOWNLOAD ERROR]", f, e)
 
-    images_cache[date] = sorted(local)
+    images_cache[date] = local
+    print(f"[INFO] {date} images:", len(local))
     return local
 
-def render_page(date, page, page_size):
+
+# ======================================================
+# PAGINATION
+# ======================================================
+def render_page(date, page):
     images = download_images_for_date(date)
-    total_pages = max(1, math.ceil(len(images) / page_size))
-    page = max(0, min(page, total_pages - 1))
-    start = page * page_size
-    end = start + page_size
-    return images[start:end], page, total_pages
+
+    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
+    subset = images[start:end]
+
+    print(f"[RENDER] {date} page {page+1}/{total_pages}")
+
+    return subset, page, f"Page {page+1} / {total_pages}"
+
+
+def nav(delta, date, page):
+    return render_page(date, page + delta)
+
+
+def change_date(date):
+    clear_date_cache(date)
+    return render_page(date, 0)
+
 
 # ======================================================
 # AUTH
 # ======================================================
-@quota_guard
 def check_password(pwd):
     if pwd != APP_PASSWORD:
         return (
             gr.update(visible=True),
             gr.update(visible=False),
             gr.update(visible=True),
-            [], 0, "", "❌ Wrong password"
+            [],
+            0,
+            "",
+            "❌ Wrong password",
         )
 
     dates = list_date_folders()
     latest = dates[0]
-    imgs, page, total = render_page(latest, 0, 6)
+
+    imgs, page, info = render_page(latest, 0)
 
     return (
         gr.update(visible=False),
@@ -137,14 +160,17 @@ def check_password(pwd):
         imgs,
         page,
         latest,
-        f"Page 1 / {total}"
+        info,
+        "✅ Access granted",
     )
 
+
 # ======================================================
-# UI
+# CSS
 # ======================================================
 css = """
 footer, .gradio-footer { display:none!important; }
+
 #ohamlab-footer {
   position:fixed;
   bottom:0;
@@ -157,10 +183,13 @@ footer, .gradio-footer { display:none!important; }
 }
 """
 
+# ======================================================
+# UI
+# ======================================================
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
-    gr.Markdown("# 🔐 OhamLab Image Showcase")
+    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")
@@ -171,57 +200,80 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
             for img in FREE_IMAGES:
                 gr.Image(img, show_label=False)
 
-    # APP
+    # -------- APP --------
     with gr.Column(visible=False) as app:
         current_date = gr.State("")
         page_state = gr.State(0)
-        page_size = gr.State(6)
 
         date_dropdown = gr.Dropdown(
             choices=list_date_folders(),
-            label="Select Date"
+            label="Select date"
+        )
+
+        manual_date = gr.Textbox(
+            label="Or enter date manually (YYYY-MM-DD)"
+        )
+
+        gallery = gr.Gallery(
+            label="Images",
+            columns=3,
+            height="auto"
         )
-        manual_date = gr.Textbox(label="Or enter date (YYYY-MM-DD)")
 
-        gallery = gr.Gallery(columns=3, height="auto")
         page_info = gr.Markdown()
 
         with gr.Row():
-            prev_btn = gr.Button("⬅ Prev")
+            prev_btn = gr.Button("⬅ Previous")
             next_btn = gr.Button("Next ➡")
 
-    # EVENTS
+    # -------- EVENTS --------
     login_btn.click(
         check_password,
-        pwd,
-        [login_box, app, gallery, page_state, current_date, page_info]
+        inputs=pwd,
+        outputs=[
+            login_box,
+            app,
+            login_box,
+            gallery,
+            page_state,
+            current_date,
+            page_info,
+            status,
+        ],
+    )
+
+    prev_btn.click(
+        lambda d, p: nav(-1, d, p),
+        inputs=[current_date, page_state],
+        outputs=[gallery, page_state, page_info],
     )
 
-    def change_date(date):
-        imgs, page, total = render_page(date, 0, 6)
-        return imgs, 0, date, f"Page 1 / {total}"
+    next_btn.click(
+        lambda d, p: nav(1, d, p),
+        inputs=[current_date, page_state],
+        outputs=[gallery, page_state, page_info],
+    )
 
     date_dropdown.change(
         change_date,
-        date_dropdown,
-        [gallery, page_state, current_date, page_info]
+        inputs=date_dropdown,
+        outputs=[gallery, page_state, page_info],
+    ).then(
+        lambda d: d,
+        inputs=date_dropdown,
+        outputs=current_date,
     )
 
     manual_date.submit(
         change_date,
-        manual_date,
-        [gallery, page_state, current_date, page_info]
+        inputs=manual_date,
+        outputs=[gallery, page_state, page_info],
+    ).then(
+        lambda d: d,
+        inputs=manual_date,
+        outputs=current_date,
     )
 
-    def nav(delta, date, page):
-        imgs, page, total = render_page(date, page + delta, 6)
-        return imgs, page, f"Page {page+1} / {total}"
-
-    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],)
-
-
-    gr.HTML("<div id='ohamlab-footer'>© OhamLab</div>")
+    gr.HTML("<div id='ohamlab-footer'>© OhamLab Copyright</div>")
 
 demo.launch()