Update app.py

app.py

@@ -69,12 +69,13 @@ except Exception as e:
 
 
 # ---------------------- 2. 定义交易决策函数 ----------------------
+
 def generate_trading_decision(ticker, analysis_date, progress=gr.Progress()):
-    """核心函数：生成交易决策，使用框架规范参数名"""
+    """核心函数：生成交易决策，显示中间分析过程"""
     logger.info(f"收到分析请求 - 股票代码: {ticker}, 分析日期: {analysis_date}")
     
     try:
-        # 1. 验证输入
+        # 1. 验证输入（与之前逻辑一致）
         if not ticker.strip():
             error_msg = "错误：股票代码不能为空！请输入如 AAPL、SPY 的代码。"
             logger.warning(error_msg)
@@ -85,7 +86,6 @@ def generate_trading_decision(ticker, analysis_date, progress=gr.Progress()):
             logger.warning(error_msg)
             return error_msg
         
-        # 验证日期格式
         try:
             time.strptime(analysis_date.strip(), "%Y-%m-%d")
         except ValueError:
@@ -97,35 +97,64 @@ def generate_trading_decision(ticker, analysis_date, progress=gr.Progress()):
         
         # 2. 显示进度
         progress(0, desc="开始初始化分析...")
-        logger.info("分析流程启动")
         time.sleep(1)
         
         progress(0.2, desc=f"获取 {ticker} 的金融数据（FinnHub）...")
-        logger.info(f"正在获取 {ticker} 的金融数据")
         time.sleep(2)
         
         progress(0.5, desc="分析师团队分析（财务+情绪+技术指标）...")
-        logger.info("分析师团队开始分析（财务+情绪+技术指标）")
         time.sleep(3)
         
         progress(0.8, desc="研究团队辩论+交易员决策+风险管理评估...")
-        logger.info("进入研究团队辩论和风险管理评估阶段")
         
-        # 3. 调用框架核心方法（使用项目规范的参数名 company_name 和 trade_date）
+        # 3. 调用框架核心方法，捕获中间结果
         logger.info(f"调用 propagate 方法 - company_name={ticker.strip()}, trade_date={analysis_date.strip()}")
         start_time = time.time()
-        _, decision = ta.propagate(
-            company_name=ticker.strip(),
-            trade_date=analysis_date.strip()
-        )
+        
+        # ---- 关键修改：调试模式下捕获流式中间结果 ----
+        all_outputs = []  # 存储所有中间输出
+        final_state = None
+        processed_signal = None
+
+        # 若处于调试模式（debug=True），使用 stream 获取中间结果
+        if ta.debug:
+            # 获取 graph.invoke 所需的参数
+            args = ta.propagator.get_graph_args()
+            init_state = ta.propagator.create_initial_state(ticker.strip(), analysis_date.strip())
+            
+            # 遍历流式结果
+            for i, chunk in enumerate(ta.graph.stream(init_state, **args)):
+                # 提取并格式化中间消息
+                if chunk.get("messages") and len(chunk["messages"]) > 0:
+                    msg = chunk["messages"][-1].content  # 假设消息内容在 .content 中
+                    all_outputs.append(f"## 中间步骤 {i+1}\n{msg}\n\n")
+                    logger.info(f"中间结果 {i+1}: {msg[:50]}...")  # 日志简略显示
+                
+                # 模拟进度（根据实际步骤数调整，这里简化）
+                progress(0.8 + 0.2 * i / 10, desc=f"处理中间步骤 {i+1}/10")
+        
+            # 流式结束后，最终状态为最后一个 chunk
+            final_state = chunk
+        else:
+            # 非调试模式，直接调用（无中间结果）
+            final_state, processed_signal = ta.propagate(
+                company_name=ticker.strip(),
+                trade_date=analysis_date.strip()
+            )
+            all_outputs.append("（非调试模式，无中间步骤显示）\n\n")
+
+        # 处理最终结果
+        if final_state:
+            decision = final_state.get("final_trade_decision", "未获取到最终交易决策")
+            all_outputs.append(f"# 最终交易决策\n{decision}\n\n")
+            
+            # 添加免责声明
+            disclaimer = "\n\n【免责声明】本结果仅用于研究目的，不构成任何财务、投资或交易建议。交易风险自负。"
+            final_result = f"# {ticker} 交易决策报告（分析日期：{analysis_date}）\n\n" + "".join(all_outputs) + disclaimer
+        
         end_time = time.time()
         logger.info(f"分析完成，耗时 {end_time - start_time:.2f} 秒")
         
-        # 4. 添加免责声明
-        disclaimer = "\n\n【免责声明】本结果仅用于研究目的，不构成任何财务、投资或交易建议。交易风险自负。"
-        final_result = f"# {ticker} 交易决策报告（分析日期：{analysis_date}）\n\n{decision}{disclaimer}"
-        logger.info(f"生成 {ticker} 的交易决策报告完成")
-        
         progress(1.0, desc="分析完成！")
         return final_result
     
@@ -141,6 +170,7 @@ def generate_trading_decision(ticker, analysis_date, progress=gr.Progress()):
         return error_msg
 
 
+
 # ---------------------- 3. 构建 Gradio 界面 ----------------------
 logger.info("开始构建 Gradio 界面...")
 with gr.Blocks(title="TradingAgents 金融交易决策工具") as demo: