TEMP

Author: adivekar-utexas

src/fmcore/algorithm/bedrock.py

@@ -54,7 +54,7 @@ class ConfigSelectionStrategy(AutoEnum):
     from botocore.exceptions import ClientError
     from PIL import Image
 
-    def _compress_image_for_bedrock(image_bytes: bytes, max_size_mb: float = 4.5) -> bytes:
+    def _compress_image_for_bedrock(image_bytes: bytes, max_size_mb: float = 3.5) -> bytes:
         """
         Compress an image to stay under the specified size limit for Bedrock.
         
@@ -101,6 +101,7 @@ def _compress_image_for_bedrock(image_bytes: bytes, max_size_mb: float = 4.5) ->
 
             # Check if we're under the size limit
             if len(compressed_bytes) <= max_size_bytes:
+                print(f'Final compressed image is {len(compressed_bytes) / (1024*1024):.2f}MB')
                 return compressed_bytes
 
             # If still too large, try reducing quality or scale
@@ -111,9 +112,8 @@ def _compress_image_for_bedrock(image_bytes: bytes, max_size_mb: float = 4.5) ->
                 quality = 95  # Reset quality when scaling
             else:
                 # If we can't compress enough, return what we have
-                Log.warning(f"Unable to compress image below {max_size_mb}MB limit. "
-                           f"Final size: {len(compressed_bytes) / (1024*1024):.2f}MB")
-                return compressed_bytes
+                raise ValueError(f"Unable to compress image below {max_size_mb}MB limit. "
+                                 f"Final size: {len(compressed_bytes) / (1024*1024):.2f}MB")
 
     def process_image_url(image_url: str) -> Optional[str]:
         """