update web_selenium.py to use try-with for headers (Significant-Gravi…

…tas#2988) Co-authored-by: Richard Beales <[email protected]>
jryals579 · May 1, 2023 · 7fc6f2a · 7fc6f2a
1 parent 94ec4a4
commit 7fc6f2a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/autogpt/commands/web_selenium.py b/autogpt/commands/web_selenium.py
@@ -175,4 +175,9 @@ def add_header(driver: WebDriver) -> None:
     Returns:
         None
     """
-    driver.execute_script(open(f"{FILE_DIR}/js/overlay.js", "r").read())
+    try:
+        with open(f"{FILE_DIR}/js/overlay.js", "r") as overlay_file:
+            overlay_script = overlay_file.read()
+        driver.execute_script(overlay_script)
+    except Exception as e:
+        print(f"Error executing overlay.js: {e}")