Upgrade run_cmd

app.py

@@ -1,24 +1,32 @@
 import os
 os.system("pip install gradio==2.9b23")
-#https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/13985
-#os.system("python -m pip install torch==2.0.1 torchvision==0.15.2")
+
 import random
 import gradio as gr
 from PIL import Image
 import torch
 from random import randint
 import sys
-from subprocess import call
 import psutil
+import subprocess
 
-
-
-  
-    
 def run_cmd(command):
     try:
-        print(command)
-        call(command, shell=True)
+        print(f"Running command: {command}")
+        # Run the command and capture both output and error
+        result = subprocess.run(command, shell=True, text=True, capture_output=True)
+
+        # Print stdout and stderr
+        if result.stdout:
+            print("Output:\n", result.stdout)
+        
+        if result.stderr:
+            print("Error:\n", result.stderr)
+        
+        # Check for command success
+        if result.returncode != 0:
+            print(f"Command failed with return code {result.returncode}")
+
     except KeyboardInterrupt:
         print("Process interrupted")
         sys.exit(1)
@@ -35,6 +43,9 @@ def inference(img,mode):
     _id = randint(1, 10000)
     INPUT_DIR = "./tmp/input_image" + str(_id) + "/"
     OUTPUT_DIR = "./tmp/output_image" + str(_id) + "/"
+
+    run_cmd("ls -l ./tmp/*/*")
+
     run_cmd("rm -rf " + INPUT_DIR)
     run_cmd("rm -rf " + OUTPUT_DIR)
     run_cmd("mkdir -p " + INPUT_DIR)