Update app.py

app.py

@@ -102,7 +102,7 @@ def preprocess_image(image_array, state):
     if image_array is None:
         return "❌ Please upload an image first.", None, state, gr.update(interactive=True), gr.update(interactive=True)
 
-    session_id = str(uuid.uuid4())
+    session_id = request.session_hash
     base_dir = os.path.join(os.environ["PIXEL3DMM_PREPROCESSED_DATA"], session_id)
     os.makedirs(base_dir, exist_ok=True)
     state.update({"session_id": session_id, "base_dir": base_dir})
@@ -237,8 +237,8 @@ def run_pipeline(image_array, state):
     return final_status, crop_img, normals_img, uv_img, track_img, mesh_file, state
 
 # Cleanup on unload
-def cleanup(state):
-    sid = state.get("session_id")
+def cleanup(request: gr.Request):
+    sid = request.session_hash
     if sid:
         d1 = os.path.join(env["PIXEL3DMM_PREPROCESSED_DATA"], sid)
         d2 = os.path.join(env["PIXEL3DMM_TRACKING_OUTPUT"], sid)
@@ -293,7 +293,7 @@ with demo:
     examples.outputs = [status, crop_img, normals_img, uv_img, track_img, mesh_file, state]
     image_in.upload(fn=reset_all, inputs=None, outputs=[crop_img, normals_img, uv_img, track_img, mesh_file, status, state, run_btn])
 
-demo.on_close(cleanup, inputs=[state], outputs=[])
+    demo.unload(cleanup)
 
 demo.queue()
 demo.launch(share=True, ssr_mode=False)