Update app.py

app.py

@@ -7,38 +7,32 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
-from selenium import webdriver
-from selenium.webdriver.chrome.options import Options
+import playwright.sync_api as playwright
 
 @tool
 def get_device_info() -> str:
     """
-    Opens a headless browser and executes JavaScript to silently retrieve device information.
+    Opens a headless browser using Playwright and executes JavaScript to silently retrieve device information.
     
     Returns:
         str: A string containing device details such as user agent, platform, and screen resolution.
     """
-    options = Options()
-    options.add_argument("--headless")
-    options.add_argument("--disable-gpu")
-    options.add_argument("--no-sandbox")
-    options.add_argument("--disable-dev-shm-usage")
+    with playwright.sync_api.sync_playwright() as p:
+        browser = p.chromium.launch(headless=True)
+        page = browser.new_page()
+        
+        script = """
+        () => JSON.stringify({
+            userAgent: navigator.userAgent,
+            platform: navigator.platform,
+            language: navigator.language,
+            screenWidth: window.screen.width,
+            screenHeight: window.screen.height
+        })
+        """
+        
+        page.goto("about:blank")
+        device_info = page.evaluate(script)
+        browser.close()
     
-    driver = webdriver.Chrome(options=options)
-    
-    script = """
-    return JSON.stringify({
-        userAgent: navigator.userAgent,
-        platform: navigator.platform,
-        language: navigator.language,
-        screenWidth: window.screen.width,
-        screenHeight: window.screen.height
-    });
-    """
-    
-    driver.get("data:text/html,<script>document.title='device_info';</script>")
-    device_info = driver.execute_script(script)
-    driver.quit()
-    
-    return device_info
-
+    return device_info