fix MemoryStream leaks + multi-attachments issue

Author: Madzionator

src/MaIN.InferPage/Components/Pages/Home.razor

@@ -199,7 +199,7 @@
     private List<MessageExt> Messages { get; set; } = new();
     private ElementReference? _bottomElement;
     private ElementReference _editorRef;
-    private List<IBrowserFile> _selectedFiles = new();
+    private List<FileInfo> _selectedFiles = new();
     private int _inputKey;
 
     private readonly MarkdownPipeline _markdownPipeline = new MarkdownPipelineBuilder()
@@ -271,17 +271,34 @@
         }
     }
 
-    private void HandleFileSelected(InputFileChangeEventArgs e)
+    private async Task HandleFileSelected(InputFileChangeEventArgs e)
     {
         foreach (var file in e.GetMultipleFiles(10))
         {
-            _selectedFiles.Add(file);
+            var ms = new MemoryStream();
+            try
+            {
+                await file.OpenReadStream(20 * 1024 * 1024).CopyToAsync(ms);
+                ms.Position = 0;
+                _selectedFiles.Add(new FileInfo
+                {
+                    Name = file.Name,
+                    Extension = Path.GetExtension(file.Name),
+                    StreamContent = ms
+                });
+            }
+            catch (Exception ex)
+            {
+                await ms.DisposeAsync();
+                _errorMessage = $"Failed to read file {file.Name}: {ex.Message}";
+            }
         }
         StateHasChanged();
     }
 
-    private void RemoveFile(IBrowserFile file)
+    private void RemoveFile(FileInfo file)
     {
+        file.StreamContent?.Dispose();
         _selectedFiles.Remove(file);
     }
 
@@ -304,29 +321,9 @@
         _isLoading = true;
         StateHasChanged();
 
+        var attachments = new List<FileInfo>(_selectedFiles);
         try
         {
-            var attachments = new List<FileInfo>();
-            foreach (var file in _selectedFiles)
-            {
-                try
-                {
-                    var ms = new MemoryStream();
-                    await file.OpenReadStream(20 * 1024 * 1024).CopyToAsync(ms, cancellationToken);
-                    ms.Position = 0;
-                    attachments.Add(new FileInfo
-                    {
-                        Name = file.Name,
-                        Extension = Path.GetExtension(file.Name),
-                        StreamContent = ms
-                    });
-                }
-                catch (Exception ex)
-                {
-                    _errorMessage = $"Failed to read file {file.Name}: {ex.Message}";
-                }
-            }
-
             _selectedFiles.Clear();
             _inputKey++;
             StateHasChanged();
@@ -445,6 +442,10 @@
         }
         finally
         {
+            foreach (var attachment in attachments)
+                attachment.StreamContent?.Dispose();
+            attachments.Clear();
+
             _isLoading = false;
             _isThinking = false;
             StateHasChanged();

src/MaIN.Services/Services/LLMService/DeepSeekService.cs

@@ -57,7 +57,7 @@ protected override void ValidateApiKey()
         CancellationToken cancellationToken = default)
     {
         var lastMsg = chat.Messages.Last();
-        var filePaths = await DocumentProcessor.ConvertToFilesContent(memoryOptions);
+        var filePaths = await DocumentProcessor.ConvertToFilesContent(memoryOptions, cancellationToken);
         var message = new Message()
         {
             Role = ServiceConstants.Roles.User,

src/MaIN.Services/Services/LLMService/GroqCloudService.cs

@@ -51,7 +51,7 @@ protected override void ValidateApiKey()
         CancellationToken cancellationToken = default)
     {
         var lastMsg = chat.Messages.Last();
-        var filePaths = await DocumentProcessor.ConvertToFilesContent(memoryOptions);
+        var filePaths = await DocumentProcessor.ConvertToFilesContent(memoryOptions, cancellationToken);
         var message = new Message()
         {
             Role = ServiceConstants.Roles.User,

src/MaIN.Services/Services/LLMService/Memory/DocumentProcessor.cs

@@ -31,7 +31,7 @@ public static string ProcessDocument(string filePath)
         };
     }
 
-    public static async Task<string[]> ConvertToFilesContent(ChatMemoryOptions options)
+    public static async Task<string[]> ConvertToFilesContent(ChatMemoryOptions options, CancellationToken cancellationToken = default)
     {
         var files = new List<string>();
         foreach (var fData in options.FilesData)
@@ -43,14 +43,14 @@ public static async Task<string[]> ConvertToFilesContent(ChatMemoryOptions optio
         {
             var path = Path.Combine(Path.GetTempPath(), sData.Key);
             await using var fileStream = new FileStream(path, FileMode.Create, FileAccess.Write);
-            await sData.Value.CopyToAsync(fileStream);
+            await sData.Value.CopyToAsync(fileStream, cancellationToken);
             files.Add(path);
         }
 
         foreach (var txt in options.TextData)
         {
             var path = Path.Combine(Path.GetTempPath(), $"{txt.Key}.txt");
-            await File.WriteAllTextAsync(path, txt.Value);
+            await File.WriteAllTextAsync(path, txt.Value, cancellationToken);
             files.Add(path);
         }
 
@@ -60,8 +60,8 @@ public static async Task<string[]> ConvertToFilesContent(ChatMemoryOptions optio
             foreach (var web in options.WebUrls)
             {
                 var path = Path.Combine(Path.GetTempPath(), $"{Guid.NewGuid()}.html");
-                var html = await client.GetStringAsync(web);
-                await File.WriteAllTextAsync(path, html);
+                var html = await client.GetStringAsync(web, cancellationToken);
+                await File.WriteAllTextAsync(path, html, cancellationToken);
                 files.Add(path);
             }
         }

src/MaIN.Services/Services/LLMService/XaiService.cs

@@ -50,7 +50,7 @@ protected override void ValidateApiKey()
         CancellationToken cancellationToken = default)
     {
         var lastMsg = chat.Messages.Last();
-        var filePaths = await DocumentProcessor.ConvertToFilesContent(memoryOptions);
+        var filePaths = await DocumentProcessor.ConvertToFilesContent(memoryOptions, cancellationToken);
         var message = new Message()
         {
             Role = ServiceConstants.Roles.User,