Update app.py

app.py

@@ -6,6 +6,7 @@ from cachetools import TTLCache, cached
 import random
 import os
 import urllib.parse
+import hashlib
 
 app = Flask(__name__)
 
@@ -20,8 +21,12 @@ if not keys:
 def get_random_key():
     return random.choice(keys)
 
+def generate_cache_key(prompt, width, height, seed, model_name):
+    # Создаем уникальный ключ на основе всех параметров
+    return hashlib.md5(f"{prompt}_{width}_{height}_{seed}_{model_name}".encode()).hexdigest()
+
 @cached(cache)
-def generate_cached_image(prompt, width, height, seed, model_name):
+def generate_cached_image(cache_key, prompt, width, height, seed, model_name):
     api_key = get_random_key()
     headers = {
         "Authorization": f"Bearer {api_key}",
@@ -36,8 +41,7 @@ def generate_cached_image(prompt, width, height, seed, model_name):
         },
         "width": width,
         "height": height,
-        "seed": seed,
-      #  generator=torch.Generator("cpu").manual_seed(0)
+        "seed": seed
     }
 
     try:
@@ -60,8 +64,11 @@ def get_image(prompt):
     # Декодируем URL-кодированный prompt
     prompt = urllib.parse.unquote(prompt)
 
+    # Генерируем уникальный ключ для кэша
+    cache_key = generate_cache_key(prompt, width, height, seed, model_name)
+
     try:
-        image = generate_cached_image(prompt, width, height, seed, model_name)
+        image = generate_cached_image(cache_key, prompt, width, height, seed, model_name)
     except Exception as e:
         app.logger.error(f"Error generating image: {e}")
         abort(500, description="Error generating image")