Update app.py

app.py

@@ -11,6 +11,16 @@ import spaces
 # Path to the cloned UniRig repository directory within the Space
 UNIRIG_REPO_DIR = os.path.join(os.path.dirname(__file__), "UniRig")
 
+# Path to the setup script
+SETUP_SCRIPT = os.path.join(os.path.dirname(__file__), "setup_blender.sh")
+
+# Check if Blender is installed
+if not os.path.exists("/usr/local/bin/blender"):
+    print("Blender not found. Installing...")
+    subprocess.run(["bash", SETUP_SCRIPT], check=True)
+else:
+    print("Blender is already installed.")
+
 if not os.path.isdir(UNIRIG_REPO_DIR):
     print(f"ERROR: UniRig repository not found at {UNIRIG_REPO_DIR}. Please clone it there.")
     # Consider raising an error or displaying it in the UI if UniRig is critical for startup
@@ -52,7 +62,6 @@ def run_unirig_command(command_list, step_name):
     process_env["PYTHONPATH"] = os.pathsep.join(filter(None, new_pythonpath_parts)) # filter(None,...) handles empty existing_pythonpath
     print(f"Set PYTHONPATH for subprocess: {process_env['PYTHONPATH']}")
 
-
     try:
         # Execute the command from the UniRig directory (UNIRIG_REPO_DIR)
         # This is crucial for the bash scripts to find their relative paths (e.g., to Python scripts)
@@ -78,7 +87,6 @@ def run_unirig_command(command_list, step_name):
         print(f"An unexpected Python exception occurred in run_unirig_command for {step_name}: {e_general}")
         raise gr.Error(f"Unexpected Python error during {step_name}: {str(e_general)[:500]}")
 
-
 @spaces.GPU # Decorator for ZeroGPU
 def rig_glb_mesh_multistep(input_glb_file_obj):
     """
@@ -161,7 +169,6 @@ def rig_glb_mesh_multistep(input_glb_file_obj):
             print(f"Cleaned up temporary directory: {processing_temp_dir}")
         raise gr.Error(f"An unexpected error occurred during processing: {str(e)[:500]}")
 
-
 # --- Gradio Interface ---
 theme = gr.themes.Soft(
     primary_hue=gr.themes.colors.sky,