Update app.py

app.py

@@ -13,7 +13,7 @@ from huggingface_hub import snapshot_download
 # QUOTA CONFIG
 # ======================================================
 LIMIT = 3
-WINDOW = 86400  # 24 hours
+WINDOW = 86400
 usage = {}
 
 
@@ -23,7 +23,6 @@ def quota_guard(fn):
         now = time.time()
 
         count, start = usage.get(session, (0, now))
-
         if now - start > WINDOW:
             count = 0
             start = now
@@ -50,23 +49,21 @@ if not APP_PASSWORD:
 
 # 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"
+FREE_VIDEO_FOLDER = "showcase/video"   # ✅ FIXED PATH
 MAX_FREE_VIDEOS = 5
 
 
 # ======================================================
-# LOAD FREE VIDEOS (MODEL REPO – SAFE)
+# LOAD FREE VIDEOS (MODEL REPO – CORRECT PATH)
 # ======================================================
 def load_free_videos():
     local_dir = snapshot_download(
         repo_id=FREE_VIDEO_REPO,
-        repo_type="model",                 # ✅ IMPORTANT
+        repo_type="model",                        # ✅ correct
+        allow_patterns=["showcase/video/*"],     # ✅ optional but good
         local_dir=tempfile.mkdtemp(),
         local_dir_use_symlinks=False,
     )
@@ -83,7 +80,7 @@ def load_free_videos():
         if f.lower().endswith((".mp4", ".webm", ".mov"))
     ]
 
-    print(f"[INFO] Loaded {len(videos)} free videos")
+    print(f"[INFO] Loaded free videos: {videos}")
     return videos
 
 
@@ -97,7 +94,7 @@ def pick_free_videos(max_count=MAX_FREE_VIDEOS):
     if len(FREE_VIDEOS) >= max_count:
         return random.sample(FREE_VIDEOS, max_count)
 
-    # repeat videos if fewer than required
+    # repeat if fewer than max_count
     result = []
     while len(result) < max_count:
         result.extend(FREE_VIDEOS)
@@ -109,47 +106,34 @@ def pick_free_videos(max_count=MAX_FREE_VIDEOS):
 # LOAD PRO DATASET
 # ======================================================
 dataset = load_dataset(PRO_DATASET, split="test")
-df = dataset.to_pandas()
-df = df[[VIDEO_COL, TEXT_COL, DATE_COL]].dropna().reset_index(drop=True)
+df = dataset.to_pandas()[["video", "text", "date"]].dropna().reset_index(drop=True)
 
 
 # ======================================================
-# PAGINATION LOGIC
+# PAGINATION
 # ======================================================
 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))
 
-    start = page * page_size
-    end = start + page_size
-    batch = df.iloc[start:end]
+    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):
@@ -165,16 +149,16 @@ def reset_page(page_size):
 
 
 # ======================================================
-# AUTH GATE
+# AUTH
 # ======================================================
 @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 box
-            gr.update(visible=False),  # free preview
-            gr.update(visible=True),   # pro content
+            gr.update(visible=False),
+            gr.update(visible=False),
+            gr.update(visible=True),
             html,
             page,
             "✅ Access granted"
@@ -190,57 +174,19 @@ def check_password(user_password, request: gr.Request):
     )
 
 
-# ======================================================
-# CSS
-# ======================================================
-css = """
-.grid {
-    display: grid;
-    grid-template-columns: repeat(auto-fill, minmax(300px, 1fr));
-    gap: 18px;
-}
-
-.card {
-    background: #0b1220;
-    border-radius: 14px;
-    padding: 12px;
-}
-
-.card video {
-    width: 100%;
-    border-radius: 12px;
-}
-
-footer, .gradio-footer {
-    display: none !important;
-}
-
-#ohamlab-footer {
-    position: fixed;
-    bottom: 0;
-    width: 100%;
-    text-align: center;
-    padding: 8px;
-    font-size: 12px;
-    background: #f8f9fb;
-    border-top: 1px solid #e5e7eb;
-}
-"""
-
-
 # ======================================================
 # UI
 # ======================================================
-with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
     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")
 
@@ -248,30 +194,14 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
 
         with gr.Row():
             for v in free_videos:
-                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)
-            page_state = gr.State(0)
+                gr.Video(v)
 
+    # PRO CONTENT
+    with gr.Column(visible=False) as app_content:
+        page_size = gr.Dropdown([1, 5, 10], value=5)
+        page_state = gr.State(0)
         gallery = gr.HTML()
 
-        with gr.Row():
-            prev_btn = gr.Button("⬅ Previous")
-            next_btn = gr.Button("Next ➡")
-
-        prev_btn.click(prev_page, [page_state, page_size], [gallery, page_state])
-        next_btn.click(next_page, [page_state, page_size], [gallery, page_state])
-        page_size.change(reset_page, [page_size], [gallery, page_state])
-
-    # -------- EVENTS --------
     login_btn.click(
         check_password,
         inputs=password_input,
@@ -285,7 +215,4 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         ],
     )
 
-    gr.HTML("<div id='ohamlab-footer'>© OhamLab</div>")
-
-
 demo.launch()