Update app.py

app.py

@@ -3,6 +3,8 @@ import requests
 import random
 import urllib.parse
 import logging
+from PIL import Image
+from io import BytesIO
 
 # Настройка логирования
 logging.basicConfig(level=logging.INFO)
@@ -62,7 +64,7 @@ def generate_image(prompt, style, width, height, seed):
         "Экспериментальный": "experimental style, abstract, unconventional, detailed patterns, vibrant colors"
     }
     
-    prompt = style_tags.get(style, "") + ". " + prompt if style_tags.get(style, "") else prompt
+    prompt = style_tags.get(style, "") + " " + prompt if style_tags.get(style, "") else prompt
     prompt = prompt.strip()
     
     # URL-кодирование промпта
@@ -75,7 +77,9 @@ def generate_image(prompt, style, width, height, seed):
         response = requests.get(url, timeout=150)
         if response.status_code == 200:
             logger.info("Image generated successfully")
-            return response.content
+            # Преобразуем байты в изображение
+            image = Image.open(BytesIO(response.content))
+            return image
         else:
             logger.error(f"Failed to generate image. Status code: {response.status_code}, Response: {response.text}")
             return None