Update app.py

app.py

@@ -1,126 +1,180 @@
 # -*- coding: utf-8 -*-
 import gradio as gr
 import asyncio
-import json
 import threading
+import json
 import time
 from datetime import datetime
 
-# 全局控制
+# 全局状态
+引擎实例 = None
 运行线程 = None
-运行状态 = {"运行中": False, "净值": 1.0, "回撤": 0.0, "日志": []}
-
-def 启动模拟(合约, 周期, 参数json):
-    global 运行线程, 运行状态
-    
-    if 运行状态["运行中"]:
-        return "已有任务运行中", "", ""
-    
-    # 解析参数
+状态锁 = threading.Lock()
+运行状态 = {
+    "运行中": False,
+    "合约": "",
+    "周期": 0,
+    "净值": 1.0,
+    "回撤": 0.0,
+    "日志": []
+}
+
+# 默认参数（示例，可直接替换为你的最优参数JSON）
+默认参数 = {
+    "ver": "v1",
+    "adx_thr": 25,
+    "atr_buf_tr": 0.1,
+    "rsi_low": 30,
+    "rsi_high": 70,
+    "mr_stop_atr": 2.0,
+    "trend_logic": "adx",
+    "alloc_mode": "fixed"
+}
+
+def _记录(msg: str):
+    with 状态锁:
+        运行状态["日志"].append(f"[{datetime.now().strftime('%H:%M:%S')}] {msg}")
+        # 控制日志长度
+        if len(运行状态["日志"]) > 500:
+            运行状态["日志"] = 运行状态["日志"][-300:]
+
+def _后台任务(合约: str, 周期: int, 参数: dict):
+    """
+    后台线程任务：启动引擎并运行。此函数内开自己的事件循环。
+    """
+    global 引擎实例
     try:
-        参数 = json.loads(参数json) if 参数json else {}
-    except:
-        return "参数JSON格式错误", "", ""
-    
-    # 启动异步任务
-    def 运行任务():
         from 插件_引擎.多策略引擎 import 多策略引擎
         from 插件_交易.策略_自适应 import 自适应策略
-        
+
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+
         async def 主循环():
-            运行状态["运行中"] = True
-            运行状态["日志"] = []
-            
-            引擎 = 多策略引擎()
+            global 引擎实例
+            引擎实例 = 多策略引擎()
             策略 = 自适应策略(f"{合约}_{周期}分", 参数)
-            引擎.注册策略(f"{合约}_{周期}", 策略, 合约, int(周期))
-            
+            引擎实例.注册策略(f"{合约}_{周期}", 策略, 合约, int(周期))
+            _记录(f"引擎启动：{合约} / {周期} 分")
             try:
-                await 引擎.启动([合约])
+                await 引擎实例.启动([合约])
             except Exception as e:
-                运行状态["日志"].append(f"错误: {e}")
+                _记录(f"引擎异常：{e}")
             finally:
-                运行状态["运行中"] = False
-        
-        asyncio.run(主循环())
-    
-    运行线程 = threading.Thread(target=运行任务)
+                with 状态锁:
+                    运行状态["运行中"] = False
+                _记录("引擎结束")
+
+        loop.run_until_complete(主循环())
+    except Exception as e:
+        _记录(f"后台任务异常：{e}")
+        with 状态锁:
+            运行状态["运行中"] = False
+
+def 启动模拟(合约: str, 周期: str, 参数json: str):
+    """
+    点击“启动模拟”时调用
+    """
+    global 运行线程
+    with 状态锁:
+        if 运行状态["运行中"]:
+            return f"已有任务运行中：{运行状态['合约']} / {运行状态['周期']}分", "\n".join(运行状态["日志"][-100:])
+
+    # 解析参数
+    if 参数json.strip():
+        try:
+            参数 = json.loads(参数json)
+        except Exception as e:
+            return f"参数JSON格式错误：{e}", "\n".join(运行状态["日志"][-100:])
+    else:
+        参数 = 默认参数
+
+    # 启动后台线程
+    try:
+        p = int(周期)
+    except:
+        return "周期必须是数字（30/60/120/240）", "\n".join(运行状态["日志"][-100:])
+
+    with 状态锁:
+        运行状态["运行中"] = True
+        运行状态["合约"] = 合约
+        运行状态["周期"] = p
+        运行状态["净值"] = 1.0
+        运行状态["回撤"] = 0.0
+        运行状态["日志"].clear()
+
+    _记录(f"准备启动：合约={合约} 周期={p} 参数={参数}")
+
+    运行线程 = threading.Thread(target=_后台任务, args=(合约, p, 参数), daemon=True)
     运行线程.start()
-    
-    return "模拟启动成功", "", ""
+
+    return "模拟启动成功", "\n".join(运行状态["日志"][-100:])
 
 def 停止模拟():
-    global 运行状态
-    运行状态["运行中"] = False
-    return "已发送停止信号"
+    """
+    点击“停止模拟”时调用
+    说明：这里只设置引擎停止处理信号（不强制关闭WS）。
+    """
+    global 引擎实例
+    if 引擎实例 is None:
+        return "当前没有运行中的任务"
+    try:
+        引擎实例.停止()
+        _记录("已请求停止（引擎将不再处理新信号）")
+        return "已发送停止信��"
+    except Exception as e:
+        _记录(f"停止失败：{e}")
+        return f"停止失败：{e}"
 
 def 获取状态():
