Update app.py

app.py

@@ -1,172 +1,242 @@
 import os
 import math
 import time
+import random
 import tempfile
-from datetime import datetime, timedelta
-
 import gradio as gr
 from huggingface_hub import HfApi, hf_hub_download
-from PIL import Image
+
+# ======================================================
+# 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.\n"
+                f"Please try again after 24 hours."
+            )
+
+        usage[session] = (count + 1, start)
+        return fn(*args)
+    return wrapper
+
 
 # ======================================================
 # CONFIG
 # ======================================================
-REPO_ID = "rahul7star/WAN22-Feb-2026"
-IMAGE_EXTS = (".png", ".jpg", ".jpeg", ".webp")
+REPO_ID = "rahul7star/ohamlab"
+FREE_REPO = "rahul7star/ohamlab"
+FREE_IMAGE_DIR = "showcase/image"
 
-PAGE_SIZE = 6
-TMP_DIR = tempfile.mkdtemp()
+APP_PASSWORD = os.getenv("APP_PASSWORD")
+if not APP_PASSWORD:
+    raise RuntimeError("APP_PASSWORD env var not set")
 
 api = HfApi()
 
+
 # ======================================================
-# GLOBAL INDEX (FAST, BUILT ONCE)
+# FREE IMAGE PREVIEW
 # ======================================================
-FILE_INDEX: dict[str, list[str]] = {}
+def load_free_images():
+    tmp = tempfile.mkdtemp()
+    files = api.list_repo_files(FREE_REPO)
 
+    images = [
+        f for f in files
+        if f.startswith(FREE_IMAGE_DIR + "/")
+        and f.lower().endswith((".png", ".jpg", ".jpeg", ".webp"))
+    ]
 
-def build_index():
-    """
-    Build an in-memory index:
-    {
-      "2026-02-03": [
-          "2026-02-03/Upload-xxx/image.png",
-          ...
-      ]
-    }
-    """
-    if FILE_INDEX:
-        return
+    if not images:
+        return []
 
-    print("📦 Indexing HuggingFace repo once…")
-    files = api.list_repo_files(repo_id=REPO_ID)
-
-    for f in files:
-        if not f.lower().endswith(IMAGE_EXTS):
-            continue
+    random.shuffle(images)
+    picked = (images * 5)[:5] if len(images) < 5 else images[:5]
 
-        parts = f.split("/")
-        if len(parts) < 3:
-            continue  # must be date/Upload-xxx/image
+    local = []
+    for img in picked:
+        try:
+            local.append(hf_hub_download(FREE_REPO, img, local_dir=tmp))
+        except Exception:
+            pass
 
-        date_folder = parts[0]
-        FILE_INDEX.setdefault(date_folder, []).append(f)
+    return local
 
-    # stable ordering
-    for k in FILE_INDEX:
-        FILE_INDEX[k].sort()
 
-    print(f"✅ Indexed {sum(len(v) for v in FILE_INDEX.values())} images")
+FREE_IMAGES = load_free_images()
 
 
 # ======================================================
-# DATE HELPERS
+# PRO HELPERS
 # ======================================================
-def last_20_days_available():
-    today = datetime.utcnow().date()
-    dates = []
+def list_date_folders():
+    files = api.list_repo_files(REPO_ID)
+    return sorted({f.split("/")[0] for f in files if "/" in f})
 
-    for i in range(20):
-        d = (today - timedelta(days=i)).isoformat()
-        if d in FILE_INDEX:
-            dates.append(d)
 
-    return dates
+def load_images_for_date(date_folder):
+    files = api.list_repo_files(REPO_ID)
+    items = []
+
+    for f in files:
+        if f.startswith(date_folder + "/") and f.lower().endswith((".png", ".jpg", ".jpeg", ".webp")):
+            url = f"https://huggingface.co/{REPO_ID}/resolve/main/{f}"
+            items.append(url)
+
+    return items
 
 
 # ======================================================
-# IMAGE LOADING (PAGINATED)
+# PAGINATION
 # ======================================================
-def load_page(date: str, page: int):
-    images = FILE_INDEX.get(date, [])
-    total_pages = max(1, math.ceil(len(images) / PAGE_SIZE))
+def render_grid(date_folder, page, page_size):
+    data = load_images_for_date(date_folder)
+    total_pages = max(1, math.ceil(len(data) / page_size))
 
-    page = max(0, min(page, total_pages - 1))
-    start = page * PAGE_SIZE
-    end = start + PAGE_SIZE
+    page = max(0, min(int(page), total_pages - 1))
+    start, end = page * page_size, (page + 1) * page_size
 
