Update app.py

app.py

@@ -31,30 +31,17 @@ def search_images_for_presentation(query: str) -> str:
     return image_path
 
 
-# Modified tool to capture screenshots and return image path
-def save_image_from_browser(memory_step: ActionStep, agent: CodeAgent) -> None:
-    sleep(1.0)  # Allow browser elements to load
-    driver = helium.get_driver()
-    current_step = memory_step.step_number
-    if driver is not None:
-        # Capture browser screenshot
-        png_bytes = driver.get_screenshot_as_png()
-        image = Image.open(BytesIO(png_bytes))
-        image_dir = "images"
-        os.makedirs(image_dir, exist_ok=True)
-        image_path = f"{image_dir}/captured_image_{current_step}.png"
-        
-        # Save image locally
-        image.save(image_path)
-        print(f"Captured and saved an image: {image_path}")
-        
-        memory_step.observations_images = [image.copy()]
-
-    # Update observations with the current URL for reference:
-    url_info = f"Current url: {driver.current_url}"
-    memory_step.observations = (
-        url_info if memory_step.observations is None else memory_step.observations + "\n" + url_info
-    )
+@tool
+def save_image_from_browser(image_url: str) -> str:
+    """
+    Saves the image from the given URL to a file.
+    Args:
+        image_url: The URL of the image to be saved.
+    Returns:
+        str: The path to the saved image.
+    """
+    image_path = f"/path/to/save/directory/{image_url.split('/')[-1]}"  # Example: use image's filename from URL
+    # Add image download and save logic here
     return image_path