feat: InferPage visual & functional improvements

Examples/Examples/Chat/ChatWithImageGenExample.cs

@@ -1,5 +1,6 @@
 using Examples.Utils;
 using MaIN.Core.Hub;
+using MaIN.Domain.Models.Abstract;
 
 namespace Examples.Chat;
 
@@ -10,7 +11,7 @@ public async Task Start()
         Console.WriteLine("ChatExample with image gen is running!");
 
         var result = await AIHub.Chat()
-            .EnableVisual()
+            .WithModel(new GenericLocalModel("FLUX.1_Shnell"), imageGen: true)
             .WithMessage("Generate cyberpunk godzilla cat warrior")
             .CompleteAsync();
 

Examples/Examples/Chat/ChatWithImageGenGeminiExample.cs

@@ -1,5 +1,7 @@
 using Examples.Utils;
 using MaIN.Core.Hub;
+using MaIN.Domain.Configuration;
+using MaIN.Domain.Models.Abstract;
 
 namespace Examples.Chat;
 
@@ -11,7 +13,7 @@ public async Task Start()
         GeminiExample.Setup(); // We need to provide Gemini API key
 
         var result = await AIHub.Chat()
-            .EnableVisual()
+            .WithModel(new GenericCloudModel("imagen-3", BackendType.Gemini), imageGen: true)
             .WithMessage("Generate hamster as a astronaut on the moon")
             .CompleteAsync();
 

Examples/Examples/Chat/ChatWithImageGenOpenAiExample.cs

@@ -13,7 +13,6 @@ public async Task Start()
 
         var result = await AIHub.Chat()
             .WithModel<DallE3>()
-            .EnableVisual()
             .WithMessage("Generate rock style cow playing guitar")
             .CompleteAsync();
 

MaIN.Core.IntegrationTests/ChatTests.cs

@@ -2,6 +2,7 @@
 using MaIN.Core.Hub;
 using MaIN.Core.IntegrationTests.Helpers;
 using MaIN.Domain.Entities;
+using MaIN.Domain.Models.Abstract;
 using MaIN.Domain.Models.Concrete;
 
 namespace MaIN.Core.IntegrationTests;
@@ -97,7 +98,7 @@ public async Task Should_GenerateImage_BasedOnPrompt()
         const string extension = "png";
 
         var result = await AIHub.Chat()
-            .EnableVisual()
+            .WithModel(new GenericLocalModel("FLUX.1_Shnell"), imageGen: true)
             .WithMessage("Generate cat in Rome. Sightseeing, colloseum, ancient builidngs, Italy.")
             .CompleteAsync();
 

Releases/0.10.0.md

@@ -0,0 +1,7 @@
+# 0.10.0 release
+
+Improve InferPage:
+- Refreshed chat UI layout with improved theming and smarter scroll behavior
+- Extended attachments (drag & drop, paste), image previews, and improved image generation
+- Added support for unregistered models and vision-based image handling (no OCR)
+- Stability fixes, proper cancellation support, and internal service refactoring

src/MaIN.Core.UnitTests/ChatContextTests.cs

@@ -88,7 +88,7 @@ public async Task CompleteAsync_ShouldCallChatService()
         };
 
 
-        _mockChatService.Setup(s => s.Completions(It.IsAny<Chat>(), It.IsAny<bool>(), It.IsAny<bool>(), null))
+        _mockChatService.Setup(s => s.Completions(It.IsAny<Chat>(), It.IsAny<bool>(), It.IsAny<bool>(), null, It.IsAny<CancellationToken>()))
             .ReturnsAsync(chatResult);
 
         _chatContext.WithMessage("User message");
@@ -98,7 +98,7 @@ public async Task CompleteAsync_ShouldCallChatService()
         var result = await _chatContext.CompleteAsync();
 
         // Assert
-        _mockChatService.Verify(s => s.Completions(It.IsAny<Chat>(), false, false, null), Times.Once);
+        _mockChatService.Verify(s => s.Completions(It.IsAny<Chat>(), false, false, null, It.IsAny<CancellationToken>()), Times.Once);
         Assert.Equal(chatResult, result);
     }
 
@@ -128,6 +128,6 @@ await _chatContext.WithModel(model)
             .CompleteAsync();
 
         // Assert
-        _mockChatService.Verify(s => s.Completions(It.Is<Chat>(c => c.ModelId == _testModelId && c.ModelInstance == model), It.IsAny<bool>(), It.IsAny<bool>(), null), Times.Once);
+        _mockChatService.Verify(s => s.Completions(It.Is<Chat>(c => c.ModelId == _testModelId && c.ModelInstance == model), It.IsAny<bool>(), It.IsAny<bool>(), null, It.IsAny<CancellationToken>()), Times.Once);
     }
 }

src/MaIN.Core/.nuspec

