Update app.py

app.py

@@ -215,10 +215,7 @@ def step4_track(session_id):
 
 # New: run all steps sequentially
 @spaces.GPU()
-def run_pipeline(image_array, request: gr.Request):
-
-    session_id = request.session_hash
-
+def generate_results_and_mesh(image, session_id):
     # Step 1
     status1, crop_img, _, _ = preprocess_image(image_array)
     if "❌" in status1:
@@ -245,6 +242,11 @@ def cleanup(request: gr.Request):
         shutil.rmtree(d1, ignore_errors=True)
         shutil.rmtree(d2, ignore_errors=True)
 
+def start_session(request: gr.Request):
+    session_hash = request.session_hash
+
+    return {"session_id": session_hash}
+    
 css = """
 #col-container {
     margin: 0 auto;
@@ -298,11 +300,13 @@ with gr.Blocks(css=css) as demo:
                     fn=run_pipeline,
                     cache_examples=True  # cache results of examples for faster loading
                 )
-    
+
+        session_state = gr.State()
+        demo.load(start_session, outputs=[session_state])
 
         run_btn.click(
-            fn=run_pipeline,
-            inputs=[image_in],
+            fn=generate_results_and_mesh,
+            inputs=[image_in, session_state["session_id"]],
             outputs=[status, crop_img, normals_img, uv_img, track_img, mesh_file]
         )
         examples.outputs = [status, crop_img, normals_img, uv_img, track_img, mesh_file]