Update app.py

app.py

@@ -1,25 +1,38 @@
-import subprocess, shlex
+import subprocess
 import os
 
-# 定义命令列表（每个命令作为一个子进程运行）
+
+BASE_DIR = Path("/home/user/app/JarvisIR")
+
 commands = [
-    "cd JarvisIR && pip install -r requirements.txt",
-    "cd JarvisIR/package/agent_tools/Retinexformer && python3 setup.py develop --no_cuda_ext",
-    "cd JarvisIR/package/agent_tools/RIDCP && python3 setup.py develop",
-    "cd JarvisIR/package && pip install -e .",
-    "pip install basicsr"
+    ("pip install -r requirements.txt", BASE_DIR),
+    ("python3 setup.py develop --no_cuda_ext", BASE_DIR / "package/agent_tools/Retinexformer"),
+    ("python3 setup.py develop", BASE_DIR / "package/agent_tools/RIDCP"),
+    ("pip install -e .", BASE_DIR / "package")
 ]
 
-# 执行所有命令
-for cmd in commands:
+def run_command(cmd, cwd=None):
     try:
-        subprocess.run(shlex.split(cmd))
+        result = subprocess.run(
+            cmd,  # 注意：这里不再使用 shlex.split()
+            cwd=str(cwd) if cwd else None,
+            shell=True,  # 需要 shell=True 来支持 && 等操作符
+            check=True,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True
+        )
+        print(f"[SUCCESS] {cmd}")
+        if result.stdout: print(result.stdout)
+        return True
     except subprocess.CalledProcessError as e:
-        print(f"Command failed: {cmd}")
+        print(f"[FAILED] {cmd}")
         print(f"Error: {e.stderr}")
-        # 可以选择是否在此处退出或继续
-        # raise  # 取消注释以抛出异常停止程序
+        return False
 
+for cmd, cwd in commands:
+    run_command(cmd, cwd)
+    
 import spaces
 import re
 import random