Update app.py

app.py

@@ -1,4 +1,3 @@
-import os
 import math
 import gradio as gr
 import pandas as pd
@@ -14,25 +13,21 @@ VIDEO_COL = "video"
 TEXT_COL = "text"
 DATE_COL = "date"
 
-HF_TOKEN = os.getenv("HF_TOKEN")
-if HF_TOKEN is None:
-    raise RuntimeError("HF_TOKEN environment variable not set")
-
 # ======================================================
 # LOAD DATA (PRIVATE DATASET)
+# Uses HUGGINGFACE_HUB_TOKEN automatically
 # ======================================================
 csv_path = hf_hub_download(
     repo_id=DATASET_REPO,
     filename=CSV_FILE,
-    repo_type="dataset",
-    token=HF_TOKEN
+    repo_type="dataset"
 )
 
 df = pd.read_csv(csv_path)
 df = df[[VIDEO_COL, TEXT_COL, DATE_COL]].dropna().reset_index(drop=True)
 
 # ======================================================
-# PAGINATION LOGIC
+# PAGINATION
 # ======================================================
 def render_grid(page, page_size):
     page = int(page)
@@ -40,7 +35,6 @@ def render_grid(page, page_size):
 
     total_items = len(df)
     total_pages = max(1, math.ceil(total_items / page_size))
-
     page = max(0, min(page, total_pages - 1))
 
     start = page * page_size
@@ -72,7 +66,6 @@ def render_grid(page, page_size):
         Page {page + 1} / {total_pages}
     </div>
     """
-
     return html, page
 
 
@@ -96,30 +89,21 @@ css = """
     grid-template-columns: repeat(auto-fill, minmax(300px, 1fr));
     gap: 18px;
 }
-
 .card {
     background: #0b1220;
     border-radius: 14px;
     padding: 12px;
     box-shadow: 0 10px 30px rgba(0,0,0,0.45);
 }
-
 .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 {
     font-size: 14px;
     color: #e5e7eb;
@@ -127,29 +111,18 @@ css = """
     max-height: 4.2em;
     overflow: hidden;
 }
-
 .page-info {
     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  
-        Browse and play videos from a **private Hugging Face dataset**.
-        """
-    )
+    gr.Markdown("# 🎬 Wan Video Dataset Showcase")
 
     with gr.Row():
-        page_size = gr.Dropdown(
-            choices=[1, 5, 10],
-            value=5,
-            label="Videos per page"
-        )
+        page_size = gr.Dropdown([1, 5, 10], value=5, label="Videos per page")
         page_state = gr.State(0)
 
     gallery = gr.HTML()
@@ -158,28 +131,9 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
         prev_btn = gr.Button("⬅ Previous")
         next_btn = gr.Button("Next ➡")
 
-    demo.load(
-        render_grid,
-        inputs=[page_state, page_size],
-        outputs=[gallery, page_state]
-    )
-
-    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]
-    )
-
-    page_size.change(
-        reset_page,
-        inputs=[page_size],
-        outputs=[gallery, page_state]
-    )
+    demo.load(render_grid, [page_state, page_size], [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])
 
 demo.launch()