Update app.py

app.py

@@ -427,13 +427,17 @@ def build_ui():
             with gr.Row():
                 with gr.Column(scale=1):
                     query_box = gr.Textbox(placeholder="Enter your question…", label="Query", lines=4)
-                    k_slider = gr.Slider(minimum=1, maximum=50, step=1, label="Number of results (k)", value=5)
+                    k_slider = gr.Slider(minimum=1, maximum=10, step=1, label="Number of results (k)", value=5)
                     search_button = gr.Button("🔍 Search", variant="primary")
+                    get_pages_button = gr.Button("🔍 Get Pages", variant="primary")
+                    
                 with gr.Column(scale=2):
                     output_text = gr.Textbox(label="Indices (0-based)", lines=12, placeholder="[0, 1, 2, ...]")
+                    output_payload = gr.Textbox(label="Indices (0-based)", lines=12, placeholder="[0, 1, 2, ...]")
 
 
             search_button.click(search, inputs=[query_box, k_slider], outputs=[output_text])
+            get_pages_button.click(get_pages, inputs=[output_text], outputs=[output_payload])
 
         with gr.Tab("3) Agent (Streaming)"):
             with gr.Row(equal_height=True):