Update app.py

app.py

@@ -5,28 +5,24 @@ from PIL import Image
 from deep_translator import GoogleTranslator
 import aiohttp
 from quart import Quart, request, jsonify, send_file, render_template_string
-from werkzeug.contrib.cache import SimpleCache
+from flask_caching import Cache  # flask-cachingを使用
 
 # アプリケーションの設定
 app = Quart(__name__)
-cache = SimpleCache()
+
+# キャッシュの設定
+cache_config = {
+    "CACHE_TYPE": "SimpleCache",  # メモリベースのシンプルなキャッシュ
+    "CACHE_DEFAULT_TIMEOUT": 60 * 60 * 24  # 24時間
+}
+app.config.from_mapping(cache_config)
+cache = Cache(app)
 
 API_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-dev"
 API_TOKEN = os.getenv("HF_READ_TOKEN")
 headers = {"Authorization": f"Bearer {API_TOKEN}"}
 timeout = 50000  # タイムアウトを300秒に設定
 
-# キャッシュの設定
-def get_cached_image(prompt):
-    cached = cache.get(prompt)
-    if cached is not None:
-        return cached
-    return None
-
-def set_cache_image(prompt, image):
-    cache.set(prompt, image, timeout=60*60*24)  # キャッシュを24時間保存
-
-
 # 非同期APIリクエストの実行関数
 async def query_async(prompt, negative_prompt="", steps=35, cfg_scale=7, sampler="DPM++ 2M Karras", seed=-1, strength=0.7, width=1024, height=1024, num_inference_steps=30, guidance_scale=7.5, top_k=50, top_p=0.9, eta=0.1):
     if not prompt:
@@ -127,9 +123,9 @@ async def generate_image():
     eta = float(request.args.get("eta", 0.1))
 
     # キャッシュを確認
-    cached_image = get_cached_image(prompt)
+    cached_image = cache.get(prompt)
     if cached_image:
-        return await send_file(cached_image, mimetype='image/png')
+        return await send_file(io.BytesIO(cached_image), mimetype='image/png')
 
     image, error = await query_async(prompt, negative_prompt, steps, cfg_scale, sampler, seed, strength, width, height, num_inference_steps, guidance_scale, top_k, top_p, eta)
 
@@ -141,7 +137,7 @@ async def generate_image():
     img_bytes.seek(0)
 
     # 画像をキャッシュに保存
-    set_cache_image(prompt, img_bytes)
+    cache.set(prompt, img_bytes.getvalue())
 
     return await send_file(img_bytes, mimetype='image/png')