Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import math
 import gradio as gr
 import pandas as pd
@@ -12,15 +13,14 @@ 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")
+
 # ======================================================
 # LOAD DATASET (PRIVATE-SAFE)
 # ======================================================
-# Auth is handled automatically via:
-# - `huggingface-cli login`
-# - or HUGGINGFACE_HUB_TOKEN env var
-dataset = load_dataset(DATASET_NAME, split="test")
-
-# Convert to pandas for easy slicing
+dataset = load_dataset(DATASET_NAME, split="train")
 df = dataset.to_pandas()
 df = df[[VIDEO_COL, TEXT_COL, DATE_COL]].dropna().reset_index(drop=True)
 
@@ -79,6 +79,28 @@ def prev_page(page, page_size):
 def reset_page(page_size):
     return render_grid(0, page_size)
 
+# ======================================================
+# AUTH GATE
+# ======================================================
+def check_password(user_password):
+    if user_password == APP_PASSWORD:
+        html, page = render_grid(0, 5)
+        return (
+            gr.update(visible=False),   # hide password box
+            gr.update(visible=True),    # show app content
+            html,
+            page,
+            "✅ Access granted"
+        )
+    else:
+        return (
+            gr.update(visible=True),
+            gr.update(visible=False),
+            "",
+            0,
+            "❌ Incorrect password"
+        )
+
 # ======================================================
 # UI
 # ======================================================
@@ -99,17 +121,13 @@ css = """
 .card video {
     width: 100%;
     border-radius: 12px;
-    outline: none;
 }
 
-.meta {
-    margin-top: 10px;
-}
+.meta { margin-top: 10px; }
 
 .date {
     font-size: 12px;
     color: #94a3b8;
-    margin-bottom: 4px;
 }
 
 .caption {
@@ -124,54 +142,42 @@ css = """
     margin-top: 14px;
     text-align: center;
     color: #c7d2fe;
-    font-weight: 500;
 }
 """
 
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
-    gr.Markdown(
-        """
-        # 🎬 Wan Video Dataset Showcase  
-        Powered by **Hugging Face Datasets** (private-safe).
-        """
-    )
-
-    with gr.Row():
-        page_size = gr.Dropdown(
-            choices=[1, 5, 10],
-            value=5,
-            label="Videos per page"
+    gr.Markdown("# 🔐 Wan Video Dataset (Password Protected)")
+
+    # -------- Password Gate --------
+    with gr.Column(visible=True) as login_box:
+        password_input = gr.Textbox(
+            label="Enter Password",
+            type="password",
+            placeholder="Password"
         )
-        page_state = gr.State(0)
+        login_btn = gr.Button("Unlock")
+        login_status = gr.Markdown()
 
-    gallery = gr.HTML()
+    # -------- Main App (Hidden) --------
+    with gr.Column(visible=False) as app_content:
+        with gr.Row():
+            page_size = gr.Dropdown([1, 5, 10], value=5, label="Videos per page")
+            page_state = gr.State(0)
 
-    with gr.Row():
-        prev_btn = gr.Button("⬅ Previous")
-        next_btn = gr.Button("Next ➡")
+        gallery = gr.HTML()
 
-    demo.load(
-        render_grid,
-        inputs=[page_state, page_size],
-        outputs=[gallery, page_state]
-    )
+        with gr.Row():
+            prev_btn = gr.Button("⬅ Previous")
+            next_btn = gr.Button("Next ➡")
 
-    next_btn.click(
-        next_page,
-        inputs=[page_state, page_size],
-        outputs=[gallery, page_state]
-    )
-
-    prev_btn.click(
-        prev_page,
-        inputs=[page_state, page_size],
-        outputs=[gallery, page_state]
-    )
+        next_btn.click(next_page, [page_state, page_size], [gallery, page_state])
+        prev_btn.click(prev_page, [page_state, page_size], [gallery, page_state])
+        page_size.change(reset_page, [page_size], [gallery, page_state])
 
-    page_size.change(
-        reset_page,
-        inputs=[page_size],
-        outputs=[gallery, page_state]
+    login_btn.click(
+        check_password,
+        inputs=password_input,
+        outputs=[login_box, app_content, gallery, page_state, login_status]
     )
 
 demo.launch()