Resize file in screen toolkit

Author: zakiali

src/goose/toolkit/screen.py

@@ -1,5 +1,9 @@
 import subprocess
 import uuid
+from typing import Optional
+
+from rich.markdown import Markdown
+from rich.panel import Panel
 
 from goose.toolkit.base import Toolkit, tool
 
@@ -8,17 +12,30 @@ class Screen(Toolkit):
     """Provides an instructions on when and how to work with screenshots"""
 
     @tool
-    def take_screenshot(self) -> str:
+    def take_screenshot(self, resize: Optional[int]=None) -> str:
         """
         Take a screenshot to assist the user in debugging or designing an app. They may need help with moving a screen element, or interacting in some way where you could do with seeing the screen.
+        Optionally, resize the image using the `sips` cli tool. You may do this fs there is potential for error in the payload size.
 
+        Args:
+            resize (Optional[int]): Optional argument to resize the image.
         Return:
             (str) a path to the screenshot file, in the format of image: followed by the path to the file.
         """  # noqa: E501
         # Generate a random tmp filename for screenshot
         filename = f"/tmp/goose_screenshot_{uuid.uuid4().hex}.png"
+        screen_capture_command = ["screencapture", "-x", filename]
+        resize_command = []
+        resize_command_str = ''
+
+        subprocess.run(screen_capture_command)
+
+        if resize:
+            resize_command = ["sips", "-Z", resize, filename]
+            subprocess.run(resize_command)
+            resize_command_str = ' '.join(resize_command)
 
-        subprocess.run(["screencapture", "-x", filename])
+        self.notifier.log(Panel.fit(Markdown(f"```bash\n{' '.join(screen_capture_command)}\n{resize_command_str}"), title="screen"))
 
         return f"image:{filename}"