V2.0 五大核心增强: 锚点定位/原生图表/插件架构/WebSocket/LLM智能

ppt_manager_v2/ai/llm_analyst.py

@@ -0,0 +1,124 @@
+from typing import Dict, Any, List, Optional
+from loguru import logger
+import json
+
+class LLMAnalyst:
+    def __init__(self, config: Dict[str, Any] = None):
+        self.config = config or {}
+        self.provider = self.config.get('provider', 'mock')
+        logger.info(f"LLM分析师初始化，提供者: {self.provider}")
+    
+    def generate_analysis(self, data_context: Dict[str, Any], 
+                           prompt_template: str = None,
+                           max_words: int = 200) -> str:
+        default_prompt = """
+基于以下宏观经济数据，用专业分析师的口吻撰写市场洞察：
+{data_context}
+
+要求：
+1. 不超过 {max_words} 汉字
+2. 专业、客观、具有洞察力
+3. 突出核心指标的边际变化
+4. 适合放在PPT首页作为摘要
+"""
+        
+        prompt = prompt_template or default_prompt
+        data_str = json.dumps(data_context, ensure_ascii=False, indent=2)
+        final_prompt = prompt.format(data_context=data_str, max_words=max_words)
+        
+        if self.provider == 'mock':
+            return self._mock_analysis(data_context)
+        elif self.provider == 'openai':
+            return self._call_openai(final_prompt)
+        elif self.provider == 'tongyi':
+            return self._call_tongyi(final_prompt)
+        else:
+            return self._mock_analysis(data_context)
+    
+    def _mock_analysis(self, context: Dict) -> str:
+        gdp = context.get('gdp_growth', 'N/A')
+        cpi = context.get('cpi', 'N/A')
+        unemployment = context.get('unemployment', 'N/A')
+        
+        return f"""本月宏观经济洞察：GDP增长{gdp}%，经济扩张动能平稳。CPI同比{cpi}%，通胀水平温和，为货币政策留出空间。就业市场{self._format_unemployment(unemployment)}，青年失业率需重点关注。整体来看，经济处于弱复苏通道，建议关注基建投资与消费修复的进度。"""
+    
+    def _format_unemployment(self, val):
+        try:
+            v = float(val)
+            if v > 5.5:
+                return f"压力较大({val}%)"
+            elif v > 5:
+                return f"基本稳定({val}%)"
+            else:
+                return f"表现良好({val}%)"
+        except:
+            return "数据待更新"
+    
+    def _call_openai(self, prompt: str) -> str:
+        try:
+            from openai import OpenAI
+            client = OpenAI(api_key=self.config.get('api_key'))
+            response = client.chat.completions.create(
+                model="gpt-3.5-turbo",
+                messages=[{"role": "user", "content": prompt}]
+            )
+            return response.choices[0].message.content
+        except Exception as e:
+            logger.warning(f"OpenAI调用失败: {e}")
+            return self._mock_analysis({})
+    
+    def _call_tongyi(self, prompt: str) -> str:
+        try:
+            import dashscope
+            dashscope.api_key = self.config.get('api_key')
+            response = dashscope.Generation.call(
+                model='qwen-turbo',
+                prompt=prompt
+            )
+            return response.output.text
+        except Exception as e:
+            logger.warning(f"通义千问调用失败: {e}")
+            return self._mock_analysis({})
+
+class DiffAnalyzer:
+    def __init__(self):
+        self.changes = []
+    
+    def compare_dataframes(self, df_old, df_new, key_cols: List = None):
+        key_cols = key_cols or df_old.columns[:1].tolist()
+        
+        for col in df_old.columns:
+            if col not in key_cols and col in df_new.columns:
+                try:
+                    old_val = df_old[col].iloc[-1] if len(df_old) > 0 else 0
+                    new_val = df_new[col].iloc[-1] if len(df_new) > 0 else 0
+                    
+                    if isinstance(old_val, (int, float)) and isinstance(new_val, (int, float)):
+                        diff = float(new_val) - float(old_val)
+                        if abs(diff) > 0.01:
+                            self.changes.append({
+                                'indicator': col,
+                                'old': round(float(old_val), 2),
+                                'new': round(float(new_val), 2),
+                                'delta': round(diff, 2),
+                                'direction': 'up' if diff > 0 else 'down'
+                            })
+                except:
+                    pass
+        
+        logger.info(f"Diff分析发现 {len(self.changes)} 项显著变动")
+        return self.changes
+    
+    def generate_diff_report(self) -> str:
+        if not self.changes:
+            return "本月与上月数据相比无显著变动。"
+        
+        report_parts = ["📊 数据异动 Diff 简报：\n"]
+        for chg in self.changes[:5]:
+            arrow = "↑" if chg['direction'] == 'up' else "↓"
+            report_parts.append(
+                f"• {chg['indicator']}: {chg['old']} → {chg['new']} "
+                f"({arrow}{abs(chg['delta'])})"
+            )
+        
+        return "\n".join(report_parts)