Update app.py

app.py

@@ -272,6 +272,48 @@ class SpyAgent(BasicAgent):
                 logger.info(f"成功推理出卧底词: '{spy_word}'")
                 return spy_word
         
+        return None
+    def analyze_similar_words(self, general_concept):
+        """当推断出的平民词和卧底词相同时，进行深度分析尝试细分概念"""
+        speak_history = self.memory.load_variable('speak_history')
+        if not speak_history:
+            return None
+            
+        my_word = self.memory.load_variable("word")
+        
+        prompt = []
+        prompt.append(f'在《谁是卧底》游戏中，初步分析显示玩家们可能在描述"{general_concept}"这个概念。')
+        prompt.append(f'你的词是"{my_word}"。游戏规则要求5名玩家拿到相同的平民词，1名玩家拿到不同但相关的卧底词。')
+        prompt.append('由于平民词和卧底词通常是近义词或同类事物，我需要你进行更精细的分析，找出可能的具体词语。')
+        prompt.append('请分析以下玩家发言，尝试识别出更具体的词语类别：\n')
+        
+        # 添加玩家发言供分析
+        for name, speaks in speak_history.items():
+            if name != self.memory.load_variable('name'):
+                for i, speak in enumerate(speaks):
+                    if speak.strip():
+                        prompt.append(f"[轮次{i+1}] {name}: {speak}\n")
+                        
+        prompt.append(f'\n基于"{general_concept}"这个概念，请分析出两个可能的更具体词语：')
+        prompt.append('1. 多数人可能持有的具体词语（平民词）')
+        prompt.append('2. 少数人可能持有的相关但不同的词语（卧底词）')
+        prompt.append('\n请用JSON格式回答：{"possible_civilian": "具体平民词", "possible_spy": "具体卧底词"}')
+        
+        result = self.llm_caller(''.join(prompt))
+        logger.info(f"词语细化分析原始返回: '{result}'")
+        
+        try:
+            # 尝试提取JSON内容
+            import json
+            import re
+            
+            json_match = re.search(r'{.*}', result, re.DOTALL)
+            if json_match:
+                json_str = json_match.group(0)
+                return json.loads(json_str)
+        except Exception as e:
+            logger.error(f"词语细化分析JSON解析失败: {e}")
+            
         return None
     def detect_civilian_word(self):
         """分析历史发言，尝试识别平民词"""
@@ -361,6 +403,26 @@ class SpyAgent(BasicAgent):
             # 在发言前更新对手词
             civilian_word = self.detect_civilian_word()
             spy_word = self.detect_spy_word()
+            # 添加这段处理逻辑
+            if civilian_word and spy_word and civilian_word == spy_word:
+                # 平民词和卧底词推断结果相同的特殊处理
+                logger.info(f"检测到平民词'{civilian_word}'与卧底词'{spy_word}'推断结果相同，执行深度分析")
+                
+                refined_analysis = self.analyze_similar_words(civilian_word)
+                if refined_analysis:
+                    # 尝试获取更精确的词语区分
+                    possible_civilian = refined_analysis.get("possible_civilian")
+                    possible_spy = refined_analysis.get("possible_spy")
+                    
+                    if possible_civilian and possible_civilian != civilian_word:
+                        logger.info(f"细化分析后的平民词: '{possible_civilian}'")
+                        self.memory.set_variable('sim_word', possible_civilian)
+                        civilian_word = possible_civilian
+                        
+                    if possible_spy and possible_spy != spy_word:
+                        logger.info(f"细化分析后的卧底词: '{possible_spy}'")
+                        self.memory.set_variable('spy_word', possible_spy)
+                        spy_word = possible_spy
             if civilian_word:
                 self.memory.set_variable('sim_word', civilian_word)