@@ -2,7 +2,7 @@
 <package>
   <metadata>
     <id>MaIN.NET</id>
-    <version>0.9.4</version>
+    <version>0.10.0</version>
     <authors>Wisedev</authors>
     <owners>Wisedev</owners>
     <icon>favicon.png</icon>

src/MaIN.Core/Hub/Contexts/ChatContext.cs

@@ -40,9 +40,10 @@ internal ChatContext(IChatService chatService, Chat existingChat)
         _chat = existingChat;
     }
 
-    public IChatMessageBuilder WithModel(AIModel model)
+    public IChatMessageBuilder WithModel(AIModel model, bool? imageGen = null)
     {
         SetModel(model);
+        _chat.ImageGen = imageGen ?? model is IImageGenerationModel;
         return this;
     }
 
@@ -82,12 +83,7 @@ private void SetModel(AIModel model)
         _chat.ModelId = model.Id;
         _chat.ModelInstance = model;
         _chat.Backend = model.Backend;
-    }
-
-    public IChatMessageBuilder EnableVisual()
-    {
-        _chat.Visual = true;
-        return this;
+        _chat.ImageGen = model.HasImageGeneration;
     }
 
     public IChatMessageBuilder EnsureModelDownloaded()
@@ -116,7 +112,7 @@ public IChatConfigurationBuilder WithMemoryParams(MemoryParams memoryParams)
 
     public IChatConfigurationBuilder Speak(TextToSpeechParams speechParams)
     {
-        _chat.Visual = false;
+        _chat.ImageGen = false;
         _chat.TextToSpeechParams = speechParams;
         return this;
     }
@@ -205,7 +201,8 @@ public IChatConfigurationBuilder DisableCache()
     public async Task<ChatResult> CompleteAsync(
         bool translate = false, // Move to WithTranslate
         bool interactive = false, // Move to WithInteractive
-        Func<LLMTokenValue?, Task>? changeOfValue = null)
+        Func<LLMTokenValue?, Task>? changeOfValue = null,
+        CancellationToken cancellationToken = default)
     {
         if (_chat.ModelInstance is null)
         {
@@ -231,7 +228,7 @@ public async Task<ChatResult> CompleteAsync(
         {
             await _chatService.Create(_chat);
         }
-        var result = await _chatService.Completions(_chat, translate, interactive, changeOfValue);
+        var result = await _chatService.Completions(_chat, translate, interactive, changeOfValue, cancellationToken);
         _files = [];
         return result;
     }

src/MaIN.Core/Hub/Contexts/Interfaces/ChatContext/IChatBuilderEntryPoint.cs

@@ -1,4 +1,4 @@
-namespace MaIN.Core.Hub.Contexts.Interfaces.ChatContext;
+namespace MaIN.Core.Hub.Contexts.Interfaces.ChatContext;
 
 public interface IChatBuilderEntryPoint : IChatActions
 {
@@ -9,7 +9,7 @@ public interface IChatBuilderEntryPoint : IChatActions
     /// <param name="model">The name of the AI model to be used.</param>
     /// <returns>The context instance implementing <see cref="IChatMessageBuilder"/> for method chaining.</returns>
     IChatMessageBuilder WithModel(string model);
-    
+
     /// <summary>
     /// Configures a custom model with a specific path and project context.
     /// </summary>
@@ -18,18 +18,11 @@ public interface IChatBuilderEntryPoint : IChatActions
     /// <param name="mmProject">Optional multi-modal project identifier.</param>
     /// <returns>The context instance implementing <see cref="IChatMessageBuilder"/> for method chaining.</returns>
     IChatMessageBuilder WithCustomModel(string model, string path, string? mmProject = null);
-
-    /// <summary>
-    /// Enables visual/image generation mode. Use this method now if you do not plan to explicitly define the model.
-    /// Otherwise, you will be able to use this method in the next step, after defining the model.
-    /// </summary>
-    /// <returns>The context instance implementing <see cref="IChatMessageBuilder"/> for method chaining.</returns>
-    IChatMessageBuilder EnableVisual();
-
+
     /// <summary>
     /// Loads an existing chat session from the database using its unique identifier.
     /// </summary>
     /// <param name="chatId">The GUID of the existing chat.</param>
     /// <returns>The context instance implementing <see cref="IChatConfigurationBuilder"/> for method chaining.</returns>
     Task<IChatConfigurationBuilder> FromExisting(string chatId);
-}
+}

src/MaIN.Core/Hub/Contexts/Interfaces/ChatContext/IChatConfigurationBuilder.cs

@@ -104,5 +104,5 @@ public interface IChatConfigurationBuilder : IChatActions
     /// <param name="interactive">A flag indicating whether the chat session should be interactive. Default is false.</param>
     /// <param name="changeOfValue">An optional callback invoked whenever a new token or update is received during streaming.</param>
     /// <returns>A <see cref="ChatResult"/> object containing the result of the completed chat session.</returns>
-    Task<ChatResult> CompleteAsync(bool translate = false, bool interactive = false, Func<LLMTokenValue?, Task>? changeOfValue = null);
+    Task<ChatResult> CompleteAsync(bool translate = false, bool interactive = false, Func<LLMTokenValue?, Task>? changeOfValue = null, CancellationToken cancellationToken = default);
 }