-    gallery = []
-    for path in images[start:end]:
-        local = hf_hub_download(
-            repo_id=REPO_ID,
-            filename=path,
-            local_dir=TMP_DIR,
-            local_dir_use_symlinks=False,
-        )
-        gallery.append(Image.open(local))
+    cards = [
+        f"<div class='card'><img src='{img}'></div>"
+        for img in data[start:end]
+    ]
 
-    return (
-        gallery,
-        page,
-        page > 0,
-        end < len(images),
-        f"Page {page + 1} / {total_pages}",
-    )
+    return f"""
+    <div class="grid">{''.join(cards)}</div>
+    <div class="page-info">Page {page+1} / {total_pages}</div>
+    """, page
 
 
-def on_date_change(date):
-    return load_page(date, 0)
+def next_page(date, page, size):
+    return render_grid(date, page + 1, size)
 
+def prev_page(date, page, size):
+    return render_grid(date, page - 1, size)
 
-def next_page(date, page):
-    return load_page(date, page + 1)
+def reset_page(date, size):
+    return render_grid(date, 0, size)
 
 
-def prev_page(date, page):
-    return load_page(date, page - 1)
+# ======================================================
+# 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",
+             latest
+        )
+
+    dates = list_date_folders()
+    latest = dates[-1]
+    html, page = render_grid(latest, 0, 6)
+
+    return (
+        gr.update(visible=False),
+        gr.update(visible=True),
+        gr.update(visible=False),
+        html,
+        page,
+        "✅ Access granted",
+        latest
+    )
 
 
 # ======================================================
-# UI
+# CSS
 # ======================================================
-def create_app(dates):
-    with gr.Blocks() as demo:
-        gr.Markdown("## 📅 OhamLab Daily Image Browser")
+css = """
+.grid {
+  display:grid;
+  grid-template-columns:repeat(auto-fill,minmax(220px,1fr));
+  gap:16px;
+}
+.card {
+  background:#0b1220;
+  padding:10px;
+  border-radius:14px;
+}
+.card img {
+  width:100%;
+  border-radius:10px;
+}
+
+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;
+}
+"""
 
-        with gr.Row():
-            date_dd = gr.Dropdown(
-                choices=dates,
-                value=dates[0] if dates else None,
-                label="Select date (last 20 days)",
-            )
 
-        gallery = gr.Gallery(columns=3, height="auto")
+# ======================================================
+# UI
+# ======================================================
+with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
+    gr.Markdown("# 🔐 OhamLab Image Showcase- For PRO Users 18+ only unlock")
+
+    # ---------- LOGIN ----------
+    with gr.Column(visible=True) as login_box:
+        pwd = gr.Textbox(type="password", label="Password")
+        login_btn = gr.Button("Unlock")
+        status = gr.Markdown()
+
+        with gr.Column(visible=True) as free_preview:
+            gr.Markdown("### 🎁 Free Image Preview")
+            with gr.Row():
+                for img in FREE_IMAGES:
+                    gr.Image(img, show_label=False)
+
+    # ---------- PRO ----------
+    with gr.Column(visible=False) as app:
+        current_date = gr.State("")
         page_state = gr.State(0)
-        page_info = gr.Markdown()
+        page_size = gr.State(6)
+
+        gallery = gr.HTML()
 
         with gr.Row():
-            prev_btn = gr.Button("⬅ Previous", interactive=False)
-            next_btn = gr.Button("Next ➡", interactive=False)
-
-        # Events
-        date_dd.change(
-            on_date_change,
-            date_dd,
-            [gallery, page_state, prev_btn, next_btn, page_info],
-        )
+            prev_btn = gr.Button("⬅ Previous")
+            next_btn = gr.Button("Next ➡")
 
-        next_btn.click(
-            next_page,
-            [date_dd, page_state],
-            [gallery, page_state, prev_btn, next_btn, page_info],
-        )
+        prev_btn.click(prev_page, [current_date, page_state, page_size], [gallery, page_state])
+        next_btn.click(next_page, [current_date, page_state, page_size], [gallery, page_state])
 
-        prev_btn.click(
-            prev_page,
-            [date_dd, page_state],
-            [gallery, page_state, prev_btn, next_btn, page_info],
-        )
-
-        return demo
+    login_btn.click(
+        check_password,
+        inputs=pwd,
+        outputs=[login_box, app, free_preview, gallery, page_state, status, current_date]
+    )
 
+    gr.HTML("<div id='ohamlab-footer'>© OhamLab Copyright</div>")
 
-# ======================================================
-# ENTRY POINT (IMPORTANT FOR PYTHON 3.10)
-# ======================================================
-if __name__ == "__main__":
-    build_index()
-    available_dates = last_20_days_available()
 
-    app = create_app(available_dates)
-    app.launch()
-    
+demo.launch()