Upload app.py

app.py

@@ -1,14 +1,31 @@
 import subprocess
+import select
 from hf_api import restart_space
 
 try:
     # 启动另一个程序，并通过管道捕获其输出
-    process = subprocess.Popen(["python", "sub_app.py"], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-    while True:
-        output = process.stdout.readline()
-        if output:
-            print(output.decode("utf-8").strip())
-        if process.poll() is not None:
-            break
+    process = subprocess.Popen(["python", "sub_app.py"],
+                               stdout=subprocess.PIPE,
+                               stderr=subprocess.PIPE,
+                               bufsize=1, universal_newlines=True)
+    while process.poll() is None:
+        # 使用 select 模块检查是否有可读数据
+        ready_reads, _, _ = select.select([process.stdout, process.stderr], [], [], 1.0)
+        for ready in ready_reads:
+            # 读取输出并打印
+            output = ready.readline()
+            if output:
+                print(output, end='')
+
+    # 读取剩余的输出
+    for output in process.stdout.readlines() + process.stderr.readlines():
+        print(output, end='')
+
+    # 检查进程的返回代码以确定是否成功结束
+    if process.returncode == 0:
+        print("Process has terminated successfully.")
+    else:
+        print(f"Process has terminated with an error. {process.returncode}")
+
 finally:
     restart_space()