From 82cda5ee66f525c5c51c961218be9ea57629eed1 Mon Sep 17 00:00:00 2001
From: Thorsten Sommer <mail@tsommer.org>
Date: Fri, 12 Jul 2024 20:50:47 +0200
Subject: [PATCH] Added option to start new chat in the same workspace

---
 .../Components/Pages/Chat.razor               | 20 ++++++---
 .../Components/Pages/Chat.razor.cs            | 44 +++++++++++++++++--
 2 files changed, 55 insertions(+), 9 deletions(-)

diff --git a/app/MindWork AI Studio/Components/Pages/Chat.razor b/app/MindWork AI Studio/Components/Pages/Chat.razor
index ee454273..f37c427d 100644
--- a/app/MindWork AI Studio/Components/Pages/Chat.razor	
+++ b/app/MindWork AI Studio/Components/Pages/Chat.razor	
@@ -50,13 +50,23 @@
                     </MudTooltip>
                 }
 
-                <MudTooltip Text="Start new chat" Placement="@TOOLBAR_TOOLTIP_PLACEMENT">
-                    <MudIconButton Icon="@Icons.Material.Filled.AddComment" OnClick="() => this.StartNewChat()"/>
+                <MudTooltip Text="Start temporary chat" Placement="@TOOLBAR_TOOLTIP_PLACEMENT">
+                    <MudIconButton Icon="@Icons.Material.Filled.AddComment" OnClick="() => this.StartNewChat(useSameWorkspace: false)"/>
                 </MudTooltip>
 
-                <MudTooltip Text="Move chat to workspace" Placement="@TOOLBAR_TOOLTIP_PLACEMENT">
-                    <MudIconButton Icon="@Icons.Material.Filled.MoveToInbox" Disabled="@(!this.CanThreadBeSaved)" OnClick="() => this.MoveChatToWorkspace()"/>
-                </MudTooltip>
+                @if (!string.IsNullOrWhiteSpace(this.currentWorkspaceName))
+                {
+                    <MudTooltip Text="@this.TooltipAddChatToWorkspace" Placement="@TOOLBAR_TOOLTIP_PLACEMENT">
+                        <MudIconButton Icon="@Icons.Material.Filled.CommentBank" OnClick="() => this.StartNewChat(useSameWorkspace: true)"/>
+                    </MudTooltip>
+                }
+
+                @if (this.SettingsManager.ConfigurationData.WorkspaceStorageBehavior is not WorkspaceStorageBehavior.DISABLE_WORKSPACES)
+                {
+                    <MudTooltip Text="Move chat to (another) workspace" Placement="@TOOLBAR_TOOLTIP_PLACEMENT">
+                        <MudIconButton Icon="@Icons.Material.Filled.MoveToInbox" Disabled="@(!this.CanThreadBeSaved)" OnClick="() => this.MoveChatToWorkspace()"/>
+                    </MudTooltip>
+                }
             </MudToolBar>
         </MudPaper>
     </FooterContent>
diff --git a/app/MindWork AI Studio/Components/Pages/Chat.razor.cs b/app/MindWork AI Studio/Components/Pages/Chat.razor.cs
index 1569a905..817c43b6 100644
--- a/app/MindWork AI Studio/Components/Pages/Chat.razor.cs	
+++ b/app/MindWork AI Studio/Components/Pages/Chat.razor.cs	
@@ -37,6 +37,7 @@ public partial class Chat : ComponentBase, IAsyncDisposable
     private bool isStreaming;
     private string userInput = string.Empty;
     private string currentWorkspaceName = string.Empty;
+    private Guid currentWorkspaceId = Guid.Empty;
     private bool workspacesVisible;
     private Workspaces? workspaces;
     
@@ -64,6 +65,8 @@ public partial class Chat : ComponentBase, IAsyncDisposable
     private string InputLabel => this.IsProviderSelected ? $"Your Prompt (use selected instance '{this.selectedProvider.InstanceName}', provider '{this.selectedProvider.UsedProvider.ToName()}')" : "Select a provider first";
     
     private bool CanThreadBeSaved => this.chatThread is not null && this.chatThread.Blocks.Count > 0;
+
+    private string TooltipAddChatToWorkspace => $"Start new chat in workspace \"{this.currentWorkspaceName}\"";
     
     private async Task SendMessage()
     {
@@ -77,7 +80,7 @@ public partial class Chat : ComponentBase, IAsyncDisposable
         {
             this.chatThread = new()
             {
-                WorkspaceId = Guid.Empty,
+                WorkspaceId = this.currentWorkspaceId,
                 ChatId = Guid.NewGuid(),
                 Name = threadName,
                 Seed = this.RNG.Next(),
@@ -218,7 +221,7 @@ public partial class Chat : ComponentBase, IAsyncDisposable
         return threadName;
     }
 
-    private async Task StartNewChat()
+    private async Task StartNewChat(bool useSameWorkspace = false)
     {
         if (this.hasUnsavedChanges)
         {
@@ -233,11 +236,29 @@ public partial class Chat : ComponentBase, IAsyncDisposable
                 return;
         }
         
-        this.chatThread = null;
         this.isStreaming = false;
         this.hasUnsavedChanges = false;
         this.userInput = string.Empty;
-        this.currentWorkspaceName = string.Empty;
+
+        if (!useSameWorkspace)
+        {
+            this.chatThread = null;
+            this.currentWorkspaceId = Guid.Empty;
+            this.currentWorkspaceName = string.Empty;
+        }
+        else
+        {
+            this.chatThread = new()
+            {
+                WorkspaceId = this.currentWorkspaceId,
+                ChatId = Guid.NewGuid(),
+                Name = string.Empty,
+                Seed = this.RNG.Next(),
+                SystemPrompt = "You are a helpful assistant!",
+                Blocks = [],
+            };
+        }
+
         await this.inputField.Clear();
     }
 
@@ -249,6 +270,19 @@ public partial class Chat : ComponentBase, IAsyncDisposable
         if(this.workspaces is null)
             return;
         
+        if (this.hasUnsavedChanges)
+        {
+            var confirmationDialogParameters = new DialogParameters
+            {
+                { "Message", "Are you sure you want to move this chat? All unsaved changes will be lost." },
+            };
+        
+            var confirmationDialogReference = await this.DialogService.ShowAsync<ConfirmDialog>("Unsaved Changes", confirmationDialogParameters, DialogOptions.FULLSCREEN);
+            var confirmationDialogResult = await confirmationDialogReference.Result;
+            if (confirmationDialogResult.Canceled)
+                return;
+        }
+        
         var dialogParameters = new DialogParameters
         {
             { "Message", "Please select the workspace where you want to move the chat to." },
@@ -280,6 +314,7 @@ public partial class Chat : ComponentBase, IAsyncDisposable
         this.chatThread!.WorkspaceId = workspaceId;
         await this.SaveThread();
         
+        this.currentWorkspaceId = this.chatThread.WorkspaceId;
         this.currentWorkspaceName = await this.workspaces.LoadWorkspaceName(this.chatThread.WorkspaceId);
     }
 
@@ -291,6 +326,7 @@ public partial class Chat : ComponentBase, IAsyncDisposable
         this.isStreaming = false;
         this.hasUnsavedChanges = false;
         this.userInput = string.Empty;
+        this.currentWorkspaceId = this.chatThread?.WorkspaceId ?? Guid.Empty;
         this.currentWorkspaceName = this.chatThread is null ? string.Empty : await this.workspaces.LoadWorkspaceName(this.chatThread.WorkspaceId);
         
         await this.inputField.Clear();