Update my_model/state_manager.py

my_model/state_manager.py

@@ -256,33 +256,48 @@ class StateManager:
         """
         return st.session_state['images_data']
 
-    
-    def resize_image(self, image_input, new_width, new_height):
+
+
+    def resize_image_aspect_ratio(self,image_input, base_width=None, base_height=None):
         """
-        Resize an image given either a file path or a PIL Image object.
+        Resize an image while maintaining its aspect ratio.
     
         Args:
-        image_input (str or Image.Image): The image file path or a PIL Image object.
-        new_width (int): The new width of the image.
-        new_height (int): The new height of the image.
+        image (PIL.Image.Image): The image to resize.
+        base_width (int, optional): The target width, with the height adjusted to maintain aspect ratio.
+        base_height (int, optional): The target height, with the width adjusted to maintain aspect ratio.
     
         Returns:
-        Image.Image: The resized PIL Image object.
+        PIL.Image.Image: The resized image.
         """
-        
-        if isinstance(image_input, str):
-            self.col2.write("str")
+
+        image = copy.deepcopy(image_input)
+        if isinstance(image, str):
             # Open the image from a file path
-            image = Image.open(image_input)
-        elif isinstance(image_input, Image.Image):
-            self.col2.write("Image")
+            image = Image.open(image)
+        elif isinstance(image, Image.Image):
             # Use the image directly if it's already a PIL Image object
             image = image_input
         else:
             raise ValueError("image_input must be a file path or a PIL Image object")
+
+        # Get original dimensions
+        original_width, original_height = image.size
+    
+        # Calculate new dimensions
+        if base_width is not None:
+            ratio = (base_width / float(original_width))
+            new_height = int((float(original_height) * float(ratio)))
+            resized_image = image.resize((base_width, new_height))
+        elif base_height is not None:
+            ratio = (base_height / float(original_height))
+            new_width = int((float(original_width) * float(ratio)))
+            resized_image = image.resize((new_width, base_height))
+        else:
+            raise ValueError("Either base_width or base_height must be provided")
     
-        resized_image = image.resize((new_width, new_height))
         return resized_image
+
     
     
     def update_image_data(self, image_key, caption, detected_objects_str, analysis_done):