Update start_jupyterlab.py

start_jupyterlab.py

@@ -1,6 +1,7 @@
 import os
 import subprocess
 import sys
+import threading
 
 def modify_jupyter_config():
     """修改 Jupyter 配置文件以允许远程访问并禁用 token"""
@@ -41,17 +42,38 @@ def modify_jupyter_config():
         print(f"修改配置文件时出错: {e}")
 
 def start_jupyter_lab():
-    """启动 JupyterLab 服务"""
+    """启动 JupyterLab 服务并将输出重定向到主窗口"""
     try:
         print("启动 JupyterLab 服务...")
+
         command = [
             sys.executable, "-m", "jupyter", "lab",
             "--no-browser",          # 禁止自动打开浏览器
             "--ip=0.0.0.0",          # 允许外部访问
-            "--port=18888",           # 指定端口为 18888
+            "--port=18888",          # 指定端口
         ]
-        subprocess.Popen(command, stdout=sys.stdout, stderr=sys.stderr)
-        print("JupyterLab 已启动，请通过 http://<服务器IP>:8888 访问。")
+
+        process = subprocess.Popen(
+            command,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True,
+            bufsize=1
+        )
+
+        def stream_output(pipe, pipe_name):
+            """实时打印子进程的输出"""
+            with pipe:
+                for line in pipe:
+                    print(f"[{pipe_name}] {line.strip()}")
+
+        # 分别启动两个线程，读取标准输出和标准错误
+        threading.Thread(target=stream_output, args=(process.stdout, "STDOUT"), daemon=True).start()
+        threading.Thread(target=stream_output, args=(process.stderr, "STDERR"), daemon=True).start()
+
+        # 等待子进程结束
+        process.wait()
+        print("JupyterLab 服务已停止。")
     except Exception as e:
         print(f"启动 JupyterLab 时出错: {e}")
 
@@ -60,4 +82,4 @@ if __name__ == "__main__":
     modify_jupyter_config()
     
     # 2. 启动 JupyterLab
-    start_jupyter_lab()
+    start_jupyter_lab()