fix stuck on loading bug

app.py

@@ -53,17 +53,20 @@ def paginate(rows, page, page_size):
 
 def refresh(dataset_sel, model_sel, query, page, page_size):
     rows = filter_rows(dataset_sel, model_sel, query)
+    if len(rows) == 0:
+        return [], "Page 1/1 — 0 images", 1
     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")
+        query_tb = gr.Textbox(label="Search filename", placeholder="e.g., 17.jpg or gsm8k")
         page_size = gr.Slider(6, 60, value=24, step=6, label="Thumbnails per page")
 
     with gr.Row():
@@ -74,26 +77,31 @@ with gr.Blocks(title="Self‑Imagine — Sample Gallery") as demo:
 
     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 _initial():
+        items, info, page = refresh("All", "All", "", 1, 24)
+        return items, info, page
+    demo.load(_initial, inputs=None, outputs=[gallery, page_info, page_num])
 
     def do_search(dataset_sel, model_sel, query, page_size):
-        return (*refresh(dataset_sel, model_sel, query, 1, page_size), 1)
+        items, info, page = refresh(dataset_sel, model_sel, query, 1, page_size)
+        return items, info, 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])
+    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))
+        newp = max(1, int(page) - 1)
+        items, info, page = refresh(dataset_sel, model_sel, query, newp, page_size)
+        return items, info, newp
 
     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)
+        total_pages = max(1, math.ceil(len(rows) / max(1, int(page_size))))
+        newp = min(total_pages, int(page) + 1)
+        items, info, page = refresh(dataset_sel, model_sel, query, newp, page_size)
+        return items, info, 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])