-    日志文本 = "\n".join(运行状态["日志"][-50:])  # 最近50条
-    状态文本 = f"运行中: {运行状态['运行中']} | 净值: {运行状态['净值']:.4f} | 回撤: {运行状态['回撤']:.2%}"
+    """
+    定时器/“刷新状态”调用：读取净值、回撤、日志
+    """
+    with 状态锁:
+        # 从引擎里读净值/回撤（如果有）
+        try:
+            if 引擎实例 is not None and 引擎实例.交易所 is not None:
+                # 交易所.净值 会在引擎每根收盘时更新；此处只是读
+                nv = float(getattr(引擎实例.交易所, "净值", 1.0) or 1.0)
+                dd = float(getattr(引擎实例.交易所, "回撤", 0.0) or 0.0)
+                运行状态["净值"] = nv
+                运行状态["回撤"] = dd
+        except Exception as e:
+            _记录(f"读取状态失败：{e}")
+
+        状态文本 = f"运行中: {运行状态['运行中']} | 合约: {运行状态['合约']} | 周期: {运行状态['周期']}m | 净值: {运行状态['净值']:.4f} | 回撤: {运行状态['回撤']:.2%}"
+        日志文本 = "\n".join(运行状态["日志"][-120:])
     return 状态文本, 日志文本
 
-# 默认参数（你的60分钟V1参数）
-默认参数 = {
-    "ver": "v1",
-    "adx_thr": 25,
-    "atr_buf_tr": 0.05,
-    "rsi_low": 30,
-    "rsi_high": 70,
-    "mr_stop_atr": 2.0,
-    "trend_logic": "adx",
-    "alloc_mode": "fixed"
-}
+# Gradio 界面
+with gr.Blocks(title="模拟实盘（OKX 公共WS + 多策略引擎）") as demo:
+    gr.Markdown("# 📈 模拟实盘（订阅OKX公共行情 → 聚合K线 → 策略 → 纸上交易所）")
+    gr.Markdown("提示：此为演示用。停止按钮仅令引擎停止处理新信号，WebSocket仍保持连接，避免强制断开导致报错。")
 
-# Gradio界面
-with gr.Blocks(title="模拟实盘") as demo:
-    gr.Markdown("# 📈 加密货币模拟实盘交易")
-    
     with gr.Row():
-        with gr.Column():
-            合约输入 = gr.Textbox(
-                label="合约ID",
-                value="UXLINK-USDT-SWAP",
-                placeholder="例: BTC-USDT-SWAP"
-            )
-            周期选择 = gr.Radio(
-                label="周期（分钟）",
-                choices=["30", "60", "120", "240"],
-                value="60"
-            )
-            参数输入 = gr.Textbox(
-                label="策略参数（JSON）",
-                value=json.dumps(默认参数, indent=2),
-                lines=10
-            )
-            
+        with gr.Column(scale=1):
+            合约输入 = gr.Textbox(label="合约ID", value="UXLINK-USDT-SWAP", placeholder="例：BTC-USDT-SWAP / ETH-USDT-SWAP")
+            周期选择 = gr.Radio(label="周期（分钟）", choices=["30", "60", "120", "240"], value="60")
+            参数输入 = gr.Textbox(label="策略参数（JSON）", value=json.dumps(默认参数, ensure_ascii=False, indent=2), lines=12)
+
             with gr.Row():
                 启动按钮 = gr.Button("🚀 启动模拟", variant="primary")
                 停止按钮 = gr.Button("⏹️ 停止模拟", variant="stop")
-        
-        with gr.Column():
+                刷新按钮 = gr.Button("🔄 刷新状态")
+
+        with gr.Column(scale=1):
             状态显示 = gr.Textbox(label="运行状态", lines=1)
-            日志显示 = gr.Textbox(label="实时日志", lines=20)
-            刷新按钮 = gr.Button("🔄 刷新状态")
-    
+            日志显示 = gr.Textbox(label="实时日志（最近120行）", lines=22)
+
     # 事件绑定
-    启动按钮.click(
-        fn=启动模拟,
-        inputs=[合约输入, 周期选择, 参数输入],
-        outputs=[状态显示, 日志显示, gr.Textbox(visible=False)]
-    )
-    
-    停止按钮.click(
-        fn=停止模拟,
-        outputs=状态显示
-    )
-    
-    刷新按钮.click(
-        fn=获取状态,
-        outputs=[状态显示, 日志显示]
-    )
-    
-    # 自动刷新
-    demo.load(获取状态, outputs=[状态显示, 日志显示], every=2)
+    启动按钮.click(fn=启动模拟, inputs=[合约输入, 周期选择, 参数输入], outputs=[状态显示, 日志显示])
+    停止按钮.click(fn=停止模拟, outputs=状态显示)
+    刷新按钮.click(fn=获取状态, outputs=[状态显示, 日志显示])
+
+    # 页面加载时刷新一次
+    demo.load(获取状态, outputs=[状态显示, 日志显示])
+    # Gradio 4：用 Timer 每2秒刷新
+    gr.Timer(2.0, fn=获取状态, outputs=[状态显示, 日志显示])
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)