Zip and download

app.py

@@ -124,6 +124,9 @@ def zip_and_download():
     temp_dir = tempfile.mkdtemp()
     zip_path = os.path.join(temp_dir, ZIP_NAME)
     
+    if not os.path.exists(FOLDER_TO_ZIP):
+        raise gr.Error("WORKSPACE Folder does not exist!")
+
     # Create ZIP archive
     shutil.make_archive(zip_path.replace(".zip", ""), "zip", FOLDER_TO_ZIP)
     if os.path.exists(FOLDER_TO_ZIP):