Update app.py

app.py

@@ -10,6 +10,10 @@ import hashlib
 
 app = Flask(__name__)
 
+# Максимальные значения для ширины и высоты
+MAX_WIDTH = 850
+MAX_HEIGHT = 850
+
 # Кэш на 10 минут
 cache = TTLCache(maxsize=100, ttl=600)
 
@@ -25,6 +29,18 @@ def generate_cache_key(prompt, width, height, seed, model_name):
     # Создаем уникальный ключ на основе всех параметров
     return hashlib.md5(f"{prompt}_{width}_{height}_{seed}_{model_name}".encode()).hexdigest()
 
+def scale_dimensions(width, height, max_width, max_height):
+    """Масштабирует размеры изображения, сохраняя соотношение сторон."""
+    aspect_ratio = width / height
+    if width > max_width or height > max_height:
+        if width / max_width > height / max_height:
+            width = max_width
+            height = int(width / aspect_ratio)
+        else:
+            height = max_height
+            width = int(height * aspect_ratio)
+    return width, height
+
 @cached(cache)
 def generate_cached_image(cache_key, prompt, width, height, seed, model_name):
     api_key = get_random_key()
@@ -38,10 +54,7 @@ def generate_cached_image(cache_key, prompt, width, height, seed, model_name):
             "width": width,
             "height": height,
             "seed": seed
-        },
-        "width": width,
-        "height": height,
-        "seed": seed,
+        }
     }
 
     try:
@@ -64,6 +77,9 @@ def get_image(prompt):
     # Декодируем URL-кодированный prompt
     prompt = urllib.parse.unquote(prompt)
 
+    # Масштабируем размеры изображения, если они превышают максимальные значения
+    width, height = scale_dimensions(width, height, MAX_WIDTH, MAX_HEIGHT)
+
     # Генерируем уникальный ключ для кэша
     cache_key = generate_cache_key(prompt, width, height, seed, model_name)
 
@@ -87,4 +103,4 @@ def health_check():
     return "OK", 200
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860, debug=True)
+    app.run(host='0.0.0.0', port=7860, debug=False)