Update app.py

app.py

@@ -28,9 +28,10 @@ def get_random_key():
     return random.choice(keys)
 
 def generate_cache_key(prompt, width, height, seed, model_name):
-    # Создаем уникальный ключ на основе всех параметров, кроме ключа API
+    # Создаем уникальный ключ на основе всех параметров
     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
@@ -47,8 +48,8 @@ def scale_dimensions(width, height, max_width, max_height):
     height = (height + 3) // 8 * 8
     return width, height
 
-@cached(cache)
-def generate_cached_image(cache_key, prompt, width, height, seed, model_name, api_key):
+@cached(cache, key=lambda prompt, width, height, seed, model_name, api_key: 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"
@@ -117,11 +118,8 @@ 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 = generate_cached_image(cache_key, prompt, width, height, seed, model_name, api_key)
+        image = generate_cached_image(prompt, width, height, seed, model_name, api_key)
         if image is None:
             return send_error_image()
     except Exception as e: