Update app.py

app.py

@@ -36,6 +36,44 @@ def add_noise(image, intensity=25):
     noisy_image = Image.fromarray(noisy_array)
     return noisy_image
 
+def resize_and_crop(image, target_width=768, target_height=1024):
+    """
+    Подгоняет изображение под размер target_width x target_height с сохранением пропорций.
+    Если изображение не соответствует соотношению сторон, обрезает его по бокам.
+    :param image: Изображение (PIL.Image)
+    :param target_width: Целевая ширина
+    :param target_height: Целевая высота
+    :return: Изображение с измененным размером (PIL.Image)
+    """
+    # Сохраняем исходные пропорции
+    original_width, original_height = image.size
+    target_ratio = target_width / target_height
+    original_ratio = original_width / original_height
+
+    # Масштабируем изображение с сохранением пропорций
+    if original_ratio > target_ratio:
+        # Если изображение шире, чем нужно, обрезаем по бокам
+        new_height = target_height
+        new_width = int(original_width * (target_height / original_height))
+        resized_image = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
+        left = (new_width - target_width) / 2
+        top = 0
+        right = (new_width + target_width) / 2
+        bottom = target_height
+    else:
+        # Если изображение уже, чем нужно, обрезаем сверху и снизу
+        new_width = target_width
+        new_height = int(original_height * (target_width / original_width))
+        resized_image = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
+        left = 0
+        top = (new_height - target_height) / 2
+        right = target_width
+        bottom = (new_height + target_height) / 2
+
+    # Обрезаем изображение до целевого размера
+    cropped_image = resized_image.crop((left, top, right, bottom))
+    return cropped_image
+
 def query(prompt):
     if prompt == "" or prompt == None:
         return None
@@ -61,14 +99,7 @@ def query(prompt):
         image_bytes = response.content
         image = Image.open(io.BytesIO(image_bytes))
 
-        width, height = image.size
-        new_width = 480
-        new_height = 640
-        left = (width - new_width) / 2
-        top = (height - new_height) / 2
-        right = (width + new_width) / 2
-        bottom = (height + new_height) / 2
-        image = image.crop((left, top, right, bottom))
+        image = resize_and_crop(image, 480, 640)
         
         # Изменение насыщенности
         enhancer = ImageEnhance.Color(image)