Added a button to regenerate the last AI response (#247)

app/MindWork AI Studio/Chat/ChatThread.cs

@@ -97,17 +97,45 @@ public sealed record ChatThread
         logger.LogInformation(logMessage);
         return systemPromptText;
     }
-    
+
     /// <summary>
     /// Removes a content block from this chat thread.
     /// </summary>
     /// <param name="content">The content block to remove.</param>
-    public void Remove(IContent content)
+    /// <param name="removeForRegenerate">Indicates whether the content block is removed for
+    /// regeneration purposes. True, when the content block is removed for regeneration purposes,
+    /// which will not remove the previous user block if it is hidden from the user.</param>
+    public void Remove(IContent content, bool removeForRegenerate = false)
     {
         var block = this.Blocks.FirstOrDefault(x => x.Content == content);
         if(block is null)
             return;
-        
+
+        //
+        // Remove the previous user block if it is hidden from the user. Otherwise,
+        // the experience might be confusing for the user.
+        //
+        // Explanation, using the ERI assistant as an example:
+        // - The ERI assistant generates for every file a hidden user prompt.
+        // - In the UI, the user can only see the AI's responses, not the hidden user prompts.
+        // - Now, the user removes one AI response
+        // - The hidden user prompt is still there, but the user can't see it.
+        // - Since the user prompt is hidden, neither is it possible to remove nor edit it.
+        // - This method solves this issue by removing the hidden user prompt when the AI response is removed.
+        //
+        if (block.Role is ChatRole.AI && !removeForRegenerate)
+        {
+            var sortedBlocks = this.Blocks.OrderBy(x => x.Time).ToList();
+            var index = sortedBlocks.IndexOf(block);
+            if (index > 0)
+            {
+                var previousBlock = sortedBlocks[index - 1];
+                if (previousBlock.Role is ChatRole.USER && previousBlock.HideFromUser)
+                    this.Blocks.Remove(previousBlock);
+            }
+        }
+
+        // Remove the block from the chat thread:
         this.Blocks.Remove(block);
     }
 }

app/MindWork AI Studio/Chat/ContentBlockComponent.razor

@@ -12,6 +12,12 @@
             <MudText Typo="Typo.body1">@this.Role.ToName() (@this.Time)</MudText>
         </CardHeaderContent>
         <CardHeaderActions>
+            @if (this.IsLastContentBlock && this.Role is ChatRole.AI && this.RegenerateFunc is not null)
+            {
+                <MudTooltip Text="Regenerate" Placement="Placement.Bottom">
+                    <MudIconButton Icon="@Icons.Material.Filled.Recycling" Color="Color.Tertiary" OnClick="@this.RegenerateBlock"/>
+                </MudTooltip>
+            }
             @if (this.RemoveBlockFunc is not null)
             {
                 <MudTooltip Text="Removes this block" Placement="Placement.Bottom">

app/MindWork AI Studio/Chat/ContentBlockComponent.razor.cs

@@ -40,10 +40,19 @@ public partial class ContentBlockComponent : ComponentBase
     /// </summary>
     [Parameter]
     public string Class { get; set; } = string.Empty;
+    
+    [Parameter]
+    public bool IsLastContentBlock { get; set; } = false;
 
     [Parameter]
     public Func<IContent, Task>? RemoveBlockFunc { get; set; }
     
+    [Parameter]
+    public Func<IContent, Task>? RegenerateFunc { get; set; }
+    
+    [Parameter]
+    public Func<bool> RegenerateEnabled { get; set; } = () => false;
+    
     [Inject]
     private RustService RustService { get; init; } = null!;
     
@@ -143,4 +152,22 @@ public partial class ContentBlockComponent : ComponentBase
         if (remove.HasValue && remove.Value)
             await this.RemoveBlockFunc(this.Content);
     }
+    
+    private async Task RegenerateBlock()
+    {
+        if (this.RegenerateFunc is null)
+            return;
+        
+        if(this.Role is not ChatRole.AI)
+            return;
+        
+        var regenerate = await this.DialogService.ShowMessageBox(
+            "Regenerate Message",
+            "Do you really want to regenerate this message?",
+            "Yes, regenerate it",
+            "No, keep it");
+        
+        if (regenerate.HasValue && regenerate.Value)
+            await this.RegenerateFunc(this.Content);
+    }
 }