EmailBill/Service/AI/SmartHandleService.cs

using Service.Transaction;

namespace Service.AI;

public interface ISmartHandleService
{
    Task SmartClassifyAsync(long[] transactionIds, Action<(string type, string data)> chunkAction);

    Task AnalyzeBillAsync(string userInput, Action<string> chunkAction);

    Task<TransactionParseResult?> ParseOneLineBillAsync(string text);

    /// <summary>
    /// 从邮件正文中使用AI提取交易记录（AI兜底方案）
    /// </summary>
    Task<(string card, string reason, decimal amount, decimal balance, TransactionType type, DateTime? occurredAt)[]?> ParseEmailByAiAsync(string emailBody);

    /// <summary>
    /// 为分类生成多个SVG图标（定时任务使用）
    /// </summary>
    Task<List<string>?> GenerateCategoryIconsAsync(string categoryName, TransactionType categoryType, int iconCount = 5);

    /// <summary>
    /// 为分类生成单个SVG图标（手动触发使用）
    /// </summary>
    Task<string?> GenerateSingleCategoryIconAsync(string categoryName, TransactionType categoryType);

    /// <summary>
    /// 生成预算执行报告（HTML格式）
    /// </summary>
    /// <param name="promptWithData">完整的Prompt（包含数据和格式要求）</param>
    /// <param name="year">年份</param>
    /// <param name="month">月份</param>
    Task<string?> GenerateBudgetReportAsync(string promptWithData, int year, int month);
}

