Update app.py

app.py

@@ -40,6 +40,24 @@ def query(prompt, is_negative=False):
     try:
         image_bytes = response.content
         image = Image.open(io.BytesIO(image_bytes))
+
+        width, height = image.size
+        new_width = 640
+        new_height = 1024
+        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))
+        
+        # Изменение экспозиции (яркости)
+        enhancer = ImageEnhance.Brightness(image)
+        image = enhancer.enhance(1.2)  # Увеличиваем яркость на 20%
+        
+        # Изменение насыщенности
+        enhancer = ImageEnhance.Color(image)
+        image = enhancer.enhance(0.8)  # Увеличиваем насыщенность на 50%
+        
         print(f'\033[1mGeneration {key} completed!\033[0m ({prompt})')
         return image
     except Exception as e: