fix

2026-02-15 10:10:28 +08:00
parent e51a3edd50
commit a88556c784
92 changed files with 6751 additions and 776 deletions
--- a/Service/AI/ClassificationIconPromptProvider.cs
+++ b/Service/AI/ClassificationIconPromptProvider.cs
@@ -0,0 +1,113 @@
+namespace Service.AI;
+
+/// <summary>
+/// 分类图标生成提示词提供器实现
+/// </summary>
+public class ClassificationIconPromptProvider : IClassificationIconPromptProvider
+{
+    private readonly ILogger<ClassificationIconPromptProvider> _logger;
+    private readonly IconPromptSettings _config;
+    private readonly Random _random = new();
+
+    public ClassificationIconPromptProvider(
+        ILogger<ClassificationIconPromptProvider> logger,
+        IOptions<IconPromptSettings> config)
+    {
+        _logger = logger;
+        _config = config.Value;
+    }
+
+    public string GetPrompt(string categoryName, TransactionType categoryType)
+    {
+        var typeText = GetCategoryTypeText(categoryType);
+        var useNewPrompt = ShouldUseNewPrompt();
+
+        var template = useNewPrompt
+            ? _config.DefaultPromptTemplate
+            : _config.OldDefaultPromptTemplate;
+
+        string prompt;
+        if (useNewPrompt)
+        {
+            prompt = PromptTemplateEngine.ReplaceForIconGeneration(
+                template,
+                categoryName,
+                typeText,
+                _config.ColorScheme,
+                _config.StyleStrength);
+        }
+        else
+        {
+            prompt = PromptTemplateEngine.ReplaceForIconGeneration(
+                template,
+                categoryName,
+                typeText,
+                _config.ColorScheme,
+                0);
+        }
+
+        _logger.LogDebug("使用 {PromptType} 提示词生成图标，分类：{CategoryName}",
+            useNewPrompt ? "新版" : "旧版",
+            categoryName);
+
+        return prompt;
+    }
+
+    public string GetSingleIconPrompt(string categoryName, TransactionType categoryType)
+    {
+        var typeText = GetCategoryTypeText(categoryType);
+        var useNewPrompt = ShouldUseNewPrompt();
+
+        var template = useNewPrompt
+            ? _config.SingleIconPromptTemplate
+            : _config.OldSingleIconPromptTemplate;
+
+        string prompt;
+        if (useNewPrompt)
+        {
+            prompt = PromptTemplateEngine.ReplaceForIconGeneration(
+                template,
+                categoryName,
+                typeText,
+                _config.ColorScheme,
+                _config.StyleStrength);
+        }
+        else
+        {
+            prompt = PromptTemplateEngine.ReplaceForIconGeneration(
+                template,
+                categoryName,
+                typeText,
+                _config.ColorScheme,
+                0);
+        }
+
+        _logger.LogDebug("使用 {PromptType} 提示词生成单个图标，分类：{CategoryName}",
+            useNewPrompt ? "新版" : "旧版",
+            categoryName);
+
+        return prompt;
+    }
+
+    private bool ShouldUseNewPrompt()
+    {
+        if (!_config.EnableNewPrompt)
+        {
+            return false;
+        }
+
+        var randomValue = _random.NextDouble();
+        return randomValue < _config.GrayScaleRatio;
+    }
+
+    private static string GetCategoryTypeText(TransactionType categoryType)
+    {
+        return categoryType switch
+        {
+            TransactionType.Expense => "支出",
+            TransactionType.Income => "收入",
+            TransactionType.None => "不计入收支",
+            _ => "未知"
+        };
+    }
+}
--- a/Service/AI/IClassificationIconPromptProvider.cs
+++ b/Service/AI/IClassificationIconPromptProvider.cs
@@ -0,0 +1,23 @@
+namespace Service.AI;
+
+/// <summary>
+/// 分类图标生成提示词提供器接口
+/// </summary>
+public interface IClassificationIconPromptProvider
+{
+    /// <summary>
+    /// 获取分类图标生成的提示词（生成 5 个图标）
+    /// </summary>
+    /// <param name="categoryName">分类名称</param>
+    /// <param name="categoryType">分类类型（收入/支出）</param>
+    /// <returns>用于生成图标的提示词</returns>
+    string GetPrompt(string categoryName, TransactionType categoryType);
+
+    /// <summary>
+    /// 获取单个图标生成的提示词（仅生成 1 个图标）
+    /// </summary>
+    /// <param name="categoryName">分类名称</param>
+    /// <param name="categoryType">分类类型（收入/支出）</param>
+    /// <returns>用于生成单个图标的提示词</returns>
+    string GetSingleIconPrompt(string categoryName, TransactionType categoryType);
+}
--- a/Service/AI/PromptTemplateEngine.cs
+++ b/Service/AI/PromptTemplateEngine.cs
@@ -0,0 +1,66 @@
+namespace Service.AI;
+
+/// <summary>
+/// 提示词模板引擎，处理占位符替换
+/// </summary>
+public class PromptTemplateEngine
+{
+    /// <summary>
+    /// 替换模板中的占位符
+    /// </summary>
+    /// <param name="template">模板字符串，支持 {{key}} 格式的占位符</param>
+    /// <param name="placeholders">占位符字典，key 为占位符名称（不含 {{ }}），value 为替换值</param>
+    /// <returns>替换后的字符串</returns>
+    public static string ReplacePlaceholders(string template, Dictionary<string, string> placeholders)
+    {
+        if (string.IsNullOrEmpty(template) || placeholders == null || placeholders.Count == 0)
+        {
+            return template;
+        }
+
+        var result = template;
+        foreach (var placeholder in placeholders)
+        {
+            var key = placeholder.Key;
+            var value = placeholder.Value ?? string.Empty;
+            result = result.Replace($"{{{{{key}}}}}", value);
+        }
+
+        return result;
+    }
+
+    /// <summary>
+    /// 替换模板中的占位符（简化版本）
+    /// </summary>
+    /// <param name="template">模板字符串</param>
+    /// <param name="categoryName">分类名称</param>
+    /// <param name="categoryType">分类类型</param>
+    /// <param name="colorScheme">颜色方案</param>
+    /// <param name="styleStrength">风格强度（0.0-1.0）</param>
+    /// <returns>替换后的字符串</returns>
+    public static string ReplaceForIconGeneration(
+        string template,
+        string categoryName,
+        string categoryType,
+        string colorScheme,
+        double styleStrength)
+    {
+        var strengthDescription = styleStrength switch
+        {
+            >= 0.9 => "极度简约（仅保留最核心元素）",
+            >= 0.7 => "高度简约（去除所有装饰）",
+            >= 0.5 => "简约（保留必要细节）",
+            _ => "适中"
+        };
+
+        var placeholders = new Dictionary<string, string>
+        {
+            ["category_name"] = categoryName,
+            ["category_type"] = categoryType,
+            ["color_scheme"] = colorScheme,
+            ["style_strength"] = $"{styleStrength:F1} - {strengthDescription}"
+        };
+
+        return ReplacePlaceholders(template, placeholders);
+    }
+}
--- a/Service/AI/SmartHandleService.cs
+++ b/Service/AI/SmartHandleService.cs
@@ -41,7 +41,8 @@ public class SmartHandleService(
    ILogger<SmartHandleService> logger,
    ITransactionCategoryRepository categoryRepository,
    IOpenAiService openAiService,
-    IConfigService configService
+    IConfigService configService,
+    IClassificationIconPromptProvider iconPromptProvider
 ) : ISmartHandleService
 {
    public async Task SmartClassifyAsync(long[] transactionIds, Action<(string, string)> chunkAction)
@@ -541,6 +542,32 @@ public class SmartHandleService(
        };
    }

