Claudeでの解析

Epub/KoeBook.Epub/Services/AnalyzerService.cs

@@ -1,5 +1,4 @@
-using System.Text;
-using System.Text.RegularExpressions;
+using System.Text.RegularExpressions;
 using KoeBook.Core;
 using KoeBook.Core.Contracts.Services;
 using KoeBook.Core.Models;
@@ -49,22 +48,8 @@ public async ValueTask<BookScripts> AnalyzeAsync(BookProperties bookProperties,
                 return p.ScriptLine = new ScriptLine(line, "", "");
             }).ToList();
 
-        // 800文字以上になったら１チャンクに分ける
-        var chunks = new List<string>();
-        var chunk = new StringBuilder();
-        foreach (var line in scriptLines)
-        {
-            if (chunk.Length + line.Text.Length > 800)
-            {
-                chunks.Add(chunk.ToString());
-                chunk.Clear();
-            }
-            chunk.AppendLine(line.Text);
-        }
-        if (chunk.Length > 0) chunks.Add(chunk.ToString());
-
-        // GPT4による話者、スタイル解析
-        var bookScripts = await _llmAnalyzerService.LlmAnalyzeScriptLinesAsync(bookProperties, scriptLines, chunks, cancellationToken);
+        // LLMによる話者、スタイル解析
+        var bookScripts = await _llmAnalyzerService.LlmAnalyzeScriptLinesAsync(bookProperties, scriptLines, cancellationToken)!;
 
         return bookScripts;
     }

KoeBook.Core/Contracts/Services/IClaudeService.cs

@@ -0,0 +1,8 @@
+using Claudia;
+
+namespace KoeBook.Core.Contracts.Services;
+
+public interface IClaudeService
+{
+    IMessages? Messages { get; }
+}

KoeBook.Core/Contracts/Services/ILlmAnalyzerService.cs

@@ -4,5 +4,5 @@ namespace KoeBook.Core.Contracts.Services;
 
 public interface ILlmAnalyzerService
 {
-    ValueTask<BookScripts> LlmAnalyzeScriptLinesAsync(BookProperties bookProperties, List<ScriptLine> scriptLines, List<string> chunks, CancellationToken cancellationToken);
+    ValueTask<BookScripts> LlmAnalyzeScriptLinesAsync(BookProperties bookProperties, List<ScriptLine> scriptLines, CancellationToken cancellationToken);
 }

KoeBook.Core/EbookException.cs

@@ -45,6 +45,12 @@ public enum ExceptionType
     [EnumMember(Value = "GPT4による話者・スタイル設定に失敗しました")]
     Gpt4TalkerAndStyleSettingFailed,
 
+    [EnumMember(Value = "APIキーが設定されていません")]
+    ApiKeyNotSet,
+
+    [EnumMember(Value = "Claudeによる話者・スタイル設定に失敗しました")]
+    ClaudeTalkerAndStyleSettingFailed,
+
     [EnumMember(Value = "webページの解析に失敗しました")]
     WebScrapingFailed,
 

KoeBook.Core/KoeBook.Core.csproj

@@ -9,7 +9,8 @@
   </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="Betalgo.OpenAI" Version="7.4.6" />
+    <PackageReference Include="Betalgo.OpenAI" Version="8.1.1" />
+    <PackageReference Include="Claudia" Version="1.2.0" />
     <PackageReference Include="FastEnum" Version="1.8.0" />
     <PackageReference Include="NAudio" Version="2.2.1" />
     <PackageReference Include="System.Security.Cryptography.ProtectedData" Version="8.0.0" />

KoeBook.Core/Services/ChatGptAnalyzerService.cs

