Update app.py

app.py

@@ -195,6 +195,31 @@ def run_unirig_command(python_script_path: str, script_args: List[str], step_nam
     process_env["LD_LIBRARY_PATH"] = f"{blender_lib_path}{os.pathsep}{existing_ld_path}" if existing_ld_path else blender_lib_path
     print(f"Subprocess LD_LIBRARY_PATH: {process_env['LD_LIBRARY_PATH']}")
 
+    # Test import of 'src' module
+    test_cmd = [
+        blender_executable_to_use,
+        "--background",
+        "--python", "-c", "import src; print('src module imported successfully')"
+    ]
+    test_result = subprocess.run(
+        test_cmd,
+        cwd=UNIRIG_REPO_DIR,
+        capture_output=True,
+        text=True,
+        env=process_env
+    )
+    if test_result.returncode != 0:
+        print(f"Test import of 'src' failed with return code {test_result.returncode}")
+        print(f"Test STDOUT: {test_result.stdout}")
+        print(f"Test STDERR: {test_result.stderr}")
+        raise gr.Error("Failed to import 'src' module in Blender environment. Check PYTHONPATH and module structure.")
+    elif "src module imported successfully" not in test_result.stdout:
+        print(f"Test import of 'src' did not produce expected output.")
+        print(f"Test STDOUT: {test_result.stdout}")
+        print(f"Test STDERR: {test_result.stderr}")
+        raise gr.Error("Unexpected output from 'src' import test. Check script behavior.")
+    else:
+        print("Test import of 'src' succeeded.")
 
     try:
         # Execute Blender with the Python script.
@@ -234,7 +259,6 @@ def run_unirig_command(python_script_path: str, script_args: List[str], step_nam
         else:
              raise gr.Error(f"Error in UniRig '{step_name}'. Check logs. Last error lines:\n{last_lines}")
 
-
     except FileNotFoundError:
         # This error means blender executable or the python script wasn't found
         print(f"ERROR: Could not find Blender executable '{blender_executable_to_use}' or script '{python_script_path}' for {step_name}.")