Update app.py

app.py

@@ -4,7 +4,6 @@ from pyserini.search.lucene import LuceneSearcher
 
 searcher = LuceneSearcher("index")
 ds = load_from_disk("data")
-PAGINATION_VISIBLE = True
 NUM_PAGES = 10 # STATIC. THIS CAN'T CHANGE BECAUSE GRADIO CAN'T DYNAMICALLY CREATE COMPONENTS. 
 RESULTS_PER_PAGE = 5 
 
@@ -25,8 +24,7 @@ def page_0(query):
     ix = [int(hit.docid) for hit in hits]
     results = ds.select(ix).shard(num_shards=NUM_PAGES, index=0, contiguous=True) # no need to shard. split ix in batches instead. (would make sense if results was cacheable)
     results = format_results(results)
-    return results, [ix]
-
+    return results, [ix], gr.update(visible=True)
 def page_i(i, ix):
     ix = ix[0]
     results = ds.select(ix).shard(num_shards=NUM_PAGES, index=i, contiguous=True)
@@ -54,23 +52,23 @@ with gr.Blocks(css="#b {min-width:15px;background:transparent;border:white;box-s
             pass
         with gr.Column(scale=13):
             c = gr.HTML(label="Results")
-            with gr.Row():
+            with gr.Row(visible=False) as pagination:
                 # left = gr.Button(value="◀", elem_id="b", visible=False).style(full_width=True)
-                page_1 = gr.Button(value="1", elem_id="b", visible=PAGINATION_VISIBLE).style(full_width=True)
-                page_2 = gr.Button(value="2", elem_id="b", visible=PAGINATION_VISIBLE).style(full_width=True)
-                page_3 = gr.Button(value="3", elem_id="b", visible=PAGINATION_VISIBLE).style(full_width=True)
-                page_4 = gr.Button(value="4", elem_id="b", visible=PAGINATION_VISIBLE).style(full_width=True)
-                page_5 = gr.Button(value="5", elem_id="b", visible=PAGINATION_VISIBLE).style(full_width=True)
-                page_6 = gr.Button(value="6", elem_id="b", visible=PAGINATION_VISIBLE).style(full_width=True)
-                page_7 = gr.Button(value="7", elem_id="b", visible=PAGINATION_VISIBLE).style(full_width=True)
-                page_8 = gr.Button(value="8", elem_id="b", visible=PAGINATION_VISIBLE).style(full_width=True)
-                page_9 = gr.Button(value="9", elem_id="b", visible=PAGINATION_VISIBLE).style(full_width=True)
-                page_10 = gr.Button(value="10", elem_id="b", visible=PAGINATION_VISIBLE).style(full_width=True)
+                page_1 = gr.Button(value="1", elem_id="b").style(full_width=True)
+                page_2 = gr.Button(value="2", elem_id="b").style(full_width=True)
+                page_3 = gr.Button(value="3", elem_id="b").style(full_width=True)
+                page_4 = gr.Button(value="4", elem_id="b").style(full_width=True)
+                page_5 = gr.Button(value="5", elem_id="b").style(full_width=True)
+                page_6 = gr.Button(value="6", elem_id="b").style(full_width=True)
+                page_7 = gr.Button(value="7", elem_id="b").style(full_width=True)
+                page_8 = gr.Button(value="8", elem_id="b").style(full_width=True)
+                page_9 = gr.Button(value="9", elem_id="b").style(full_width=True)
+                page_10 = gr.Button(value="10", elem_id="b").style(full_width=True)
                 # right = gr.Button(value="▶", elem_id="b", visible=False).style(full_width=True)
         with gr.Column(scale=1):
             pass
-    query.submit(fn=page_0, inputs=[query], outputs=[c, result_list])
-    submit_btn.click(page_0, inputs=[query], outputs=[c, result_list])
+    query.submit(fn=page_0, inputs=[query], outputs=[c, result_list, pagination])
+    submit_btn.click(page_0, inputs=[query], outputs=[c, result_list, pagination])
     with gr.Box(visible=False):
         nums = [gr.Number(i, visible=False, precision=0) for i in range(NUM_PAGES)]
     page_1.click(fn=page_i, inputs=[nums[0], result_list], outputs=[c, result_list])