Update app.py

app.py

@@ -22,6 +22,9 @@ BLENDER_EXEC = os.path.join(BLENDER_INSTALL_DIR, "blender")
 BLENDER_EXEC_SYMLINK = "/usr/local/bin/blender" # Fallback symlink
 
 SETUP_SCRIPT = os.path.join(os.path.dirname(__file__), "setup_blender.sh")
+# ** INCREASED TIMEOUT FOR SETUP SCRIPT **
+SETUP_SCRIPT_TIMEOUT = 1800 # 30 minutes (increased from 600 seconds)
+
 
 # --- Initial Checks ---
 print("--- Environment Checks ---")
@@ -37,7 +40,13 @@ else:
     if os.path.exists(SETUP_SCRIPT):
         try:
             # Run setup script if Blender not found
-            setup_result = subprocess.run(["bash", SETUP_SCRIPT], check=True, capture_output=True, text=True, timeout=600) # 10 min timeout for setup
+            setup_result = subprocess.run(
+                ["bash", SETUP_SCRIPT],
+                check=True,
+                capture_output=True,
+                text=True,
+                timeout=SETUP_SCRIPT_TIMEOUT # Use the increased timeout
+            )
             print("Setup script executed successfully.")
             print(f"Setup STDOUT:\n{setup_result.stdout}")
             if setup_result.stderr: print(f"Setup STDERR:\n{setup_result.stderr}")
@@ -55,8 +64,8 @@ else:
                  raise RuntimeError(f"Setup script ran but Blender executable still not found at {BLENDER_EXEC} or {BLENDER_EXEC_SYMLINK}.")
 
         except subprocess.TimeoutExpired:
-             print(f"ERROR: Setup script timed out: {SETUP_SCRIPT}")
-             raise gr.Error("Setup script timed out. The Space might be too slow or setup is stuck.")
+             print(f"ERROR: Setup script timed out after {SETUP_SCRIPT_TIMEOUT} seconds: {SETUP_SCRIPT}")
+             raise gr.Error(f"Setup script timed out after {SETUP_SCRIPT_TIMEOUT // 60} minutes. The Space might be too slow or setup is stuck. Check full logs for the last operation.")
         except subprocess.CalledProcessError as e:
             print(f"ERROR running setup script: {SETUP_SCRIPT}\nStderr: {e.stderr}")
             # Raise a Gradio error to notify the user
@@ -227,7 +236,7 @@ def run_unirig_command(python_script_path: str, script_args: List[str], step_nam
             text=True,
             check=True, # Raises CalledProcessError on non-zero exit codes
             env=process_env, # Pass the modified environment
-            timeout=1800 # 30 minutes timeout
+            timeout=1800 # 30 minutes timeout for UniRig steps
         )
         print(f"{step_name} STDOUT:\n{result.stdout}")
         if result.stderr:
@@ -408,12 +417,14 @@ print("--- End Diagnostic Info ---")
         # --- Execute UniRig Steps ---
         # Step 1: Skeleton Prediction
         print("\nStarting Step 1: Predicting Skeleton...")
+        # Arguments for run.py using Hydra format (key=value)
         skeleton_args = [
-            "--config-name=skeleton_config",
-            "with",
+            "--config-name=skeleton_config", # Hydra main config
+            "with", # Hydra keyword for overrides
             f"input={abs_input_glb_path}",
             f"output={abs_skeleton_output_path}",
             unirig_device_arg
+            # Add other args like "verbose=True" if needed for debugging
         ]
         run_unirig_command(unirig_script_to_run, skeleton_args, "Skeleton Prediction")
         if not os.path.exists(abs_skeleton_output_path):
@@ -425,7 +436,7 @@ print("--- End Diagnostic Info ---")
         skin_args = [
             "--config-name=skin_config",
             "with",
-            f"input={abs_skeleton_output_path}",
+            f"input={abs_skeleton_output_path}", # Input is the predicted skeleton
             f"output={abs_skin_output_path}",
             unirig_device_arg
         ]
@@ -439,10 +450,10 @@ print("--- End Diagnostic Info ---")
         merge_args = [
             "--config-name=merge_config",
             "with",
-            f"source_path={abs_skin_output_path}",
-            f"target_path={abs_input_glb_path}",
+            f"source_path={abs_skin_output_path}", # Merge skinned skeleton
+            f"target_path={abs_input_glb_path}",   # With original mesh geometry/attrs
             f"output_path={abs_final_rigged_glb_path}",
-            "mode=skin",
+            "mode=skin", # Specify merge mode
             unirig_device_arg
         ]
         run_unirig_command(unirig_script_to_run, merge_args, "Merging Results")