clean up tmp dir after each run

app.py

@@ -197,6 +197,22 @@ def add_extra_model_paths() -> None:
         print("Could not find the extra_model_paths config file.")
 
 
+def clear_tmp_directory():
+    print("Cleaning up /tmp directory...")
+    image_extensions = {'.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp'}
+    for root, dirs, files in os.walk('/tmp'):
+        for file in files:
+            if not file.startswith('advance-blurred-'):
+                ext = os.path.splitext(file)[1].lower()
+                if ext in image_extensions:
+                    full_path = os.path.join(root, file)
+                    try:
+                        os.remove(full_path)
+                        print(f"Deleted: {full_path}")
+                    except Exception as e:
+                        print(f"Failed to delete {full_path}: {e}")
+
+
 add_comfyui_directory_to_sys_path()
 add_extra_model_paths()
 
@@ -262,6 +278,10 @@ def advance_blur(input_image):
         )
         outpath = f"advance-blurred-{os.urandom(16).hex()}.jpg"
         img.save(outpath, quality=80, dpi=(72, 72))
+
+        # Clean up the /tmp directory
+        clear_tmp_directory()
+
         return outpath