Update app.py

app.py

@@ -432,11 +432,8 @@ def build_ui():
                 with gr.Column(scale=2):
                     output_text = gr.Textbox(label="Indices (0-based)", lines=12, placeholder="[0, 1, 2, ...]")
 
-            def run_direct_indices(query: str, k: int) -> str:
-                idxs = search(query=query, k=k)
-                return str(idxs)
 
-            search_button.click(run_direct_indices, inputs=[query_box, k_slider], outputs=[output_text])
+            search_button.click(search, inputs=[query_box, k_slider], outputs=[output_text])
 
         with gr.Tab("3) Agent (Streaming)"):
             with gr.Row(equal_height=True):