diff --git a/router.py b/router.py
index 111d638..08225fb 100644
--- a/router.py
+++ b/router.py
@@ -1418,30 +1418,30 @@ def resize_image_if_needed(image_data):
                 pass
         # Decode the base64 image data
         image_bytes = base64.b64decode(image_data)
-        image = Image.open(io.BytesIO(image_bytes))
-        if image.mode not in ("RGB", "L"):
-            image = image.convert("RGB")
+        with Image.open(io.BytesIO(image_bytes)) as image:
+            if image.mode not in ("RGB", "L"):
+                image = image.convert("RGB")
 
-        # Get current size
-        width, height = image.size
+            # Get current size
+            width, height = image.size
 
-        # Calculate the new dimensions while maintaining aspect ratio
-        if width > 512 or height > 512:
-            aspect_ratio = width / height
-            if aspect_ratio > 1:  # Width is larger
-                new_width = 512
-                new_height = int(512 / aspect_ratio)
-            else:  # Height is larger
-                new_height = 512
-                new_width = int(512 * aspect_ratio)
+            # Calculate the new dimensions while maintaining aspect ratio
+            if width > 512 or height > 512:
+                aspect_ratio = width / height
+                if aspect_ratio > 1:  # Width is larger
+                    new_width = 512
+                    new_height = int(512 / aspect_ratio)
+                else:  # Height is larger
+                    new_height = 512
+                    new_width = int(512 * aspect_ratio)
 
-            image = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
+                image = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
 
-        # Encode the resized image back to base64
-        buffered = io.BytesIO()
-        image.save(buffered, format="PNG")
-        resized_image_data = base64.b64encode(buffered.getvalue()).decode("utf-8")
-        return resized_image_data
+            # Encode the resized image back to base64
+            buffered = io.BytesIO()
+            image.save(buffered, format="PNG")
+            resized_image_data = base64.b64encode(buffered.getvalue()).decode("utf-8")
+            return resized_image_data
 
     except Exception as e:
         print(f"Error processing image: {e}")