Update app.py

app.py

@@ -3,6 +3,7 @@ import math
 import time
 import random
 import tempfile
+
 import gradio as gr
 import pandas as pd
 from datasets import load_dataset
@@ -12,7 +13,7 @@ from huggingface_hub import snapshot_download
 # QUOTA CONFIG
 # ======================================================
 LIMIT = 3
-WINDOW = 86400
+WINDOW = 86400  # 24 hours
 usage = {}
 
 
@@ -22,6 +23,7 @@ def quota_guard(fn):
         now = time.time()
 
         count, start = usage.get(session, (0, now))
+
         if now - start > WINDOW:
             count = 0
             start = now
@@ -46,46 +48,60 @@ APP_PASSWORD = os.getenv("APP_PASSWORD")
 if not APP_PASSWORD:
     raise RuntimeError("APP_PASSWORD environment variable not set")
 
+# PRO CONTENT
 PRO_DATASET = "rahul7star/Wan-video"
+VIDEO_COL = "video"
+TEXT_COL = "text"
+DATE_COL = "date"
+
+# FREE PREVIEW (MODEL REPO)
 FREE_VIDEO_REPO = "rahul7star/ohamlab"
 FREE_VIDEO_FOLDER = "video"
+MAX_FREE_VIDEOS = 5
 
 
 # ======================================================
-# LOAD FREE VIDEOS FROM HF REPO
+# LOAD FREE VIDEOS (MODEL REPO – SAFE)
 # ======================================================
 def load_free_videos():
     local_dir = snapshot_download(
         repo_id=FREE_VIDEO_REPO,
-        allow_patterns=[f"{FREE_VIDEO_FOLDER}/*"],
+        repo_type="model",                 # ✅ IMPORTANT
         local_dir=tempfile.mkdtemp(),
-        repo_type="model",
+        local_dir_use_symlinks=False,
     )
 
     video_dir = os.path.join(local_dir, FREE_VIDEO_FOLDER)
+
+    if not os.path.isdir(video_dir):
+        print(f"[WARN] Free video folder not found: {video_dir}")
+        return []
+
     videos = [
         os.path.join(video_dir, f)
         for f in os.listdir(video_dir)
         if f.lower().endswith((".mp4", ".webm", ".mov"))
     ]
 
+    print(f"[INFO] Loaded {len(videos)} free videos")
     return videos
 
 
 FREE_VIDEOS = load_free_videos()
 
 
-def pick_free_videos(max_count=5):
+def pick_free_videos(max_count=MAX_FREE_VIDEOS):
     if not FREE_VIDEOS:
         return []
 
     if len(FREE_VIDEOS) >= max_count:
         return random.sample(FREE_VIDEOS, max_count)
 
-    # repeat if less than max_count
+    # repeat videos if fewer than required
     result = []
     while len(result) < max_count:
         result.extend(FREE_VIDEOS)
+
     return result[:max_count]
 
 
@@ -93,34 +109,47 @@ def pick_free_videos(max_count=5):
 # LOAD PRO DATASET
 # ======================================================
 dataset = load_dataset(PRO_DATASET, split="test")
-df = dataset.to_pandas()[["video", "text", "date"]].dropna().reset_index(drop=True)
+df = dataset.to_pandas()
+df = df[[VIDEO_COL, TEXT_COL, DATE_COL]].dropna().reset_index(drop=True)
 
 
 # ======================================================
-# PAGINATION
+# PAGINATION LOGIC
 # ======================================================
 def render_grid(page, page_size):
     page = int(page)
     page_size = int(page_size)
 
-    total_pages = max(1, math.ceil(len(df) / page_size))
+    total_items = len(df)
+    total_pages = max(1, math.ceil(total_items / page_size))
     page = max(0, min(page, total_pages - 1))
 
-    batch = df.iloc[page * page_size:(page + 1) * page_size]
+    start = page * page_size
+    end = start + page_size
+    batch = df.iloc[start:end]
 
     cards = []
     for _, row in batch.iterrows():
         cards.append(f"""
         <div class="card">
-            <video src="{row['video']}" controls></video>
+            <video src="{row[VIDEO_COL]}" controls preload="metadata"></video>
             <div class="meta">
-                <div class="date">{row['date']}</div>
-                <div class="caption">{row['text']}</div>
+                <div class="date">{row[DATE_COL]}</div>
+                <div class="caption">{row[TEXT_COL]}</div>
             </div>
         </div>
         """)
 
-    return f"<div class='grid'>{''.join(cards)}</div>", page
+    html = f"""
+    <div class="grid">
+        {''.join(cards)}
+    </div>
+    <div class="page-info">
+        Page {page + 1} / {total_pages}
+    </div>
+    """
+
+    return html, page
 
 
 def next_page(page, page_size):
@@ -136,14 +165,14 @@ def reset_page(page_size):
 
 
 # ======================================================
-# AUTH
+# AUTH GATE
 # ======================================================
 @quota_guard
 def check_password(user_password, request: gr.Request):
     if user_password == APP_PASSWORD:
         html, page = render_grid(0, 5)
         return (
-            gr.update(visible=False),  # login
+            gr.update(visible=False),  # login box
             gr.update(visible=False),  # free preview
             gr.update(visible=True),   # pro content
             html,
@@ -203,23 +232,30 @@ footer, .gradio-footer {
 # UI
 # ======================================================
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
-    gr.Markdown("# 🔐 OhamLab Video Showcase - For PRO Users 18+")
+    gr.Markdown("# 🔐 OhamLab Video Showcase")
 
-    # ---------- LOGIN ----------
+    # -------- LOGIN --------
     with gr.Column(visible=True) as login_box:
         password_input = gr.Textbox(type="password", label="Password")
         login_btn = gr.Button("Unlock")
         login_status = gr.Markdown()
 
-    # ---------- FREE PREVIEW ----------
+    # -------- FREE PREVIEW --------
     with gr.Column(visible=True) as free_preview:
         gr.Markdown("### 🎬 Free Preview")
-        free_videos = pick_free_videos(5)
+
+        free_videos = pick_free_videos()
+
         with gr.Row():
             for v in free_videos:
-                gr.Video(v, autoplay=False)
-
-    # ---------- PRO CONTENT ----------
+                gr.Video(
+                    value=v,
+                    autoplay=False,
+                    loop=False,
+                    show_download_button=False
+                )
+
+    # -------- PRO CONTENT --------
     with gr.Column(visible=False) as app_content:
         with gr.Row():
             page_size = gr.Dropdown([1, 5, 10], value=5)
@@ -235,7 +271,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         next_btn.click(next_page, [page_state, page_size], [gallery, page_state])
         page_size.change(reset_page, [page_size], [gallery, page_state])
 
-    # ---------- EVENTS ----------
+    # -------- EVENTS --------
     login_btn.click(
         check_password,
         inputs=password_input,