Update agent.py

agent.py

@@ -32,17 +32,6 @@ def build_gemini_llm():
         raise ValueError("GOOGLE_API_KEY environment variable is not set.")
     return ChatGoogleGenerativeAI(model = "gemini-3.7-flash", temperature = 0, max_output_tokens = 1025, include_thoughts=True)
 
-
-@tool
-def search_web(query:str) -> str:
-    """
-    Searches the web for the answer to a given question or topic.
-    Args:
-        query (str): the question or topic to search for.
-    """
-    print(f"Query to ddsg: {query}")
-    return DuckDuckGoSearchRun()
-
 @tool
 def extract_text_from_image(img_path: str) -> str:
     """
@@ -184,6 +173,7 @@ def wikipedia_search(query: str):
         print(f"Wikipedia search failed for {query}: {e}")
         return f"Wikipedia search failed for {query} due to {e}."
 
+web_search = DuckDuckGoSearchRun()
 model = build_gemini_llm()
 tools = [
     search_web,
@@ -195,7 +185,7 @@ def assistant(state: AgentState):
     response = model_with_tools.invoke(state["messages"])
     print("--- Thinking Section ---")
     print(response.content[0].get("thinking").strip())
-    print("--- End of Thinking ---")
+    print("--- End of Thinking ---\n\n")
     return {
         "messages": [response],
         "file_path": state["file_path"],