Update app.py

app.py

@@ -243,6 +243,7 @@ def search_synthetize(query: str, k: int = 5) -> List[int]:
         results.append((images[idx], f"Page {page_num}"))
 
     # Generate grounded response
+    print("[waiting for ai response]", query)
     ai_response = query_gpt(query, results)
     print("[search_synthetize]", ai_response)
     return ai_response
@@ -308,7 +309,6 @@ Final deliverable (must be clear and standalone):
 # =============================
 # MCP config (search-only)
 # =============================
-VISUAL_REASONING = True
 DEFAULT_MCP_SERVER_URL = "https://manu-mcp-test.hf.space/gradio_api/mcp/"
 DEFAULT_MCP_SERVER_LABEL = "colpali_rag"
 
@@ -448,7 +448,7 @@ def stream_agent(question: str,
                             yield round_state["final_text"] or " ", round_state["summary_text"] or " ", "\n".join(log_lines[-400:])
 
                     # Capture tool RESULT text and try to parse indices
-                    elif etype.startswith("response.output_item.done"):
+                    elif etype.startswith("response.output_item.done") and visual_reasoning:
                         delta_text = getattr(event.item, "output", "")
                         if delta_text:
                             tool_result_buffer += str(delta_text)
@@ -487,7 +487,7 @@ def stream_agent(question: str,
 
         # If the model returned indices via the tool, use them in a fresh call
         next_indices = round_state.get("last_search_indices") or []
-        if next_indices:
+        if next_indices and visual_reasoning:
             # Neighbor expansion for context
             base = set(next_indices)
             expanded = set(base)