src/MaIN.Core/Hub/Contexts/Interfaces/ChatContext/IChatMessageBuilder.cs

@@ -4,21 +4,14 @@ namespace MaIN.Core.Hub.Contexts.Interfaces.ChatContext;
 
 public interface IChatMessageBuilder : IChatActions
 {
-    /// <summary>
-    /// Enables the visual output for the current chat session. This flag allows the AI to generate and return visual content,
-    /// such as images or charts, as part of its response.
-    /// </summary>
-    /// <returns>The context instance implementing <see cref="IChatMessageBuilder"/> for method chaining.</returns>
-    IChatMessageBuilder EnableVisual();
-
     /// <summary>
     /// Flags the chat to automatically ensure the selected local model is downloaded before completing.
     /// If the model is already present the download is skipped; cloud models are silently ignored.
     /// The actual download is deferred until <see cref="IChatConfigurationBuilder.CompleteAsync"/> is called.
     /// </summary>
     /// <returns>The context instance implementing <see cref="IChatMessageBuilder"/> for method chaining.</returns>
     IChatMessageBuilder EnsureModelDownloaded();
-    
+
     /// <summary>
     /// Adds a user message to the chat. This method captures the message content and assigns the "User" role to it.
     /// It also timestamps the message for proper ordering.

src/MaIN.Domain/Entities/Chat.cs

@@ -9,29 +9,35 @@ public class Chat
 {
     public string Id { get; init; } = string.Empty;
     public required string Name { get; init; }
+    private string? _modelId;
     public required string ModelId
     {
-        get => _modelInstance?.Id ?? string.Empty;
+        get => _modelInstance?.Id ?? _modelId ?? string.Empty;
         set
         {
+            _modelId = value;
             if (string.IsNullOrEmpty(value))
             {
                 _modelInstance = null;
                 return;
             }
 
-            _modelInstance = ModelRegistry.GetById(value);
+            ModelRegistry.TryGetById(value, out _modelInstance);
         }
     }
     private AIModel? _modelInstance;
     public AIModel? ModelInstance
     {
         get => _modelInstance;
-        set => (_modelInstance, ModelId) = (value, value?.Id ?? string.Empty);
+        set
+        {
+            _modelInstance = value;
+            _modelId = value?.Id ?? string.Empty;
+        }
     }
     public List<Message> Messages { get; set; } = [];
     public ChatType Type { get; set; } = ChatType.Conversation;
-    public bool Visual { get; set; }
+    public bool ImageGen { get; set; }
     public InferenceParams InterferenceParams { get; set; } = new();
     public MemoryParams MemoryParams { get; set; } = new();
     public ToolsConfiguration? ToolsConfiguration { get; set; }

src/MaIN.Domain/Entities/Message.cs

@@ -19,7 +19,19 @@ public Message()
     public List<LLMTokenValue> Tokens { get; set; } = [];
     public bool Tool { get; init; }
     public DateTime Time { get; set; }
-    public byte[]? Image { get; init; }
+    public List<byte[]>? Images { get; set; }
+
+    // Backward-compat wrapper – single image access
+    public byte[]? Image
+    {
+        get => Images?.Count > 0 ? Images[0] : null;
+        set
+        {
+            if (value == null) Images = null;
+            else Images = [value];
+        }
+    }
+
     public byte[]? Speech { get; set; }
     public List<FileInfo>? Files { get; set; } 
     public Dictionary<string, string> Properties { get; set; } = [];

src/MaIN.Domain/Models/Abstract/AIModel.cs

@@ -34,6 +34,9 @@ public abstract record AIModel(
 
     /// <summary> Checks if model supports vision/image input. </summary>
     public bool HasVision => this is IVisionModel;
+
+    /// <summary> Checks if model generates images from text prompts. </summary>
+    public bool HasImageGeneration => this is IImageGenerationModel;
 }
 
 /// <summary> Base class for local models. </summary>

src/MaIN.Domain/Models/Abstract/IModelCapabilities.cs

@@ -45,3 +45,8 @@ public interface IEmbeddingModel
 /// Interface for models that support text-to-speech.
 /// </summary>
 public interface ITTSModel;
+
+/// <summary>
+/// Interface for models that generate images from text prompts.
+/// </summary>
+public interface IImageGenerationModel;