Update app.py

app.py

@@ -16,8 +16,14 @@ def zip_folder(dataset_name, folder_name, output_zip):
     Returns:
         str: ZIPファイルのパス。
     """
-    # データセットをロード
-    dataset = load_dataset(dataset_name)
+    try:
+        # データセットをロード
+        print(f"ロード対象データセット: {dataset_name}")
+        dataset = load_dataset(dataset_name)
+    except FileNotFoundError:
+        return f"指定されたデータセット '{dataset_name}' が見つかりません。"
+    except Exception as e:
+        return f"エラーが発生しました: {str(e)}"
 
     # 一時ディレクトリに保存
     temp_dir = "temp_dataset"
@@ -31,6 +37,7 @@ def zip_folder(dataset_name, folder_name, output_zip):
     # フォルダパスを指定
     folder_path = os.path.join(temp_dir, folder_name)
     if not os.path.exists(folder_path):
+        shutil.rmtree(temp_dir)
         return f"指定されたフォルダ {folder_name} が見つかりませんでした。"
 
     # ZIPファイルに圧縮