Update app.py

app.py

@@ -86,7 +86,7 @@ except Exception as e:
     raise
 
 # Screenshot callback
-def save_screenshot(memory_step: ActionStep, agent: CodeAgent) -> Image.Image:
+def save_screenshot(memory_step: ActionStep, agent: CodeAgent) -> str:
     from time import sleep
     sleep(1.0)
     driver = helium.get_driver()
@@ -111,8 +111,7 @@ def save_screenshot(memory_step: ActionStep, agent: CodeAgent) -> Image.Image:
         memory_step.observations = (
             url_info if memory_step.observations is None else memory_step.observations + "\n" + url_info
         )
-        memory_step.observations_images = image
-        return image
+        return screenshot_path
 
 # Initialize model and agent
 model = LiteLLMModel("gemini/gemini-2.0-flash")