manu commited on
Commit
e8c4fd1
·
verified ·
1 Parent(s): cbc536a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -163,6 +163,7 @@ def search(query: str, k: int = 5) -> List[int]:
163
  # Score and select top-k
164
  scores = processor.score(q_vecs, ds, device=device)
165
  top_k_indices = scores[0].topk(k).indices.tolist()
 
166
 
167
  # Neighbor expansion for context
168
  base = set(top_k_indices)
@@ -180,7 +181,7 @@ def get_pages(indices: List[int]) -> Dict[str, Any]:
180
  Return page images (as data URLs) for the given 0-based indices.
181
 
182
  MCP tool description:
183
- - name: mcp_get_pages
184
  - description: Given 0-based indices from mcp_test_search, return the corresponding page images as data URLs for vision reasoning.
185
  - input_schema:
186
  type: object
@@ -196,6 +197,7 @@ def get_pages(indices: List[int]) -> Dict[str, Any]:
196
  {"images": [{"index": int, "page": int, "image_url": str}], "count": int}
197
  """
198
  global images
 
199
 
200
  if not images:
201
  return {"images": [], "count": 0}
 
163
  # Score and select top-k
164
  scores = processor.score(q_vecs, ds, device=device)
165
  top_k_indices = scores[0].topk(k).indices.tolist()
166
+ print(query, top_k_indices)
167
 
168
  # Neighbor expansion for context
169
  base = set(top_k_indices)
 
181
  Return page images (as data URLs) for the given 0-based indices.
182
 
183
  MCP tool description:
184
+ - name: mcp_test_get_pages
185
  - description: Given 0-based indices from mcp_test_search, return the corresponding page images as data URLs for vision reasoning.
186
  - input_schema:
187
  type: object
 
197
  {"images": [{"index": int, "page": int, "image_url": str}], "count": int}
198
  """
199
  global images
200
+ print("indices to get", indices)
201
 
202
  if not images:
203
  return {"images": [], "count": 0}