Update app.py

app.py

@@ -105,7 +105,8 @@ def preprocess_image(image_array, session_id):
     if image_array is None:
         return "❌ Please upload an image first.", None, gr.update(interactive=True), gr.update(interactive=True)
 
-    os.makedirs(os.environ["PIXEL3DMM_PREPROCESSED_DATA"], session_id, exist_ok=True)
+    base_dir = os.path.join(os.environ["PIXEL3DMM_PREPROCESSED_DATA"], session_id)
+    os.makedirs(base_dir, exist_ok=True)
 
     img = Image.fromarray(image_array)
     saved_image_path = os.path.join(os.environ["PIXEL3DMM_PREPROCESSED_DATA"], session_id, f"{session_id}.png")
@@ -117,7 +118,7 @@ def preprocess_image(image_array, session_id):
         ], check=True, capture_output=True, text=True)
     except subprocess.CalledProcessError as e:
         err = f"❌ Preprocess failed (exit {e.returncode}).\n\n{e.stdout}\n{e.stderr}"
-        shutil.rmtree(os.environ["PIXEL3DMM_PREPROCESSED_DATA"], session_id)
+        shutil.rmtree(base_dir, ignore_errors=True)
         return err, None, {}, gr.update(interactive=True), gr.update(interactive=True)
 
     crop_dir = os.path.join(os.environ["PIXEL3DMM_PREPROCESSED_DATA"], session_id, "cropped")