Update app.py

app.py

@@ -102,7 +102,7 @@ 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(base_dir, exist_ok=True)
+    os.makedirs(os.environ["PIXEL3DMM_PREPROCESSED_DATA"], session_id, 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")
@@ -114,10 +114,10 @@ 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(base_dir)
+        shutil.rmtree(os.environ["PIXEL3DMM_PREPROCESSED_DATA"], session_id)
         return err, None, {}, gr.update(interactive=True), gr.update(interactive=True)
 
-    crop_dir = os.path.join(base_dir, "cropped")
+    crop_dir = os.path.join(os.environ["PIXEL3DMM_PREPROCESSED_DATA"], session_id, "cropped")
     image = first_image_from_dir(crop_dir)
     return "✅ Step 1 complete. Ready for Normals.", image, gr.update(interactive=True), gr.update(interactive=True)