Update app.py

app.py

@@ -214,7 +214,7 @@ def search_synthetize(query: str, k: int = 5) -> List[int]:
     """
     Search within a PDF document for the most relevant pages to answer a query and synthetizes a short grounded answer using only those pages.
     MCP tool description:
-      - name: mcp_test_search
+      - name: mcp_test_search_synthetize
       - description: Search within a PDF document for the most relevant pages to answer a query and synthetizes a short grounded answer using only those pages.
       - input_schema:
           type: object
@@ -626,11 +626,14 @@ def build_ui():
                     query_box = gr.Textbox(placeholder="Enter your question…", label="Query", lines=4)
                     k_slider = gr.Slider(minimum=1, maximum=10, step=1, label="Number of results (k)", value=5)
                     search_button = gr.Button("🔍 Search", variant="primary")
+                    search_synthetize_button = gr.Button("🔍 Search & Synthetize", variant="primary")
 
                 with gr.Column(scale=2):
-                    output_text = gr.Textbox(label="Indices (0-based)", lines=12, placeholder="[0, 1, 2, ...]")
+                    output_docs = gr.Textbox(label="Indices (0-based)", lines=1, placeholder="[0, 1, 2, ...]")
+                    output_text = gr.Textbox(label="ColQwen+GPT-5 Answer", lines=12, placeholder="...")
 
-            search_button.click(search, inputs=[query_box, k_slider], outputs=[output_text])
+            search_button.click(search, inputs=[query_box, k_slider], outputs=[output_docs])
+            search_synthetize_button.click(search_synthetize, inputs=[query_box, k_slider], outputs=[output_text])
 
         # ---- Tab 3: Agent (Streaming)
         with gr.Tab("3) Deep Search"):