Update agent.py

agent.py

@@ -22,6 +22,9 @@ from tools import tools
 from langchain_core.callbacks import BaseCallbackHandler
 from langchain_core.tools import tool
 from duckduckgo_search import DDGS
+from selenium.webdriver.chrome.service import Service
+from webdriver_manager.chrome import ChromeDriverManager
+
 
 driver = None
 
@@ -31,7 +34,12 @@ def get_driver():
         options = webdriver.ChromeOptions()
         options.add_argument("--no-sandbox")
         options.add_argument("--disable-dev-shm-usage")
-        driver = webdriver.Chrome(options=options)
+        options.add_argument("--headless")  # 🔥 IMPORTANT
+
+        driver = webdriver.Chrome(
+            service=Service(ChromeDriverManager().install()),
+            options=options
+        )
     return driver
 
 @tool
@@ -69,10 +77,11 @@ def search_in_page(keyword: str, context_chars: int = 400) -> str:
 
     
 @tool
-def go_back() -> None:
+def go_back() -> str:
     """Goes back to previous page."""
     driver = get_driver()
     driver.back()
+    return "Went back to previous page"
 
 @tool
 def close_popups() -> str:
@@ -82,12 +91,16 @@ def close_popups() -> str:
     """
     driver = get_driver()
     webdriver.ActionChains(driver).send_keys(Keys.ESCAPE).perform()
+    return "Attempted to close popups"
 
 @tool
 def go_to(url: str) -> str:
     """Open a webpage using URL."""
     driver = get_driver()
     driver.get(url)
+    WebDriverWait(driver, 10).until(
+        lambda d: d.execute_script("return document.readyState") == "complete"
+    )
     return f"Opened {url}"
 
 @tool
@@ -192,13 +205,14 @@ toolsused = [
     get_page_text,
     go_back,
     close_popups,
-    get_youtube_transcript
+    get_youtube_transcript,
+    web_search   
 ]
-from tools import tools
+
 load_dotenv()
 
 # Load prompt
-with open("config.yaml") as f:
+with open("config.yaml", encoding="utf-8") as f:
     SYSTEM_PROMPT = yaml.safe_load(f)["prompt"]
 
 # LLM (local)