@@ -13,8 +13,20 @@ public partial class ChatGptAnalyzerService(IOpenAIService openAIService, IDispl
     private readonly IOpenAIService _openAiService = openAIService;
     private readonly IDisplayStateChangeService _displayStateChangeService = displayStateChangeService;
 
-    public async ValueTask<BookScripts> LlmAnalyzeScriptLinesAsync(BookProperties bookProperties, List<ScriptLine> scriptLines, List<string> chunks, CancellationToken cancellationToken)
+    public async ValueTask<BookScripts> LlmAnalyzeScriptLinesAsync(BookProperties bookProperties, List<ScriptLine> scriptLines, CancellationToken cancellationToken)
     {
+        var chunks = new List<string>();
+        var chunk = new StringBuilder();
+        foreach (var line in scriptLines)
+        {
+            if (chunk.Length + line.Text.Length > 800)
+            {
+                chunks.Add(chunk.ToString());
+                chunk.Clear();
+            }
+            chunk.AppendLine(line.Text);
+        }
+        if (chunk.Length > 0) chunks.Add(chunk.ToString());
         var progress = _displayStateChangeService.ResetProgress(bookProperties, GenerationState.Analyzing, chunks.Count);
         Queue<string> summaryList = new();
         Queue<string> characterList = new();
@@ -147,7 +159,7 @@ 3. Target Sentence
             },
             Model = OpenAI.ObjectModels.Models.Gpt_4_turbo_preview,
             MaxTokens = 4000
-        });
+        }, cancellationToken: cancellationToken);
         if (completionResult.Successful)
         {
             var result = completionResult.Choices.First().Message.Content;
@@ -261,7 +273,7 @@ 3. Story
             },
             Model = OpenAI.ObjectModels.Models.Gpt_4_turbo_preview,
             MaxTokens = 4000
-        });
+        }, cancellationToken: cancellationToken);
         if (completionResult.Successful)
         {
             var result = completionResult.Choices.First().Message.Content;
@@ -356,7 +368,7 @@ Make a table of character names and voices
             },
             Model = OpenAI.ObjectModels.Models.Gpt_4_turbo_preview,
             MaxTokens = 4000
