Update my_model/state_manager.py

my_model/state_manager.py

@@ -257,25 +257,28 @@ class StateManager:
         return st.session_state['images_data']
 
     @staticmethod
-    def resize_image(image_path, new_width, new_height):
+    def resize_image(image_input, new_width, new_height):
         """
-        Resizes an image from the specified to the given dimensions.
+        Resize an image given either a file path or a PIL Image object.
     
         Args:
-            image_path (str): Path to the image file.
-            new_width (int): Desired width for the resized image.
-            new_height (int): Desired height for the resized image.
+        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.
     
         Returns:
-            Image: The resized image object.
+        Image.Image: The resized PIL Image object.
         """
-
-        if isinstance(image_path, str):
+        if isinstance(image_input, str):
             # Open the image from a file path
-            image = Image.open(image_path)
-        elif hasattr(image_path, 'read'):
-            resized_image = image.resize((new_width, new_height))
-            
+            image = Image.open(image_input)
+        elif isinstance(image_input, 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")
+    
+        resized_image = image.resize((new_width, new_height))
         return resized_image