Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import subprocess
+from contextlib import redirect_stdout
 
 import numpy as np
 import matplotlib.pyplot as plt
@@ -54,28 +54,18 @@ def run_all(tickersymbol, download_data, download_update, use_train, method, PRE
     yield
 
 
-def run_command():
-    # 這裡以執行一段簡單的 ping 指令或 python 腳本為例
-    # 你可以替換成你實際要在 container 內執行的指令
-    command = ["python", "-c", "import time; print('Starting...'); time.sleep(1); print('Running...'); time.sleep(1); print('Done!')"]
+def run_python_code():
+    # 建立一個字串 IO 來接收輸出
+    f = io.StringIO()
     
-    # 使用 Popen 捕捉 stdout 和 stderr
-    process = subprocess.Popen(
-        command,
-        stdout=subprocess.PIPE,
-        stderr=subprocess.STDOUT, # 將錯誤訊息導向標準輸出
-        text=True,                # 將輸出解碼為字串
-        bufsize=1                 # 行緩衝 (Line buffered)
-    )
-    
-    full_output = ""
-    # 逐行讀取輸出並即時回傳給 Gradio
-    for line in iter(process.stdout.readline, ''):
-        full_output += line
-        yield full_output  # yield 會即時更新前端的 Textbox
+    with redirect_stdout(f):
+        # 這裡放你原本會 print 東西的程式碼
+        print("載入模型中...")
+        print("處理資料...")
+        print("執行完成！")
         
-    process.stdout.close()
-    process.wait()
+    # 獲取所有被捕捉到的輸出
+    return f.getvalue()
 
     
 def update_setup(is_checked):
@@ -113,11 +103,11 @@ def update_ticker_list(market_choice):
     return gr.update(choices=["No tickers loaded"], value="No tickers loaded")
 
 symbol_list = ["No tickers loaded"]
-
 today = datetime.date.today()
 yesterday = today - datetime.timedelta(days=1)
 tomorow = today + datetime.timedelta(days=1)
 
+
 with gr.Blocks() as demo:
     gr.Markdown("### 股票分析與訓練設定")
     
@@ -170,6 +160,7 @@ with gr.Blocks() as demo:
    
     with gr.Row():
         run_btn = gr.Button("啟動")
+        console_output = gr.Textbox(label="Console Output", lines=5, interactive=False)
         
         ##--觸發事件--
     market_radio.change(
@@ -214,7 +205,7 @@ with gr.Blocks() as demo:
         outputs = [config_display] # 這裡依序對應 yield 的回傳值
     )#執行按鈕的邏輯
     run_btn.click(
-        fn=run_command, 
+        fn=run_python_code, 
         inputs=[], 
         outputs=console_output
     )