diff --git a/comfyui/main.py b/comfyui/main.py
index c3ef9e3..0c8cc08 100644
--- a/comfyui/main.py
+++ b/comfyui/main.py
@@ -146,7 +146,11 @@ def checkIsFinish(driver, pid, orderId,styleNumber):
 #清除指定文件夹下的所有文件
 def clearFolder(folder):
     for file in os.listdir(folder):
-        os.remove(os.path.join(folder, file))
+        #判断如果是文件夹，则整个删除
+        if os.path.isdir(os.path.join(folder, file)):
+            shutil.rmtree(os.path.join(folder, file))
+        else:
+            os.remove(os.path.join(folder, file))
 
 
 #检测输入目录里是否有输入的文件存在