Add application file

app.py

@@ -0,0 +1,85 @@
+import math
+import os
+
+import gradio as gr
+from datasets import load_dataset
+
+REPO_ID = "teddyk251/self-imagine-sample"  # change me
+TOKEN = os.environ.get("HF_TOKEN")
+
+ds = load_dataset(REPO_ID, split="train", token=TOKEN)
+
+DATASETS = ["All"] + sorted(set(ds["dataset"]))
+MODELS = ["All"] + sorted(set(ds["model"]))
+
+def filter_rows(dataset_sel, model_sel, query):
+    rows = ds
+    if dataset_sel != "All":
+        rows = rows.filter(lambda ex: ex["dataset"] == dataset_sel)
+    if model_sel != "All":
+        rows = rows.filter(lambda ex: ex["model"] == model_sel)
+    if query:
+        q = query.lower()
+        rows = rows.filter(lambda ex: q in (ex["filename"] or "").lower())
+    return rows
+
+def paginate(rows, page, page_size):
+    total = len(rows)
+    total_pages = max(1, math.ceil(total / max(1, page_size)))
+    page = max(1, min(page, total_pages))
+    start = (page - 1) * page_size
+    end = min(start + page_size, total)
+    sub = rows.select(range(start, end))
+    gallery_items = [(sub[i]["image"], f'{sub[i]["dataset"]} / {sub[i]["model"]} — {sub[i]["filename"]}')
+                     for i in range(len(sub))]
+    return gallery_items, total, total_pages, page
+
+def refresh(dataset_sel, model_sel, query, page, page_size):
+    rows = filter_rows(dataset_sel, model_sel, query)
+    items, total, total_pages, page = paginate(rows, page, page_size)
+    info = f"Page {page}/{total_pages} — {total} images"
+    return items, info, page
+
+with gr.Blocks(title="Self‑Imagine — Sample Gallery") as demo:
+    gr.Markdown("## 🖼️ Self‑Imagine — Sample Gallery")
+
+    with gr.Row():
+        dataset_dd = gr.Dropdown(DATASETS, value="All", label="Dataset")
+        model_dd = gr.Dropdown(MODELS, value="All", label="Model")
+        query_tb = gr.Textbox(label="Search filename")
+        page_size = gr.Slider(6, 60, value=24, step=6, label="Thumbnails per page")
+
+    with gr.Row():
+        prev_btn = gr.Button("◀ Prev")
+        next_btn = gr.Button("Next ▶")
+        page_num = gr.Number(value=1, precision=0, interactive=False, label="Page")
+        page_info = gr.Markdown()
+
+    gallery = gr.Gallery(columns=[6], height=700, allow_preview=False, show_label=False)
+
+    # initial
+    items, info, page = refresh("All", "All", "", 1, 24)
+    gallery.value, page_info.value, page_num.value = items, info, page
+
+    def do_search(dataset_sel, model_sel, query, page_size):
+        return (*refresh(dataset_sel, model_sel, query, 1, page_size), 1)
+
+    dataset_dd.change(do_search, [dataset_dd, model_dd, query_tb, page_size], [gallery, page_info, page_num])
+    model_dd.change(do_search, [dataset_dd, model_dd, query_tb, page_size], [gallery, page_info, page_num])
+    query_tb.change(do_search, [dataset_dd, model_dd, query_tb, page_size], [gallery, page_info, page_num])
+    page_size.change(do_search, [dataset_dd, model_dd, query_tb, page_size], [gallery, page_info, page_num])
+
+    def on_prev(dataset_sel, model_sel, query, page, page_size):
+        return (*refresh(dataset_sel, model_sel, query, max(1, page-1), page_size), max(1, page-1))
+
+    def on_next(dataset_sel, model_sel, query, page, page_size):
+        rows = filter_rows(dataset_sel, model_sel, query)
+        total_pages = max(1, math.ceil(len(rows) / max(1, page_size)))
+        newp = min(total_pages, page+1)
+        return (*refresh(dataset_sel, model_sel, query, newp, page_size), newp)
+
+    prev_btn.click(on_prev, [dataset_dd, model_dd, query_tb, page_num, page_size], [gallery, page_info, page_num])
+    next_btn.click(on_next, [dataset_dd, model_dd, query_tb, page_num, page_size], [gallery, page_info, page_num])
+
+if __name__ == "__main__":
+    demo.launch()