public class SmartHandleService(
    ITransactionRecordRepository transactionRepository,
    ITransactionStatisticsService transactionStatisticsService,
    ITextSegmentService textSegmentService,
    ILogger<SmartHandleService> logger,
    ITransactionCategoryRepository categoryRepository,
    IOpenAiService openAiService,
    IConfigService configService,
    IClassificationIconPromptProvider iconPromptProvider
) : ISmartHandleService
{
    public async Task SmartClassifyAsync(long[] transactionIds, Action<(string, string)> chunkAction)
    {
        try
        {
            // 获取指定ID的账单（作为样本）
            var sampleRecords = await transactionRepository.GetByIdsAsync(transactionIds);

            sampleRecords = sampleRecords
                .Where(x => string.IsNullOrEmpty(x.Classify))
                .ToArray();

            if (sampleRecords.Length == 0)
            {
                // await WriteEventAsync("error", "找不到指定的账单");
                chunkAction(("error", "找不到指定的账单"));
                return;
            }

            // 重新按Reason分组所有待分类账单
            var groupedRecords = sampleRecords
                .GroupBy(r => r.Reason)
                .Select(g => new
                {
                    Reason = g.Key,
                    Ids = g.Select(r => r.Id).ToList(),
                    Count = g.Count(),
                    TotalAmount = g.Sum(r => r.Amount),
                    SampleType = g.First().Type
                })
                .OrderByDescending(g => Math.Abs(g.TotalAmount))
                .ToList();

            // 【增强功能】对每个分组的摘要进行分词，查询已分类的相似账单
            var referenceRecords = new Dictionary<string, List<TransactionRecord>>();
            foreach (var group in groupedRecords)
            {
                // 使用专业分词库提取关键词
                var keywords = textSegmentService.ExtractKeywords(group.Reason);

                if (keywords.Count > 0)
                {
                    // 查询包含这些关键词且已分类的账单（带相关度评分）
                    // minMatchRate=0.4 表示至少匹配40%的关键词才被认为是相似的
                    var similarClassifiedWithScore = await transactionStatisticsService.GetClassifiedByKeywordsWithScoreAsync(keywords, minMatchRate: 0.4, limit: 10);

                    if (similarClassifiedWithScore.Count > 0)
                    {
                        // 只取前5个最相关的
                        var topSimilar = similarClassifiedWithScore.Take(5).Select(x => x.record).ToList();
                        referenceRecords[group.Reason] = topSimilar;

                        // 记录调试信息
                        logger.LogDebug("摘要 '{Reason}' 提取关键词: {Keywords}, 找到 {Count} 个相似账单，相关度分数: {Scores}",
                            group.Reason,
                            string.Join(", ", keywords),
                            similarClassifiedWithScore.Count,
                            string.Join(", ", similarClassifiedWithScore.Select(x => $"{x.record.Reason}({x.relevanceScore:F2})")));
                    }
                    else
                    {
                        logger.LogDebug("摘要 '{Reason}' 提取关键词: {Keywords}, 未找到高相关度的相似账单",
                            group.Reason,
                            string.Join(", ", keywords));
                    }
                }
            }

            // 构建分类信息
            var categoryInfo = await GetCategoryInfoAsync();

            // 构建账单分组信息
            var billsInfo = new StringBuilder();
            foreach (var (group, index) in groupedRecords.Select((g, i) => (g, i)))
            {
                billsInfo.AppendLine($"{index + 1}. 摘要={group.Reason}, 当前类型={GetTypeName(group.SampleType)}, 当前分类={(string.IsNullOrEmpty(group.SampleType.ToString()) ? "未分类" : group.SampleType.ToString())}, 涉及金额={group.TotalAmount}");

                // 如果有相似的已分类账单，添加参考信息
                if (referenceRecords.TryGetValue(group.Reason, out var references))
                {
                    billsInfo.AppendLine("   【参考】相似且已分类的账单：");
                    foreach (var refer in references.Take(3)) // 最多显示3个参考
                    {
                        billsInfo.AppendLine($"      - 摘要={refer.Reason}, 分类={refer.Classify}, 类型={GetTypeName(refer.Type)}, 金额={refer.Amount}");
                    }
                }
            }

            var systemPrompt = $$"""
            你是一个专业的账单分类助手。请根据提供的账单分组信息和分类列表，为每个分组选择最合适的分类。

            可用的分类列表：
            {{categoryInfo}}

            分类规则：
            1. 根据账单的摘要和涉及金额，选择最匹配的分类
            2. 如果提供了【参考】信息，优先参考相似账单的分类，这些是历史上已分类的相似账单
            3. 如果无法确定分类，可以选择"其他"
            4. 每个分组可能包含多条账单，你需要为整个分组选择一个分类

            输出格式要求（强制）：
            - 请使用 NDJSON（每行一个独立的 JSON 对象，末尾以换行符分隔），不要输出数组。
            - 每行的JSON格式严格为：
            {
                "reason": "交易摘要", 
                "type": Number, // 交易类型，0=支出，1=收入，2=不计入收支
                "classify": "分类名称"
            }
            - 不要输出任何解释性文字、编号、标点或多余的文本
            - 如果无法判断分类，请不要输出改行的JSON对象

            只输出按行的JSON对象（NDJSON），不要有其他文字说明。
            """;

            var userPrompt = $$"""
            请为以下账单分组进行分类：

            {{billsInfo}}

            请逐个输出分类结果。
            """;

            // 流式调用AI
            chunkAction(("start", $"开始分类，共 {sampleRecords.Length} 条账单"));

            var classifyResults = new List<(string Reason, string Classify, TransactionType Type)>();
            var sentIds = new HashSet<long>();

            // 将流解析逻辑提取为本地函数以减少嵌套
            void HandleResult(GroupClassifyResult? result)
            {
                if (result is null || string.IsNullOrEmpty(result.Reason)) return;
                classifyResults.Add((result.Reason, result.Classify ?? string.Empty, result.Type));
                var group = groupedRecords.FirstOrDefault(g => g.Reason == result.Reason);
                if (group == null) return;
                foreach (var id in group.Ids)
                {
                    if (!sentIds.Add(id))
                    {
                        continue;
                    }

                    var resultJson = JsonSerializer.Serialize(new
                    {
                        id,
                        result.Classify,
                        result.Type
                    });
                    chunkAction(("data", resultJson));
                }
            }

            // 解析缓冲区中的所有完整 JSON 对象或数组
            void FlushBuffer(StringBuilder buffer)
            {
                var buf = buffer.ToString();
                if (string.IsNullOrWhiteSpace(buf)) return;

                // 优先尝试解析完整数组
                var trimmed = buf.TrimStart();
                if (trimmed.Length > 0 && trimmed[0] == '[')
                {
                    var lastArrEnd = buf.LastIndexOf(']');
                    if (lastArrEnd > -1)
                    {
                        var arrJson = buf.Substring(0, lastArrEnd + 1);
                        try
                        {
                            var results = JsonSerializer.Deserialize<GroupClassifyResult[]>(arrJson);
                            if (results != null)
                            {
                                foreach (var r in results) HandleResult(r);
                            }
                            buffer.Remove(0, lastArrEnd + 1);
                            buf = buffer.ToString();
                        }
                        catch (Exception exArr)
                        {
                            logger.LogDebug(exArr, "按数组解析AI返回失败，回退到逐对象解析。预览: {Preview}", arrJson.Length > 200 ? arrJson.Substring(0, 200) + "..." : arrJson);
                        }
                    }
                }

                // 逐对象解析
                var startIdx = 0;
                while (startIdx < buf.Length)
                {
                    var openBrace = buf.IndexOf('{', startIdx);
                    if (openBrace == -1) break;
                    var closeBrace = FindMatchingBrace(buf, openBrace);
                    if (closeBrace == -1) break;
                    var jsonStr = buf.Substring(openBrace, closeBrace - openBrace + 1);
                    try
                    {
                        var result = JsonSerializer.Deserialize<GroupClassifyResult>(jsonStr);
                        HandleResult(result);
                    }
                    catch (Exception ex)
                    {
                        logger.LogWarning(ex, "解析AI分类结果失败: {JsonStr}", jsonStr.Length > 200 ? jsonStr.Substring(0, 200) + "..." : jsonStr);
                    }
                    startIdx = closeBrace + 1;
                }

                if (startIdx > 0)
                {
                    buffer.Remove(0, startIdx);
                }
            }

            var buffer = new StringBuilder();
            await foreach (var chunk in openAiService.ChatStreamAsync(systemPrompt, userPrompt))
            {
                buffer.Append(chunk);
                FlushBuffer(buffer);
            }

            // 如果AI流结束但没有任何分类结果，发出错误提示
            if (classifyResults.Count == 0)
            {
                logger.LogWarning("AI未返回任何分类结果，buffer最终内容: {BufferPreview}", buffer.ToString().Length > 500 ? buffer.ToString().Substring(0, 500) + "..." : buffer.ToString());
                chunkAction(("error", "智能分类未返回任何结果，请重试或手动分类"));
            }
            else
            {
                chunkAction(("end", "分类完成"));
            }
        }
        catch (Exception ex)
        {
            logger.LogError(ex, "智能分类失败");
            chunkAction(("error", $"智能分类失败: {ex.Message}"));
        }
    }

    public async Task AnalyzeBillAsync(string userInput, Action<string> chunkAction)
    {
        try
        {
            // 构建分类信息
            var categoryInfo = await GetCategoryInfoAsync();

            // 第一步：使用AI生成聚合SQL查询
            var now = DateTime.Now;
            var sqlPrompt = $$"""
            当前日期：{{now:yyyy年M月d日}}（{{now:yyyy-MM-dd}}）
            用户问题：{{userInput}}
            
            数据库类型：SQLite
            数据库表名：TransactionRecord
            字段说明：
            - Id: bigint 主键
            - Card: nvarchar 卡号
            - Reason: nvarchar 交易原因/摘要
            - Amount: decimal 交易金额（支出为负数，收入为正数）
            - OccurredAt: datetime 交易发生时间（TEXT类型，格式：'2025-12-26 10:30:00'）
            - Type: int 交易类型（0=支出, 1=收入, 2=不计入收支）
            - Classify: nvarchar 交易分类（如：交通、餐饮、购物等）
            
            【核心原则】直接生成用户所需的聚合统计SQL，而不是查询原始记录后再统计
            
            要求：
            1. 根据用户问题判断需要什么维度的聚合数据
            2. 使用 GROUP BY 按分类、时间等维度分组
            3. 使用聚合函数：SUM(ABS(Amount)) 计算金额总和、COUNT(*) 计数、AVG()平均、MAX()最大、MIN()最小
            4. 时间范围使用 OccurredAt 字段，"最近X个月/天"基于当前日期计算
            5. 支出用 Type = 0，收入用 Type = 1
            6. 给聚合字段起有意义的别名（如 TotalAmount, TransactionCount, AvgAmount）
            7. 使用 ORDER BY 对结果排序（通常按金额降序）
            8. 只返回SQL语句，不要解释
            
            【重要】SQLite日期函数：
            - 提取年份：strftime('%Y', OccurredAt)
            - 提取月份：strftime('%m', OccurredAt)
            - 提取日期：strftime('%Y-%m-%d', OccurredAt)
            - 不要使用 YEAR()、MONTH()、DAY() 函数，SQLite不支持
            
            【重要】最终的SQL会被一下DOTNET代码执行, 请确保你生成的代码可执行，不报错
            ```C#
            public async Task<List<dynamic>> ExecuteDynamicSqlAsync(string completeSql)
            {
                var dt = await FreeSql.Ado.ExecuteDataTableAsync(completeSql);
                var result = new List<dynamic>();

                foreach (System.Data.DataRow row in dt.Rows)
                {
                    var expando = new System.Dynamic.ExpandoObject() as IDictionary<string, object>;
                    foreach (System.Data.DataColumn column in dt.Columns)
                    {
                        expando[column.ColumnName] = row[column];
                    }
                    result.Add(expando);
                }

                return result;
            }
            ```

            【重要】必须从以下分类列表中选择分类：
            {{categoryInfo}}

            只返回SQL语句。
            """;

            var sqlText = await openAiService.ChatAsync(sqlPrompt);

            // 清理SQL文本
            sqlText = sqlText?.Trim() ?? "";
            sqlText = sqlText.TrimStart('`').TrimEnd('`');
            if (sqlText.StartsWith("sql", StringComparison.OrdinalIgnoreCase))
            {
                sqlText = sqlText.Substring(3).Trim();
            }

            logger.LogInformation("AI生成的SQL: {Sql}", sqlText);

            chunkAction(
                JsonSerializer.Serialize(new
                {
                    content = $"""
                    <pre style="max-height: 80px; font-size: 8px; overflow-y: auto; padding: 8px; border: 1px solid #3c3c3c">
                    {WebUtility.HtmlEncode(sqlText)}
                    </pre>
                    """
                })
            );

            // 第二步：执行动态SQL查询
            List<dynamic> queryResults;
            try
            {
                queryResults = await transactionRepository.ExecuteDynamicSqlAsync(sqlText);
            }
            catch (Exception ex)
            {
                logger.LogError(ex, "执行AI生成的SQL失败: {Sql}", sqlText);
                // 如果SQL执行失败，返回错误
                var errorData = JsonSerializer.Serialize(new { content = "<div class='error-message'>SQL执行失败，请重新描述您的问题</div>" });
                chunkAction(errorData);
                return;
            }

            // 第三步：将查询结果序列化为JSON，直接传递给AI生成分析报告
            var dataJson = JsonSerializer.Serialize(queryResults, new JsonSerializerOptions
            {
                WriteIndented = true,
                Encoder = JavaScriptEncoder.UnsafeRelaxedJsonEscaping
            });

            var userPromptExtra = await configService.GetConfigByKeyAsync<string>("BillAnalysisPrompt");

            var dataPrompt = $"""
            当前日期：{DateTime.Now:yyyy年M月d日}
            用户问题：{userInput}
            
            【用户要求（重要）】
            {userInput}

            查询结果数据（JSON格式）：
            {dataJson}
            
            说明：以上数据是根据用户问题查询出的聚合统计结果，请基于这些数据生成分析报告。
            
            请生成一份专业的数据分析报告，严格遵守以下要求：
            
            【格式要求】
            1. 使用HTML格式（移动端H5页面风格）
            2. 生成清晰的报告标题（基于用户问题）
            3. 使用表格展示统计数据（table > thead/tbody > tr > th/td）
            4. 使用合适的HTML标签：h2（标题）、h3（小节）、p（段落）、table（表格）、ul/li（列表）、strong（强调）
            5. 支出金额用 <span class='expense-value'>金额</span> 包裹
            6. 收入金额用 <span class='income-value'>金额</span> 包裹
            7. 重要结论用 <span class='highlight'>内容</span> 高亮
            
            【样式限制（重要）】
            8. 不要包含 html、body、head 标签
            9. 不要使用任何 style 属性或 <style> 标签
            10. 不要设置 background、background-color、color 等样式属性
            11. 不要使用 div 包裹大段内容
            
            【内容要求】
            12. 准确解读数据：将JSON数据转换为易读的表格和文字说明
            13. 提供洞察分析：根据数据给出有价值的发现和趋势分析
            14. 给出实用建议：基于数据提供合理的财务建议
            15. 语言专业、清晰、简洁

            【用户补充（重要）】
            {userPromptExtra}
            
            直接输出纯净的HTML内容，不要markdown代码块标记。
            """;

            // 第四步：流式输出AI分析结果
            await foreach (var chunk in openAiService.ChatStreamAsync(dataPrompt))
            {
                var sseData = JsonSerializer.Serialize(new { content = chunk });
                chunkAction(sseData);
            }

            // 发送完成标记
            chunkAction("[DONE]");
        }
        catch (Exception ex)
        {
            logger.LogError(ex, "智能分析账单失败");
            var errorData = JsonSerializer.Serialize(new { content = $"<div class='error-message'>分析失败：{ex.Message}</div>" });
            chunkAction(errorData);
        }
    }

    public async Task<TransactionParseResult?> ParseOneLineBillAsync(string text)
    {
        // 获取所有分类
        var categories = await categoryRepository.GetAllAsync();

        // 构建分类信息
        var categoryInfo = new StringBuilder();
        foreach (var type in new[] { 0, 1, 2 })
        {
            var typeName = GetTypeName((TransactionType)type);
            categoryInfo.AppendLine($"{typeName}: ");
            var categoriesOfType = categories.Where(c => (int)c.Type == type).ToList();
            foreach (var category in categoriesOfType)
            {
                categoryInfo.AppendLine($"- {category.Name}");
            }
        }

        var sysPrompt = $$"""
        你是一个智能账单解析助手。请从用户提供的文本中提取交易信息，包括日期、金额、摘要、类型和分类。

        可用的分类列表：
        {{categoryInfo}}

        请返回 JSON 格式，包含以下字段：
        - OccurredAt: 日期时间，格式 yyyy-MM-dd HH:mm:ss。当前系统时间为{{DateTime.Now:yyyy-MM-dd HH:mm:ss}}。
        - Amount: 金额，数字。
        - Reason: 备注/摘要，原文或其他补充信息。
        - Type: 交易类型，0=支出，1=收入，2=不计入收支。根据语义判断。
        - Classify: 分类，请从以下现有分类中选择最匹配的一个：如果无法匹配，请留空。

        返回示例
        {
            "OccurredAt": "2024-06-15 14:30:00",
            "Amount": 150.75,
            "Reason": "午餐消费",
            "Type": 0,
            "Classify": "餐饮"
        }

        只返回 JSON，不要包含 markdown 标记。
        """;
        var json = await openAiService.ChatAsync(sysPrompt, text);
        if (string.IsNullOrWhiteSpace(json)) return null;

        try
        {
            // 清理可能的 markdown 标记
            json = json.Replace("```json", "").Replace("```", "").Trim();
            var options = new JsonSerializerOptions { PropertyNameCaseInsensitive = true };
            return JsonSerializer.Deserialize<TransactionParseResult>(json, options);
        }
        catch (Exception ex)
        {
            logger.LogError(ex, "解析账单失败");
            return null;
        }
    }

    /// <summary>
    /// 查找匹配的右括号
    /// </summary>
    private static int FindMatchingBrace(string str, int startPos)
    {
        var braceCount = 0;
        for (var i = startPos; i < str.Length; i++)
        {
            if (str[i] == '{') braceCount++;
            else if (str[i] == '}')
            {
                braceCount--;
                if (braceCount == 0) return i;
            }
        }
        return -1;
    }

    private static string GetTypeName(TransactionType type)
    {
        return type switch
        {
            TransactionType.Expense => "支出",
            TransactionType.Income => "收入",
            TransactionType.None => "不计入收支",
            _ => "未知"
        };
    }

    /// <summary>
    /// 清理 AI 响应中的 markdown 代码块标记
    /// </summary>
    private static string CleanMarkdownCodeBlock(string response)
    {
        var cleaned = response?.Trim() ?? string.Empty;
        if (cleaned.StartsWith("```"))
        {
            // 移除开头的 ```json 或 ```
            var firstNewLine = cleaned.IndexOf('\n');
            if (firstNewLine > 0)
            {
                cleaned = cleaned.Substring(firstNewLine + 1);
            }

            // 移除结尾的 ```
            if (cleaned.EndsWith("```"))
            {
                cleaned = cleaned.Substring(0, cleaned.Length - 3);
            }

            cleaned = cleaned.Trim();
        }
        return cleaned;
    }

    private async Task<string> GetCategoryInfoAsync()
    {
        // 获取所有分类
        var categories = await categoryRepository.GetAllAsync();

        // 构建分类信息
        var categoryInfo = new StringBuilder();
        foreach (var type in new[] { 0, 1, 2 })
        {
            var typeName = GetTypeName((TransactionType)type);
            categoryInfo.AppendLine($"{typeName}: ");
            var categoriesOfType = categories.Where(c => (int)c.Type == type).ToList();
            foreach (var category in categoriesOfType)
            {
                categoryInfo.AppendLine($"- {category.Name}");
            }
        }

        return categoryInfo.ToString();
    }

    /// <summary>
    /// 从邮件正文中使用AI提取交易记录（AI兜底方案）
    /// </summary>
    public async Task<(string card, string reason, decimal amount, decimal balance, TransactionType type, DateTime? occurredAt)[]?> ParseEmailByAiAsync(string emailBody)
    {
        var systemPrompt = $"""
        你是一个信息抽取助手。
        仅输出严格的JSON数组，不要包含任何多余文本。
        每个交易记录包含字段: card(字符串), reason(字符串), amount(数字), balance(数字), type(字符串，值为'收入'或'支出'), occurredAt(字符串，yyyy-MM-dd HH:mm:ss格式日期时间)。
        如果缺失，请推断或置空。
        [重要] 当前时间为{DateTime.Now:yyyy-MM-dd HH:mm:ss}，请根据当前时间推断交易发生的时间。
        """;
        var userPrompt = $"""
        从下面这封银行账单相关邮件正文中提取所有交易记录，返回JSON数组格式，
        每个元素包含: card, reason, amount, balance, type(收入或支出), occurredAt(非必要)。
        正文如下:\n\n{emailBody}
        """;

        var contentText = await openAiService.ChatAsync(systemPrompt, userPrompt);
        if (string.IsNullOrWhiteSpace(contentText))
        {
            logger.LogWarning("AI未返回任何内容，无法解析邮件");
            return null;
        }

        logger.LogDebug("AI返回的内容: {Content}", contentText);
        // 清理可能的 markdown 代码块标记
        contentText = contentText.Trim();
        if (contentText.StartsWith("```"))
        {
            // 移除开头的 ```json 或 ```
            var firstNewLine = contentText.IndexOf('\n');
            if (firstNewLine > 0)
            {
                contentText = contentText.Substring(firstNewLine + 1);
            }

            // 移除结尾的 ```
            if (contentText.EndsWith("```"))
            {
                contentText = contentText.Substring(0, contentText.Length - 3);
            }

            contentText = contentText.Trim();
        }

        // contentText 期望是 JSON 数组
        using var jsonDoc = JsonDocument.Parse(contentText);
        var arrayElement = jsonDoc.RootElement;

        // 如果返回的是单个对象而不是数组，尝试兼容处理
        if (arrayElement.ValueKind == JsonValueKind.Object)
        {
            logger.LogWarning("AI返回的内容是单个对象而非数组，尝试兼容处理");
            var result = ParseEmailSingleRecord(arrayElement);
            return result != null ? [result.Value] : null;
        }

        if (arrayElement.ValueKind != JsonValueKind.Array)
        {
            logger.LogWarning("AI返回的内容不是JSON数组，无法解析邮件");
            return null;
        }

        var results = new List<(string card, string reason, decimal amount, decimal balance, TransactionType type, DateTime? occurredAt)>();

        foreach (var obj in arrayElement.EnumerateArray())
        {
            var record = ParseEmailSingleRecord(obj);
            if (record != null)
            {
                logger.LogInformation("解析到一条交易记录: {@Record}", record.Value);
                results.Add(record.Value);
            }
        }

        logger.LogInformation("使用AI成功解析邮件内容，提取到 {Count} 条交易记录", results.Count);
        return results.Count > 0 ? results.ToArray() : null;
    }

    /// <summary>
    /// 为分类生成多个SVG图标（定时任务使用）
    /// </summary>
    public async Task<List<string>?> GenerateCategoryIconsAsync(string categoryName, TransactionType categoryType, int iconCount = 5)
    {
        logger.LogInformation("正在为分类 {CategoryName} 生成 {IconCount} 个图标", categoryName, iconCount);

        var systemPrompt = iconPromptProvider.GetPrompt(categoryName, categoryType);

        var response = await openAiService.ChatAsync(systemPrompt, "", timeoutSeconds: 60 * 10);

        if (string.IsNullOrWhiteSpace(response))
        {
            logger.LogWarning("AI 未返回有效的图标数据，分类: {CategoryName}", categoryName);
            return null;
        }

        // 清理可能的 markdown 代码块标记
        response = CleanMarkdownCodeBlock(response);

        if (string.IsNullOrWhiteSpace(response))
        {
            logger.LogWarning("清理后的 AI 响应为空，分类: {CategoryName}", categoryName);
            return null;
        }

        // 验证返回的是有效的 JSON 数组
        try
        {
            var icons = JsonSerializer.Deserialize<List<string>>(response);
            if (icons == null || icons.Count != iconCount)
            {
                logger.LogWarning("AI 返回的图标数量不正确（期望{IconCount}个），分类: {CategoryName}", iconCount, categoryName);
                return null;
            }

            logger.LogInformation("成功为分类 {CategoryName} 生成了 {IconCount} 个图标", categoryName, iconCount);
            return icons;
        }
        catch (JsonException ex)
        {
            logger.LogError(ex, "解析 AI 返回的图标数据失败，分类: {CategoryName}，响应内容: {Response}",
                categoryName, response);
            return null;
        }
    }

    /// <summary>
    /// 为分类生成单个SVG图标（手动触发使用）
    /// </summary>
    public async Task<string?> GenerateSingleCategoryIconAsync(string categoryName, TransactionType categoryType)
    {
        logger.LogInformation("正在为分类 {CategoryName} 生成单个图标", categoryName);

        // 使用单个图标生成的 Prompt（只生成 1 个图标，加快速度）
        var systemPrompt = iconPromptProvider.GetSingleIconPrompt(categoryName, categoryType);

        var stopwatch = System.Diagnostics.Stopwatch.StartNew();
        try
        {
            // 增加超时时间到 180 秒（3 分钟）
            var response = await openAiService.ChatAsync(systemPrompt, "", timeoutSeconds: 180);

            stopwatch.Stop();
            logger.LogInformation("AI 响应耗时: {ElapsedMs}ms，分类: {CategoryName}", stopwatch.ElapsedMilliseconds, categoryName);

            if (string.IsNullOrWhiteSpace(response))
            {
                logger.LogWarning("AI 未返回有效的图标数据，分类: {CategoryName}", categoryName);
                return null;
            }

            // 清理可能的 markdown 代码块标记
            response = CleanMarkdownCodeBlock(response);

            if (string.IsNullOrWhiteSpace(response))
            {
                logger.LogWarning("清理后的 AI 响应为空，分类: {CategoryName}", categoryName);
                return null;
            }

            // 解析返回的 JSON 数组，取第一个图标
            try
            {
                var icons = JsonSerializer.Deserialize<List<string>>(response);
                if (icons == null || icons.Count == 0)
                {
                    logger.LogWarning("AI 未返回有效的图标数据，分类: {CategoryName}", categoryName);
                    return null;
                }

                var svg = icons[0];
                logger.LogInformation("成功为分类 {CategoryName} 生成单个图标，总耗时: {ElapsedMs}ms", categoryName, stopwatch.ElapsedMilliseconds);
                return svg;
            }
            catch (JsonException ex)
            {
                logger.LogError(ex, "解析 AI 返回的图标数据失败，分类: {CategoryName}，响应内容: {Response}",
                    categoryName, response.Length > 500 ? response.Substring(0, 500) + "..." : response);
                return null;
            }
        }
        catch (TimeoutException)
        {
            stopwatch.Stop();
            logger.LogError("AI 请求超时（>180秒），分类: {CategoryName}，已等待: {ElapsedMs}ms", categoryName, stopwatch.ElapsedMilliseconds);
            throw;
        }
        catch (Exception ex)
        {
            stopwatch.Stop();
            logger.LogError(ex, "AI 调用失败，分类: {CategoryName}，耗时: {ElapsedMs}ms", categoryName, stopwatch.ElapsedMilliseconds);
            throw;
        }
    }

    /// <summary>
    /// 生成预算执行报告（HTML格式）
    /// </summary>
    public async Task<string?> GenerateBudgetReportAsync(string promptWithData, int year, int month)
    {
        logger.LogInformation("正在生成预算执行报告: {Year}年{Month}月", year, month);

        // 直接使用传入的完整prompt（包含数据和格式要求）
        var htmlReport = await openAiService.ChatAsync(promptWithData);
        if (string.IsNullOrWhiteSpace(htmlReport))
        {
            logger.LogWarning("AI 未返回有效的报告内容");
            return null;
        }

        logger.LogInformation("成功生成预算执行报告: {Year}年{Month}月", year, month);
        return htmlReport;
    }

    private (string card, string reason, decimal amount, decimal balance, TransactionType type, DateTime? occurredAt)? ParseEmailSingleRecord(JsonElement obj)
    {
        var card = obj.TryGetProperty("card", out var pCard) ? pCard.GetString() ?? string.Empty : string.Empty;
        var reason = obj.TryGetProperty("reason", out var pReason) ? pReason.GetString() ?? string.Empty : string.Empty;
        var typeStr = obj.TryGetProperty("type", out var pType) ? pType.GetString() ?? string.Empty : string.Empty;
        var occurredAtStr = obj.TryGetProperty("occurredAt", out var pOccurredAt) ? pOccurredAt.GetString() ?? string.Empty : string.Empty;

        var amount = 0m;
        if (obj.TryGetProperty("amount", out var pAmount))
        {
            if (pAmount.ValueKind == JsonValueKind.Number && pAmount.TryGetDecimal(out var d)) amount = d;
            else if (pAmount.ValueKind == JsonValueKind.String && decimal.TryParse(pAmount.GetString(), NumberStyles.Any, CultureInfo.InvariantCulture, out var ds)) amount = ds;
        }

        var balance = 0m;
        if (obj.TryGetProperty("balance", out var pBalance))
        {
            if (pBalance.ValueKind == JsonValueKind.Number && pBalance.TryGetDecimal(out var d2)) balance = d2;
            else if (pBalance.ValueKind == JsonValueKind.String && decimal.TryParse(pBalance.GetString(), NumberStyles.Any, CultureInfo.InvariantCulture, out var ds2)) balance = ds2;
        }

        if (string.IsNullOrWhiteSpace(card) || string.IsNullOrWhiteSpace(reason))
        {
            return null;
        }

        var occurredAt = (DateTime?)null;
        if (DateTime.TryParse(occurredAtStr, out var occurredAtValue))
        {
            occurredAt = occurredAtValue;
        }

        var type = DetermineTransactionType(typeStr, reason, amount);
        return (card, reason, amount, balance, type, occurredAt);
    }

    /// <summary>
    /// 判断交易类型
    /// </summary>
    private TransactionType DetermineTransactionType(string typeStr, string reason, decimal amount)
    {
        // 优先使用明确的类型字符串
        if (!string.IsNullOrWhiteSpace(typeStr))
        {
            if (typeStr.Contains("收入") || typeStr.Contains("income") || typeStr.Equals("收", StringComparison.OrdinalIgnoreCase))
                return TransactionType.Income;
            if (typeStr.Contains("支出") || typeStr.Contains("expense") || typeStr.Equals("支", StringComparison.OrdinalIgnoreCase))
                return TransactionType.Expense;
        }

        // 根据交易原因中的关键词判断
        var lowerReason = reason.ToLower();

        // 收入关键词
        string[] incomeKeywords =
        [
            "工资", "奖金", "退款",
            "返现", "收入", "转入",
            "存入", "利息", "分红",
            "入账", "收款",

            // 常见扩展
            "实发工资", "薪资", "薪水", "薪酬",
            "提成", "佣金", "劳务费",
            "报销入账", "报销款", "补贴", "补助",

            "退款成功", "退回", "退货退款",
            "返现入账", "返利", "返佣",

            "到账", "已到账", "入账成功",
            "收款成功", "收到款项", "到账金额",
            "资金转入", "资金收入",

            "转账收入", "转账入账", "他行来账",
            "工资代发", "代发工资", "单位打款",

            "利息收入", "收益", "收益发放", "理财收益",
            "分红收入", "股息", "红利",

            // 平台常用词
            "红包", "红包收入", "红包入账",
            "奖励金", "活动奖励", "补贴金",
            "现金奖励", "推广奖励", "返现奖励",

            // 存取类
            "现金存入", "柜台存入", "ATM存入",
            "他人转入", "他人汇入"
        ];
        if (incomeKeywords.Any(k => lowerReason.Contains(k)))
            return TransactionType.Income;

        // 支出关键词
        string[] expenseKeywords =
        [
            "消费", "支付", "购买",
            "转出", "取款", "支出",
            "扣款", "缴费", "付款",
            "刷卡",

            // 常见扩展
            "支出金额", "支出人民币", "已支出",
            "已消费", "消费支出", "消费人民币",
            "已支付", "成功支付", "支付成功", "交易支付",
            "已扣款", "扣款成功", "扣费", "扣费成功",
            "转账", "转账支出", "向外转账", "已转出",
            "提现", "现金支出", "现金取款",
            "扣除", "扣除金额", "记账支出",

            // 账单/通知类用语
            "本期应还", "本期应还金额", "本期账单金额",
            "本期应还人民币", "最低还款额",
            "本期欠款", "欠款金额",

            // 线上平台常见用语
            "订单支付", "订单扣款", "订单消费",
            "交易支出", "交易扣款", "交易成功支出",
            "话费充值", "流量充值", "水费", "电费", "燃气费",
            "物业费", "服务费", "手续费", "年费", "会费",
            "利息支出", "还款支出", "代扣", "代缴",

            // 信用卡/花呗等场景
            "信用卡还款", "花呗还款", "白条还款",
            "分期还款", "账单还款", "自动还款"
        ];
        if (expenseKeywords.Any(k => lowerReason.Contains(k)))
            return TransactionType.Expense;

        // 根据金额正负判断（如果金额为负数，可能是支出）
        if (amount < 0)
            return TransactionType.Expense;
        if (amount > 0)
            return TransactionType.Income;

        // 默认为支出
        return TransactionType.Expense;
    }
}

/// <summary>
/// 分组分类结果DTO（用于AI返回结果解析）
/// </summary>
public record GroupClassifyResult
{
    [JsonPropertyName("reason")]
    public string Reason { get; init; } = string.Empty;

    [JsonPropertyName("classify")]
    public string? Classify { get; init; }

    [JsonPropertyName("type")]
    public TransactionType Type { get; init; }
}

public record TransactionParseResult(string OccurredAt, string Classify, decimal Amount, string Reason, TransactionType Type);