Added support for images (#609)

app/MindWork AI Studio/Components/AttachDocuments.razor.cs

@@ -36,6 +36,9 @@ public partial class AttachDocuments : MSGComponentBase
     [Parameter]
     public bool UseSmallForm { get; set; }
     
+    [Parameter]
+    public AIStudio.Settings.Provider? Provider { get; set; }
+    
     [Inject]
     private ILogger<AttachDocuments> Logger { get; set; } = null!;
     
@@ -114,7 +117,7 @@ public partial class AttachDocuments : MSGComponentBase
 
                 foreach (var path in paths)
                 {
-                    if(!await FileExtensionValidation.IsExtensionValidWithNotifyAsync(path))
+                    if(!await FileExtensionValidation.IsExtensionValidWithNotifyAsync(FileExtensionValidation.UseCase.ATTACHING_CONTENT, path, this.Provider))
                         continue;
 
                     this.DocumentPaths.Add(FileAttachment.FromPath(path));
@@ -158,7 +161,7 @@ public partial class AttachDocuments : MSGComponentBase
             if (!File.Exists(selectedFilePath))
                 continue;
 
-            if (!await FileExtensionValidation.IsExtensionValidWithNotifyAsync(selectedFilePath))
+            if (!await FileExtensionValidation.IsExtensionValidWithNotifyAsync(FileExtensionValidation.UseCase.ATTACHING_CONTENT, selectedFilePath, this.Provider))
                 continue;
 
             this.DocumentPaths.Add(FileAttachment.FromPath(selectedFilePath));
@@ -216,7 +219,7 @@ public partial class AttachDocuments : MSGComponentBase
     {
         var dialogParameters = new DialogParameters<DocumentCheckDialog>
         {
-            { x => x.FilePath, fileAttachment.FilePath },
+            { x => x.Document, fileAttachment },
         };
 
         await this.DialogService.ShowAsync<DocumentCheckDialog>(T("Document Preview"), dialogParameters, DialogOptions.FULLSCREEN);

app/MindWork AI Studio/Components/ChatComponent.razor

@@ -83,7 +83,7 @@
             
             <ChatTemplateSelection CanChatThreadBeUsedForTemplate="@this.CanThreadBeSaved" CurrentChatThread="@this.ChatThread" CurrentChatTemplate="@this.currentChatTemplate" CurrentChatTemplateChanged="@this.ChatTemplateWasChanged"/>
 
-            <AttachDocuments Name="File Attachments" @bind-DocumentPaths="@this.chatDocumentPaths" CatchAllDocuments="true" UseSmallForm="true"/>
+            <AttachDocuments Name="File Attachments" @bind-DocumentPaths="@this.chatDocumentPaths" CatchAllDocuments="true" UseSmallForm="true" Provider="@this.Provider"/>
 
             @if (this.SettingsManager.ConfigurationData.Workspace.StorageBehavior is WorkspaceStorageBehavior.STORE_CHATS_AUTOMATICALLY)
             {

app/MindWork AI Studio/Components/ChatComponent.razor.cs

@@ -933,10 +933,17 @@ public partial class ChatComponent : MSGComponentBase, IAsyncDisposable
 
         if (this.cancellationTokenSource is not null)
         {
-            if(!this.cancellationTokenSource.IsCancellationRequested)
-                await this.cancellationTokenSource.CancelAsync();
+            try
+            {
+                if(!this.cancellationTokenSource.IsCancellationRequested)
+                    await this.cancellationTokenSource.CancelAsync();
             
-            this.cancellationTokenSource.Dispose();
+                this.cancellationTokenSource.Dispose();
+            }
+            catch
+            {
+                // ignored
+            }
         }
     }
 

app/MindWork AI Studio/Components/ReadFileContent.razor.cs

@@ -55,7 +55,7 @@ public partial class ReadFileContent : MSGComponentBase
             return;
         }
 
-        if (!await FileExtensionValidation.IsExtensionValidWithNotifyAsync(selectedFile.SelectedFilePath))
+        if (!await FileExtensionValidation.IsExtensionValidWithNotifyAsync(FileExtensionValidation.UseCase.DIRECTLY_LOADING_CONTENT, selectedFile.SelectedFilePath))
         {
             this.Logger.LogWarning("User attempted to load unsupported file: {FilePath}", selectedFile.SelectedFilePath);
             return;