Update app.py

app.py

@@ -38,7 +38,7 @@ load_dotenv(override=True)
 # CONFIGURATION & CONSTANTS
 # -----------------------------------------------------------------------------
 
-E2B_API_KEY = os.getenv("E2B_API")
+E2B_API_KEY = os.getenv("E2B_API_KEY")
 HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_API_KEY")
 if HF_TOKEN:
     login(token=HF_TOKEN)
@@ -432,6 +432,11 @@ class E2BVisionAgent(CodeAgent):
         def drag_and_drop(x1: int, y1: int, x2: int, y2: int) -> str:
             """
             Clicks [x1, y1], drags mouse to [x2, y2], then release click.
+            Args:
+                x1: The x coordinate of the start position.
+                y1: The y coordinate of the start position.
+                x2: The x coordinate of the end position.
+                y2: The y coordinate of the end position.
             """
             self.desktop.drag([x1, y1], [x2, y2])
             message = f"Dragged and dropped from [{x1}, {y1}] to [{x2}, {y2}]"
@@ -458,6 +463,8 @@ class E2BVisionAgent(CodeAgent):
         def wait(seconds: float) -> str:
             """
             Waits for the specified number of seconds.
+            Args:
+                seconds: The duration to wait in seconds.
             """
             time.sleep(seconds)
             self.logger.log(f"Waited for {seconds} seconds")
@@ -467,6 +474,8 @@ class E2BVisionAgent(CodeAgent):
         def open_url(url: str) -> str:
             """
             Directly opens a browser with the specified url.
+            Args:
+                url: The website URL to open.
             """
             if not url.startswith(("http://", "https://")):
                 url = "https://" + url
@@ -480,6 +489,8 @@ class E2BVisionAgent(CodeAgent):
         def find_on_page_ctrl_f(search_string: str) -> str:
             """
             Scroll the browser viewport to the first occurrence of the search string (Ctrl+F).
+            Args:
+                search_string: The text to search for on the page.
             """
             self.desktop.press(["ctrl", "f"])
             time.sleep(0.3)
@@ -780,7 +791,6 @@ def initialize_session(interactive_mode, browser_uuid):
         return update_html(interactive_mode, browser_uuid), browser_uuid
 
 class EnrichedGradioUI(GradioUI):
-    # REMOVED @spaces.GPU from here to prevent pickling the E2B Sandbox (which has locks)
     def interact_with_agent(
         self,
         task_input,