AI-Studio/app/MindWork AI Studio/Chat/ContentImage.cs

using System.Text.Json.Serialization;

using AIStudio.Provider;
using AIStudio.Tools.Validation;

namespace AIStudio.Chat;

/// <summary>
/// Represents an image inside the chat.
/// </summary>
public sealed class ContentImage : IContent, IImageSource
{
    #region Implementation of IContent

    /// <inheritdoc />
    [JsonIgnore]
    public bool InitialRemoteWait { get; set; }

    /// <inheritdoc />
    [JsonIgnore]
    public bool IsStreaming { get; set; }

    /// <inheritdoc />
    [JsonIgnore]
    public Func<Task> StreamingDone { get; set; } = () => Task.CompletedTask;

    /// <inheritdoc />
    [JsonIgnore]
    public Func<Task> StreamingEvent { get; set; } = () => Task.CompletedTask;

    /// <inheritdoc />
    public List<Source> Sources { get; set; } = [];

    /// <inheritdoc />
    public List<FileAttachment> FileAttachments { get; set; } = [];

    /// <inheritdoc />
    public Task<ChatThread> CreateFromProviderAsync(IProvider provider, Model chatModel, IContent? lastUserPrompt, ChatThread? chatChatThread, CancellationToken token = default)
    {
        throw new NotImplementedException();
    }
    
    /// <inheritdoc />
    public IContent DeepClone() => new ContentImage
    {
        Source = this.Source,
        InitialRemoteWait = this.InitialRemoteWait,
        IsStreaming = this.IsStreaming,
        SourceType = this.SourceType,
        Sources = [..this.Sources],
        FileAttachments = [..this.FileAttachments],
    };

    #endregion

    /// <summary>
    /// Creates a ContentImage from a local file path.
    /// </summary>
    /// <param name="filePath">The path to the image file.</param>
    /// <returns>A new ContentImage instance if the file is valid, null otherwise.</returns>
    public static async Task<ContentImage?> CreateFromFileAsync(string filePath)
    {
        if (!await FileExtensionValidation.IsImageExtensionValidWithNotifyAsync(filePath))
            return null;

        return new ContentImage
        {
            SourceType = ContentImageSource.LOCAL_PATH,
            Source = filePath,
        };
    }

    /// <summary>
    /// The type of the image source.
    /// </summary>
    /// <remarks>
    /// Is the image source a URL, a local file path, a base64 string, etc.?
    /// </remarks>
    public required ContentImageSource SourceType { get; init; }

    /// <summary>
    /// The image source.
    /// </summary>
    public required string Source { get; set; }
}