Update app.py

app.py

@@ -129,7 +129,7 @@ def index_from_url(url: str) -> Tuple[str, str]:
 # MCP Tools
 # =============================
 
-def mcp_test_search(query: str, k: int = 5) -> List[int]:
+def search(query: str, k: int = 5) -> List[int]:
     """
     Search within an indexed PDF and return ONLY the indices of the most relevant pages (0-based).
 
@@ -175,7 +175,7 @@ def mcp_test_search(query: str, k: int = 5) -> List[int]:
     return sorted(expanded)
 
 
-def mcp_get_pages(indices: List[int]) -> Dict[str, Any]:
+def get_pages(indices: List[int]) -> Dict[str, Any]:
     """
     Return page images (as data URLs) for the given 0-based indices.
 
@@ -241,7 +241,7 @@ Final deliverable:
 
 DEFAULT_MCP_SERVER_URL = "https://manu-mcp-test.hf.space/gradio_api/mcp/"
 DEFAULT_MCP_SERVER_LABEL = "colpali_rag"
-DEFAULT_ALLOWED_TOOLS = "mcp_test_search,mcp_get_pages"
+DEFAULT_ALLOWED_TOOLS = "mcp_test_search,mcp_test_get_pages"
 
 
 def stream_agent(question: str,
@@ -431,7 +431,7 @@ def build_ui():
                     output_text = gr.Textbox(label="Indices (0-based)", lines=12, placeholder="[0, 1, 2, ...]")
 
             def run_direct_indices(query: str, k: int) -> str:
-                idxs = mcp_test_search(query=query, k=k)
+                idxs = search(query=query, k=k)
                 return str(idxs)
 
             search_button.click(run_direct_indices, inputs=[query_box, k_slider], outputs=[output_text])