Update app.py

app.py

@@ -13,6 +13,9 @@ from datetime import datetime
 import gradio as gr
 from typing import Tuple
 import time
+# 상단에 Path import 추가
+from pathlib import Path
+
 
 # 스크린샷 캐시 디렉토리 설정
 CACHE_DIR = Path("screenshot_cache")
@@ -21,7 +24,18 @@ CACHE_DIR.mkdir(exist_ok=True)
 
 # 전역 변수로 스크린샷 캐시 선언
 SCREENSHOT_CACHE = {}
+# get_cached_screenshot 함수 추가
 
+def get_cached_screenshot(url: str) -> str:
+    """캐시된 스크린샷 가져오기 또는 새로 생성"""
+    cache_file = CACHE_DIR / f"{base64.b64encode(url.encode()).decode()}.png"
+    
+    if cache_file.exists():
+        with open(cache_file, "rb") as f:
+            return base64.b64encode(f.read()).decode()
+            
+    return take_screenshot(url)
+    
 def take_screenshot(url):
     """웹사이트 스크린샷 촬영 함수 (로딩 대기 시간 추가)"""
     if url in SCREENSHOT_CACHE: