Update app.py

app.py

@@ -1,14 +1,32 @@
+import gradio as gr
 import shutil
 import os
+import csv
 
-# 캐시된 예제 파일 디렉토리
-cache_dir_14 = "/home/user/app/gradio_cached_examples/14"
-cache_dir_28 = "/home/user/app/gradio_cached_examples/28"
-
-# 디렉토리가 존재하면 삭제
-for cache_dir in [cache_dir_14, cache_dir_28]:
-    if os.path.exists(cache_dir):
-        shutil.rmtree(cache_dir)
-        print(f"{cache_dir} has been removed.")
-    else:
-        print(f"{cache_dir} does not exist.")
+# CSV 필드 크기 제한 늘리기
+csv.field_size_limit(10**6)
+
+# 캐시 파일 삭제 함수
+def clear_cache():
+    cache_dir_14 = "/home/user/app/gradio_cached_examples/14"
+    cache_dir_28 = "/home/user/app/gradio_cached_examples/28"
+    
+    removed_dirs = []
+    for cache_dir in [cache_dir_14, cache_dir_28]:
+        if os.path.exists(cache_dir):
+            shutil.rmtree(cache_dir)
+            removed_dirs.append(f"{cache_dir} has been removed.")
+        else:
+            removed_dirs.append(f"{cache_dir} does not exist.")
+    return "\n".join(removed_dirs)
+
+# Gradio 인터페이스
+with gr.Blocks() as demo:
+    gr.Markdown("## Gradio Cache Cleaner")
+    clear_button = gr.Button("Clear Cache")
+    output = gr.Textbox()
+
+    clear_button.click(fn=clear_cache, outputs=output)
+
+if __name__ == "__main__":
+    demo.launch()