-        });
+        }, cancellationToken: cancellationToken);
         if (completionResult.Successful)
         {
             var result = completionResult.Choices.First().Message.Content;

KoeBook.Core/Services/ClaudeAnalyzerService.cs

@@ -0,0 +1,201 @@
+using System.Text;
+using KoeBook.Core.Contracts.Services;
+using KoeBook.Core.Models;
+
+namespace KoeBook.Core.Services;
+
+public partial class ClaudeAnalyzerService(IClaudeService claudeService, IDisplayStateChangeService displayStateChangeService, ISoundGenerationSelectorService soundGenerationSelectorService) : ILlmAnalyzerService
+{
+    private readonly IClaudeService _claudeService = claudeService;
+    private readonly IDisplayStateChangeService _displayStateChangeService = displayStateChangeService;
+    private readonly ISoundGenerationSelectorService _soundGenerationSelectorService = soundGenerationSelectorService;
+
+    public async ValueTask<BookScripts> LlmAnalyzeScriptLinesAsync(BookProperties bookProperties, List<ScriptLine> scriptLines, CancellationToken cancellationToken)
+    {
+        var lineNumberingText = LineNumbering(scriptLines);
+        if (_claudeService.Messages is null)
+        {
+            throw new EbookException(ExceptionType.ApiKeyNotSet);
+        }
+        try
+        {
+            var message1 = await _claudeService.Messages.CreateAsync(new()
+            {
+                Model = Claudia.Models.Claude3Opus,
+                MaxTokens = 4000,
+                Messages = [new()
+                {
+                    Role = "user",
+                    Content = CreateCharacterGuessPrompt(lineNumberingText)
+                }]
+            },
+                cancellationToken: cancellationToken
+            );
+            var characterList = ExtractCharacterList(message1.ToString(), scriptLines);
+
+            var message2 = await _claudeService.Messages.CreateAsync(new()
+            {
+                Model = Claudia.Models.Claude3Opus,
+                MaxTokens = 4000,
+                Messages = [new()
+                {
+                    Role = "user",
+                    Content = CreateVoiceTypeAnalyzePrompt(characterList)
+                }]
+            },
+                cancellationToken: cancellationToken
+            );
+
+            var characterVoiceMapping = ExtractCharacterVoiceMapping(message2.ToString(), characterList);
+
+            return new(bookProperties, new(characterVoiceMapping)) { ScriptLines = scriptLines };
+        }
+        catch (OperationCanceledException) { throw; }
+        catch (Exception e)
+        {
+            throw new EbookException(ExceptionType.ClaudeTalkerAndStyleSettingFailed, innerException: e);
+        }
+    }
+
+    private Dictionary<string, string> ExtractCharacterVoiceMapping(string response, List<Character> characterList)
+    {
+        return response.Split("\n")
+           .SkipWhile(l => !l.StartsWith("[Assign Voices]"))
+           .Where(l => l.StartsWith('c'))
+           .Select(l =>
+           {
+               var characterId = l[1..l.IndexOf('.')];
+               var voiceType = l[(l.IndexOf(':') + 2)..];
+               // voiceTypeが_soundGenerationSelectorService.Modelsに含まれているかチェック
+               return _soundGenerationSelectorService.Models.Any(x => x.Name == voiceType)
+                   ? (characterId, voiceType)
+                   : (characterId, string.Empty);
+           }).ToDictionary();
+    }
+
+    private static string CreateCharacterGuessPrompt(string lineNumberingText)
+    {
+        return $$"""
+                {{lineNumberingText}}
+
+                Notes:
+                - For narration parts, which are not enclosed in quotation marks, select the narrator.
+                - For dialogues enclosed in quotation marks, assign a voice other than the narrator.
+                - In the character description, include the appropriate voice characteristics.
+                Tasks: Based on the notes above, perform the following two tasks:
+                - List of characters Objective: To understand the characters appearing in the text, list the character ID, name, and description for all characters who speak at least one line.
+                - Output the speaking character or narrator for each line Objective: To identify which character is speaking in each line of the text, output the speaking character or narrator for all lines. Carefully recognize the context and output with attention. Select only one character_id per line.
+                - Revise CHARACTER LIST and VOICE ID (in the event that the CHARACTER LIST is incomplete)
+                Output Format:
+                [CHARACTER LIST]
+                c0. ナレーター: {character_and_voice_description, example:"The person who speaks the narration parts. A calm-toned male voice."}
+                c1. {character_name}: {character_and_voice_description}
+                {character_id}. {character_name}: {character_and_voice_description}
+                ...
+                [VOICE ID]
+                1. {character_id} {narration|dialogue}
+                2. {character_id} {narration|dialogue}
+                3. {character_id} {narration|dialogue}
+                ...
+                [REVISE CHARACTER LIST]
+                c0. ナレーター: {character_and_voice_description}
+                c1. {character_name}: {character_and_voice_description}
+                {character_id}. {character_name}: {character_and_voice_description}
+                ...
+                [REVISE VOICE ID]
+                1. {character_id} {narration|dialogue}
+                2. {character_id} {narration|dialogue}
+                3. {character_id} {narration|dialogue}
+                ...
+                """;
+    }
+
+    private string CreateVoiceTypeAnalyzePrompt(List<Character> characterList)
+    {
+        return $$"""
+            Assign the most fitting voice type to each character from the provided list, ensuring the chosen voice aligns with their role and attributes in the story. Only select from the available voice types.
+
+            Characters:
+            {{string.Join("\n", characterList.Select(character => $"c{character.Id}. {character.Name}: {character.Description}"))}}
+
+            Voice Types:
+            {{string.Join(",", _soundGenerationSelectorService.Models.Select(m => m.Name))}}
+
+            Output Format:
+            [Assign Voices]
+            c0. {character_name}: {voice_type}
+            c1. {character_name}: {voice_type}
+            """;
+    }
+
+    private static string LineNumbering(List<ScriptLine> scriptLines)
+    {
+        var sb = new StringBuilder();
+        foreach (var (index, scriptLine) in scriptLines.Select((x, i) => (i, x)))
+        {
+            sb.AppendLine($"{index + 1}. {scriptLine.Text}");
+        }
+        return sb.ToString();
+    }
+
+    private static List<Character> ExtractCharacterList(string response, List<ScriptLine> scriptLines)
+    {
+        var characterList = new List<Character>();
+        var lines = response.Split("\n");
+        var characterListStartIndex = 0;
+        var characterListEndIndex = 0;
+        for (var i = 0; i < lines.Length; i++)
+        {
+            if (lines[i].StartsWith("[REVISE CHARACTER LIST]"))
+            {
+                characterListStartIndex = i + 1;
+            }
+
+            if (lines[i].StartsWith("[REVISE VOICE ID]"))
+            {
+                characterListEndIndex = i;
+                break;
+            }
+        }
+        for (var i = characterListStartIndex; i < characterListEndIndex; i++)
+        {
+            var line = lines[i];
+            if (line.StartsWith('c'))
+            {
+                var characterId = line[1..line.IndexOf('.')];
+                var characterName = line[(line.IndexOf('.') + 2)..line.IndexOf(':')];
+                var characterDescription = line[(line.IndexOf(':') + 2)..];
+                characterList.Add(new Character(characterId, characterName, characterDescription));
+            }
+        }
+
+        var dic = characterList.Select(x => KeyValuePair.Create(x.Id, x.Name)).ToDictionary();
+        var lines2 = lines.AsSpan()[(characterListEndIndex + 1)..];
+
+        for (var i = 0; i < lines2.Length; i++)
+        {
+            var line = lines2[i].AsSpan().TrimEnd();
+            line = line[(line.IndexOf(' ') + 2)..];//cまで無視
+            line = line[..line.IndexOf(' ')];
+            if (dic.TryGetValue(line.ToString(), out var characterName))
+            {
+                scriptLines[i].Character = characterName;
+            }
+        }
+        return characterList;
+    }
+
+    private class Character
+    {
+        public string Id { get; }
+        public string Name { get; }
+        public string Description { get; }
+        public string VoiceType { get; set; } = string.Empty;
+        public Character(string id, string name, string description)
+        {
+            Id = id;
+            Name = name;
+            Description = description;
+        }
+    }
+}

KoeBook.Core/Services/ClaudeService.cs

@@ -0,0 +1,31 @@
+using Claudia;
+using KoeBook.Core.Contracts.Services;
+
+namespace KoeBook.Core.Services;
+
+public class ClaudeService(ISecretSettingsService secretSettingsService) : IClaudeService
+{
+    private readonly ISecretSettingsService _secretSettingsService = secretSettingsService;
+
+    private string? _apiKey;
+    private Anthropic? _anthropic;
+
+    public IMessages? Messages => GetAnthropic()?.Messages;
+
+
+    private Anthropic? GetAnthropic()
+    {
+        if (_apiKey != _secretSettingsService.ApiKey)
+        {
+            if (string.IsNullOrEmpty(_secretSettingsService.ApiKey))
+            {
+                _apiKey = _secretSettingsService.ApiKey;
+                return null;
+            }
+
+            _anthropic = new Anthropic { ApiKey = _secretSettingsService.ApiKey };
+            _apiKey = _secretSettingsService.ApiKey;
+        }
+        return _anthropic;
+    }
+}

KoeBook.Core/Services/MyOpenAiService.cs

@@ -36,6 +36,8 @@ public class MyOpenAiService(ISecretSettingsService secretSettingsService, IHttp
 
     public IAudioService Audio => GetOpenAiService()?.Audio!;
 
+    public IBatchService Batch => GetOpenAiService()?.Batch!;
+
     public void SetDefaultModelId(string modelId)
     {
         GetOpenAiService()?.SetDefaultModelId(modelId);

KoeBook/App.xaml.cs

@@ -6,7 +6,6 @@
 using KoeBook.Core.Contracts.Services;
 using KoeBook.Core.Services;
 using KoeBook.Core.Services.Mocks;
-using KoeBook.Epub;
 using KoeBook.Epub.Contracts.Services;
 using KoeBook.Epub.Services;
 using KoeBook.Models;

KoeBook/Startup.cs

@@ -33,8 +33,8 @@ public static IHostBuilder UseCoreStartup(this IHostBuilder builder)
             services.AddSingleton<IEpubGenerateService, EpubGenerateService>();
             services.AddSingleton<IEpubDocumentStoreService, EpubDocumentStoreService>();
             services.AddSingleton<IAnalyzerService, AnalyzerService>();
-            services.AddSingleton<ILlmAnalyzerService, ChatGptAnalyzerService>();
-            services.AddSingleton<OpenAI.Interfaces.IOpenAIService, MyOpenAiService>();
+            services.AddSingleton<ILlmAnalyzerService, ClaudeAnalyzerService>();
+            services.AddSingleton<IClaudeService, ClaudeService>();
 
             // Epub Services
             services