+    /// <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()
    {
        // 获取所有分类
@@ -649,46 +676,9 @@ public class SmartHandleService(
    {
        logger.LogInformation("正在为分类 {CategoryName} 生成 {IconCount} 个图标", categoryName, iconCount);

-        var typeText = categoryType == TransactionType.Expense ? "支出" : "收入";
+        var systemPrompt = iconPromptProvider.GetPrompt(categoryName, categoryType);

-        var systemPrompt = """
-            你是一个专业的 SVG 图标设计师，擅长创作精美、富有表现力的图标。
-            请根据分类名称和类型，生成 5 个风格迥异、视觉效果突出的 SVG 图标。
-            
-            设计要求：
-            1. 尺寸：24x24，viewBox="0 0 24 24"
-            2. 色彩：使用丰富的渐变色和多色搭配，让图标更有吸引力和辨识度
-               - 可以使用 <linearGradient> 或 <radialGradient> 创建渐变效果
-               - 不同元素使用不同颜色，增加层次感
-               - 根据分类含义选择合适的配色方案（如餐饮用暖色系、交通用蓝色系等）
-            3. 设计风格：5 个图标必须风格明显不同，避免雷同
-               - 第1个：扁平化风格，色彩鲜明，使用渐变
-               - 第2个：线性风格，多色描边，细节丰富
-               - 第3个：3D立体风格，使用阴影和高光效果
-               - 第4个：卡通可爱风格，圆润造型，活泼配色
-               - 第5个：现代简约风格，几何与曲线结合，优雅配色
-            4. 细节丰富：不要只用简单的几何图形，添加特征性的细节元素
-               - 例如：餐饮可以加刀叉、蒸汽、食材纹理等
-               - 交通可以加轮胎、车窗、尾气等
-               - 每个图标要有独特的视觉记忆点
-            5. 图标要直观表达分类含义，让人一眼就能识别
-            6. 只返回 JSON 数组格式，包含 5 个完整的 SVG 字符串，不要有任何其他文字说明
-            
-            重要：每个 SVG 必须是自包含的完整代码，包含所有必要的 gradient 定义。
-            """;
-
-        var userPrompt = $"""
-            分类名称：{categoryName}
-            分类类型：{typeText}
-            
-            请为这个分类生成 {iconCount} 个精美的、风格各异的彩色 SVG 图标。
-            确保每个图标都有独特的视觉特征，不会与其他图标混淆。
-            
-            返回格式（纯 JSON 数组，无其他内容）：
-            ["<svg>...</svg>", "<svg>...</svg>", "<svg>...</svg>", "<svg>...</svg>", "<svg>...</svg>"]
-            """;
-
-        var response = await openAiService.ChatAsync(systemPrompt, userPrompt, timeoutSeconds: 60 * 10);
+        var response = await openAiService.ChatAsync(systemPrompt, "", timeoutSeconds: 60 * 10);

        if (string.IsNullOrWhiteSpace(response))
        {
@@ -696,6 +686,15 @@ public class SmartHandleService(
            return null;
        }

+        // 清理可能的 markdown 代码块标记
+        response = CleanMarkdownCodeBlock(response);
+
+        if (string.IsNullOrWhiteSpace(response))
+        {
+            logger.LogWarning("清理后的 AI 响应为空，分类: {CategoryName}", categoryName);
+            return null;
+        }
+
        // 验证返回的是有效的 JSON 数组
        try
        {
@@ -724,45 +723,66 @@ public class SmartHandleService(
    {
        logger.LogInformation("正在为分类 {CategoryName} 生成单个图标", categoryName);

-        var typeText = categoryType == TransactionType.Expense ? "支出" : "收入";
+        // 使用单个图标生成的 Prompt（只生成 1 个图标，加快速度）
+        var systemPrompt = iconPromptProvider.GetSingleIconPrompt(categoryName, categoryType);

-        var systemPrompt = """
-            你是一个专业的SVG图标设计师。为预算分类生成极简风格的SVG图标。
-            
-            设计要求：
-            1. 尺寸：24x24，viewBox="0 0 24 24"
-            2. 使用丰富的渐变色和多色搭配，让图标更有吸引力
-            3. 图标要直观表达分类含义
-            4. 只返回SVG代码，不要有任何其他文字说明
-            """;
-
-        var userPrompt = $"""
-            请为「{categoryName}」{typeText}分类生成一个精美的SVG图标。
-            直接返回SVG代码，无需解释。
-            """;
-
-        var svgContent = await openAiService.ChatAsync(systemPrompt, userPrompt, timeoutSeconds: 30);
-        if (string.IsNullOrWhiteSpace(svgContent))
+        var stopwatch = System.Diagnostics.Stopwatch.StartNew();
+        try
        {
-            logger.LogWarning("AI 未返回有效的图标数据，分类: {CategoryName}", categoryName);
-            return null;
+            // 增加超时时间到 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;
+            }
        }
-
-        // 提取SVG标签
-        var svgMatch = System.Text.RegularExpressions.Regex.Match(
-            svgContent,
-            @"<svg[^>]*>.*?</svg>",
-            System.Text.RegularExpressions.RegexOptions.Singleline);
-
-        if (!svgMatch.Success)
+        catch (TimeoutException)
        {
-            logger.LogWarning("生成的内容不包含有效的SVG标签，分类: {CategoryName}", categoryName);
-            return null;
+            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;
        }
-
-        var svg = svgMatch.Value;
-        logger.LogInformation("成功为分类 {CategoryName} 生成单个图标", categoryName);
-        return svg;
    }

    /// <summary>