CLEAN image resources

cua2-core/src/cua2_core/services/agent_service.py

@@ -413,8 +413,9 @@ class AgentService:
 
             step_filename = f"{message_id}-{memory_step.step_number + 1}"
             screenshot_bytes = agent.desktop.screenshot()
-            image = Image.open(BytesIO(screenshot_bytes))
-            image = compress_image_to_max_size(image, max_size_kb=500)
+            original_image = Image.open(BytesIO(screenshot_bytes))
+            image = compress_image_to_max_size(original_image, max_size_kb=500)
+            del original_image
 
             for previous_memory_step in (
                 agent.memory.steps

cua2-core/src/cua2_core/services/utils.py

@@ -27,7 +27,9 @@ def compress_image_to_max_size(
     for compress_level in [3, 6, 9]:
         buffered = BytesIO()
         current_image.save(buffered, format="PNG", compress_level=compress_level)
-        png_size = buffered.tell()
+        png_bytes = buffered.getvalue()
+        buffered.close()
+        png_size = len(png_bytes)
         if png_size <= max_size_bytes:
             # Size is acceptable, return the image as-is (it's already in PNG format when saved)
             return current_image
@@ -52,7 +54,9 @@ def compress_image_to_max_size(
         # Check PNG size with best compression
         buffered = BytesIO()
         resized_image.save(buffered, format="PNG", compress_level=9)
-        png_size = buffered.tell()
+        png_bytes = buffered.getvalue()
+        buffered.close()
+        png_size = len(png_bytes)
 
         if png_size <= max_size_bytes:
             # Found acceptable size