Update app.py

app.py

@@ -5,6 +5,7 @@ from typing import Dict, List, Optional, Tuple
 import base64
 import anthropic
 from functools import partial, lru_cache
+import json
 
 import gradio as gr
 import modelscope_studio.components.base as ms
@@ -12,6 +13,10 @@ import modelscope_studio.components.legacy as legacy
 import modelscope_studio.components.antd as antd
 from config import DEMO_LIST, SystemPrompt
 
+# 캐시 저장소
+DEMO_CACHE = {}
+
+
 
 # 파일 상단의 import 문 아래에 추가
 def get_image_base64(image_path):
@@ -64,15 +69,27 @@ def send_to_sandbox(code):
     return f"<iframe src=\"{data_uri}\" width=\"100%\" height=\"920px\"></iframe>"
 
 
-# 캐시 구현
-@lru_cache(maxsize=len(DEMO_LIST))
-def get_demo_description(index: int) -> str:
-    return DEMO_LIST[index]['description']
+# 캐시된 데모 설명을 가져오는 함수
+def get_cached_demo_description(index: int) -> str:
+    if index not in DEMO_CACHE:
+        DEMO_CACHE[index] = DEMO_LIST[index]['description']
+        # 캐시 파일에 저장
+        with open('demo_cache.json', 'w', encoding='utf-8') as f:
+            json.dump(DEMO_CACHE, f, ensure_ascii=False, indent=2)
+    return DEMO_CACHE[index]
+
+# 시작할 때 캐시 로드
+try:
+    with open('demo_cache.json', 'r', encoding='utf-8') as f:
+        DEMO_CACHE = json.load(f)
+except FileNotFoundError:
+    # 처음 실행시 캐시 초기화
+    for i in range(len(DEMO_LIST)):
+        get_cached_demo_description(i)
 
 def demo_card_click(e: gr.EventData):
     index = e._data['component']['index']
-    return get_demo_description(index)
-
+    return get_cached_demo_description(index)
 
 with gr.Blocks(css_paths="app.css") as demo:
     history = gr.State([])