Update app.py

app.py

@@ -9,7 +9,6 @@ import urllib.parse
 import hashlib
 from deep_translator import GoogleTranslator
 from langdetect import detect
-import time
 
 app = Flask(__name__)
 
@@ -32,6 +31,7 @@ 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):
     """Масштабирует размеры изображения, сохраняя соотношение сторон, и округляет до чисел, кратных 8."""
     aspect_ratio = width / height
@@ -48,7 +48,8 @@ def scale_dimensions(width, height, max_width, max_height):
     height = (height + 3) // 8 * 8
     return width, height
 
-def generate_image(prompt, width, height, seed, model_name, api_key):
+@cached(cache, key=lambda prompt, width, height, seed, model_name: generate_cache_key(prompt, width, height, seed, model_name))
+def generate_cached_image(prompt, width, height, seed, model_name, api_key):
     headers = {
         "Authorization": f"Bearer {api_key}",
         "Content-Type": "application/json"
@@ -83,23 +84,12 @@ def generate_image(prompt, width, height, seed, model_name, api_key):
         app.logger.error(f"Request error occurred: {req_err}")
         return None
 
-def generate_with_retry(prompt, width, height, seed, model_name, api_key):
-    max_retries = 2  # Максимальное количество повторных попыток
-    for attempt in range(max_retries):
-        image = generate_image(prompt, width, height, seed, model_name, api_key)
-        if image:
-            return image
-        else:
-            app.logger.warning(f"Retrying image generation (attempt {attempt + 1}/{max_retries})...")
-            time.sleep(1)  # Небольшая задержка перед повторной попыткой
-    return None  # Возвращаем None, если все попытки не удались
-
 @app.route('/prompt/<path:prompt>')
 def get_image(prompt):
     width = request.args.get('width', type=int, default=512)
     height = request.args.get('height', type=int, default=512)
     seed = request.args.get('seed', type=int, default=25)
-    model_name = request.args.get('model', default="black-forest-labs+FLUX.1-dev").replace('+', '/')
+    model_name = request.args.get('model', default="black-forest-labs/FLUX.1-schnell").replace('+', '/')
     api_key = request.args.get('key', default=None)
 
     # Декодируем URL-кодированный prompt
@@ -128,19 +118,10 @@ def get_image(prompt):
     if api_key is None:
         api_key = get_random_key()
 
-    # Генерируем уникальный ключ для кэша
-    cache_key = generate_cache_key(prompt, width, height, seed, model_name)
-
     try:
-        # Пытаемся получить изображение из кэша
-        image = cache.get(cache_key)
+        image = generate_cached_image(prompt, width, height, seed, model_name, api_key)
         if image is None:
-            # Если нет в кэше, генерируем изображение
-            image = generate_with_retry(prompt, width, height, seed, model_name, api_key)
-            if image:
-               cache[cache_key] = image # Кэшируем только успешное изображение
-            else:
-               return send_error_image()
+            return send_error_image()
     except Exception as e:
         app.logger.error(f"Error generating image: {e}")
         return send_error_image()