Update app.py

app.py

@@ -5,6 +5,15 @@ import shutil
 import gradio as gr
 import time
 
+def clear_cache():
+    """Hugging Faceキャッシュを削除する関数。"""
+    cache_dir = os.path.expanduser("~/.cache/huggingface")
+    if os.path.exists(cache_dir):
+        print(f"Clearing cache at: {cache_dir}")
+        shutil.rmtree(cache_dir)
+    else:
+        print("No cache found to clear.")
+
 def process_and_zip_folders(huggingface_dataset_url, output_dir):
     # URLからリポジトリIDを抽出
     if huggingface_dataset_url.startswith("https://huggingface.co/"):
@@ -19,6 +28,9 @@ def process_and_zip_folders(huggingface_dataset_url, output_dir):
     dataset_path = None
     for attempt in range(retry_attempts):
         try:
+            if attempt > 0:
+                print(f"Retrying... (Attempt {attempt + 1})")
+                clear_cache()  # キャッシュ削除
             dataset_path = snapshot_download(repo_id, repo_type="dataset", force_download=True)
             break  # 成功した場合はループを終了
         except OSError as e:
@@ -63,6 +75,9 @@ def process_and_zip_folders(huggingface_dataset_url, output_dir):
 
 def gradio_interface():
     def start_process(huggingface_url, output_directory):
+        # 出力ディレクトリが存在しない場合は作成
+        if not os.path.exists(output_directory):
+            os.makedirs(output_directory, exist_ok=True)
         zip_files = process_and_zip_folders(huggingface_url, output_directory)
         return zip_files