Update app.py

app.py

@@ -2,16 +2,18 @@ import os
 import math
 import time
 import random
+import tempfile
 import gradio as gr
 import pandas as pd
 from datasets import load_dataset
+from huggingface_hub import snapshot_download
 
 # ======================================================
 # QUOTA CONFIG
 # ======================================================
 LIMIT = 3
-WINDOW = 86400  # 24 hours
-usage = {}      # session_hash -> (count, start_time)
+WINDOW = 86400
+usage = {}
 
 
 def quota_guard(fn):
@@ -20,7 +22,6 @@ def quota_guard(fn):
         now = time.time()
 
         count, start = usage.get(session, (0, now))
-
         if now - start > WINDOW:
             count = 0
             start = now
@@ -41,59 +42,58 @@ def quota_guard(fn):
 # ======================================================
 # CONFIG
 # ======================================================
-DATASET_NAME = "rahul7star/Wan-video"
-
-VIDEO_COL = "video"
-TEXT_COL = "text"
-DATE_COL = "date"
-
 APP_PASSWORD = os.getenv("APP_PASSWORD")
 if not APP_PASSWORD:
     raise RuntimeError("APP_PASSWORD environment variable not set")
 
+PRO_DATASET = "rahul7star/Wan-video"
+FREE_VIDEO_REPO = "rahul7star/ohamlab"
+FREE_VIDEO_FOLDER = "video"
+
 
 # ======================================================
-# YOUTUBE FREE PREVIEW
+# LOAD FREE VIDEOS FROM HF REPO
 # ======================================================
-YOUTUBE_LINKS = [
-    "https://www.youtube.com/watch?v=g30KR9VNXS0",
-    "https://www.youtube.com/watch?v=g30KR9VNXS0",
-    "https://www.youtube.com/watch?v=g30KR9VNXS0",
-    "https://www.youtube.com/watch?v=g30KR9VNXS0",
-    "https://www.youtube.com/watch?v=g30KR9VNXS0",
-]
+def load_free_videos():
+    local_dir = snapshot_download(
+        repo_id=FREE_VIDEO_REPO,
+        allow_patterns=[f"{FREE_VIDEO_FOLDER}/*"],
+        local_dir=tempfile.mkdtemp(),
+        repo_type="dataset",
+    )
 
+    video_dir = os.path.join(local_dir, FREE_VIDEO_FOLDER)
+    videos = [
+        os.path.join(video_dir, f)
+        for f in os.listdir(video_dir)
+        if f.lower().endswith((".mp4", ".webm", ".mov"))
+    ]
 
-def render_free_videos():
-    picks = random.sample(YOUTUBE_LINKS, 4)
-    cards = []
+    return videos
 
-    for link in picks:
-        vid = link.split("v=")[-1]
-        cards.append(f"""
-        <div class="free-card">
-            <iframe
-                src="https://www.youtube.com/embed/{vid}"
-                allowfullscreen
-                loading="lazy">
-            </iframe>
-        </div>
-        """)
 
-    return f"""
-    <h3>🎬 Free Preview</h3>
-    <div class="free-grid">
-        {''.join(cards)}
-    </div>
-    """
+FREE_VIDEOS = load_free_videos()
+
+
+def pick_free_videos(max_count=5):
+    if not FREE_VIDEOS:
+        return []
+
+    if len(FREE_VIDEOS) >= max_count:
+        return random.sample(FREE_VIDEOS, max_count)
+
+    # repeat if less than max_count
+    result = []
+    while len(result) < max_count:
+        result.extend(FREE_VIDEOS)
+    return result[:max_count]
 
 
 # ======================================================
-# LOAD DATASET
+# LOAD PRO DATASET
 # ======================================================
-dataset = load_dataset(DATASET_NAME, split="test")
-df = dataset.to_pandas()
-df = df[[VIDEO_COL, TEXT_COL, DATE_COL]].dropna().reset_index(drop=True)
+dataset = load_dataset(PRO_DATASET, split="test")
+df = dataset.to_pandas()[["video", "text", "date"]].dropna().reset_index(drop=True)
 
 
 # ======================================================
@@ -103,30 +103,24 @@ def render_grid(page, page_size):
     page = int(page)
     page_size = int(page_size)
 
-    total_items = len(df)
-    total_pages = max(1, math.ceil(total_items / page_size))
+    total_pages = max(1, math.ceil(len(df) / page_size))
     page = max(0, min(page, total_pages - 1))
 
-    batch = df.iloc[page * page_size: (page + 1) * page_size]
+    batch = df.iloc[page * page_size:(page + 1) * page_size]
 
     cards = []
     for _, row in batch.iterrows():
         cards.append(f"""
         <div class="card">
-            <video src="{row[VIDEO_COL]}" controls preload="metadata"></video>
+            <video src="{row['video']}" controls></video>
             <div class="meta">
-                <div class="date">{row[DATE_COL]}</div>
-                <div class="caption">{row[TEXT_COL]}</div>
+                <div class="date">{row['date']}</div>
+                <div class="caption">{row['text']}</div>
             </div>
         </div>
         """)
 
-    html = f"""
-    <div class="grid">{''.join(cards)}</div>
-    <div class="page-info">Page {page + 1} / {total_pages}</div>
-    """
-
-    return html, page
+    return f"<div class='grid'>{''.join(cards)}</div>", page
 
 
 def next_page(page, page_size):
@@ -149,22 +143,22 @@ 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 box
+            gr.update(visible=False),  # login
             gr.update(visible=False),  # free preview
-            gr.update(visible=True),   # app content
+            gr.update(visible=True),   # pro content
             html,
             page,
             "✅ Access granted"
         )
-    else:
-        return (
-            gr.update(visible=True),
-            gr.update(visible=True),
-            gr.update(visible=False),
-            "",
-            0,
-            "❌ Incorrect password"
-        )
+
+    return (
+        gr.update(visible=True),
+        gr.update(visible=True),
+        gr.update(visible=False),
+        "",
+        0,
+        "❌ Incorrect password"
+    )
 
 
 # ======================================================
@@ -188,20 +182,6 @@ css = """
     border-radius: 12px;
 }
 
-.free-grid {
-    display: grid;
-    grid-template-columns: repeat(auto-fill, minmax(260px, 1fr));
-    gap: 16px;
-    margin-top: 12px;
-}
-
-.free-card iframe {
-    width: 100%;
-    aspect-ratio: 16 / 9;
-    border-radius: 12px;
-}
-
-/* Hide default footer */
 footer, .gradio-footer {
     display: none !important;
 }
@@ -223,18 +203,23 @@ footer, .gradio-footer {
 # UI
 # ======================================================
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
-    gr.Markdown("# 🔐 OhamLab Video Showcase")
+    gr.Markdown("# 🔐 OhamLab Video Showcase - For PRO Users 18+")
 
-    # -------- LOGIN + FREE PREVIEW --------
+    # ---------- 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 ----------
     with gr.Column(visible=True) as free_preview:
-        free_html = gr.HTML(render_free_videos())
+        gr.Markdown("### 🎬 Free Preview")
+        free_videos = pick_free_videos(5)
+        with gr.Row():
+            for v in free_videos:
+                gr.Video(v, autoplay=False)
 
-    # -------- PRO CONTENT --------
+    # ---------- PRO CONTENT ----------
     with gr.Column(visible=False) as app_content:
         with gr.Row():
             page_size = gr.Dropdown([1, 5, 10], value=5)
@@ -250,7 +235,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,