Update app.py

app.py

@@ -55,6 +55,7 @@ class SpyAgent(BasicAgent):
         prompt.append('\n请用以下格式回答：')
         prompt.append('平民词：[词语或"未知"/"信息不足"]')
         prompt.append('卧底词：[词语或"未知"/"信息不足"]')
+        prompt.append('可疑玩家：[玩家名字或"未知"]')  # 新增这行
         prompt.append('分析理由：[简要分析发言中的线索和推理过程]')
         
         analysis_result = self.llm_caller(''.join(prompt))
@@ -87,7 +88,16 @@ class SpyAgent(BasicAgent):
             reasoning = reason_match.group(1).strip()
             # 保存分析理由供下一轮使用
             self.memory.set_variable('word_analysis_reasoning', reasoning)
-        
+         # 提取可疑玩家
+        suspect_match = re.search(r'可疑玩家[：:]\s*(.+?)(?:\n|$)', analysis_result)
+        if suspect_match:
+            suspect = suspect_match.group(1).strip()
+            if not any(pattern in suspect.lower() for pattern in ["未知", "不清楚", "无法确定", "不确定", "信息不足"]):
+                for player in speak_history.keys():
+                    if player in suspect and player != self.memory.load_variable('name'):
+                        self.memory.set_variable('spy_player', player)
+                        logger.info(f"集成分析识别可疑卧底: '{player}'")
+                        break
         return civilian_word, spy_word
     def process_speak(self, name, speak):
         prompt = []