using System;

using System.ComponentModel;

using System.Collections.Generic;

using System.IO;

using System.Threading;

using Azure;

using Azure.Storage.Blobs;

using Azure.Storage.Blobs.Models;

​

// (josar):

// OPTION 2: Separate properties for each model type

//

// This is similar to OPTION 1, but the assumption it makes is a little different:

// - In a way, OPTION 1 assumes that the union of the StorageBlobImageData and Base64ImageData models is the common scenario, i.e. it assumes that they will almost always be used together as a union.

// - On the other hand, this other option assumes that they are independent, and that using them as a union is the exceptional case. Therefore, the responsibility of making them behave as a union falls on the consumer classes (which in this case is the ImageGenerations class).

//

// Given that currently we have two uses cases for these models (one in OpenAI and another one in Content Safety), and that in both cases they are used together as a union, I still favor OPTION 1.

public partial class StorageBlobImageData

{

    public Uri Url { get; set; }    

}

​

public partial class Base64ImageData

{

    public BinaryData Base64Json { get; set; }  

}

​

public partial class ImageGenerations

{

    public IReadOnlyList<StorageBlobImageData> StorageBlobData { get; }

    public IReadOnlyList<Base64ImageData> Base64Data { get; }

    // Other properties here...

}

​

// Here is an example of how to use it:

public class Program

{

    public static void Main()

    {

        UseUrl();

        UseBase64Json();

    }

    // Example: Use the storage blob URL to save the image as a file.

    public static void UseUrl()

    {

        // Request that the images be returned as URLs.

        ImageGenerationOptions options = new ImageGenerationOptions();

        options.ResposeFormat = ImageGenerationResponseFormat.Url;  

        OpenAIClient client = new OpenAIClient();

        ImageGenerations result = client.GenerateImages(options);

        foreach (var generatedImage in result.StorageBlobData)

        {

            // Use the BlobClient from Azure.Storage.Blobs to download the image

            // as BinaryData and save it to a file.

            BlobClient blobClient = new BlobClient(generatedImage.Url);

            BlobDownloadResult download = blobClient.DownloadContent();

            SaveToFile(download.Content.ToStream());    

        }

    }

    // Example: Use the base64 JSON representation to save the image as a file.

    public static void UseBase64Json()

    {

        // Request that the images be returned as base64 encoded PNGs.

        ImageGenerationOptions options = new ImageGenerationOptions();

        options.ResposeFormat = ImageGenerationResponseFormat.Base64;

        OpenAIClient client = new OpenAIClient();

        ImageGenerations result = client.GenerateImages(options);

        foreach (Base64ImageData generatedImage in result.Base64Data)

        {

            // Save the image to file.

            SaveToFile(generatedImage.Base64Json.ToStream());   

        }

    }

    public static void SaveToFile(Stream stream)

    {

        using (var fileStream = File.Create("MyImage.png"))

        {

            stream.Seek(0, SeekOrigin.Begin);

            stream.CopyTo(fileStream);

        }

    }

}

​

/* ***************************************************************** */

​

# region Ignore

​

public partial class OpenAIClient

{

    public virtual Response<ImageGenerations> GenerateImages(

        ImageGenerationOptions imageGenerationOptions, 

        CancellationToken cancellationToken = default)

    {

        throw new NotImplementedException();

    }

    // Other methods and properties here...

}

​

public partial class ImageGenerationOptions

{   

    public ImageGenerationResponseFormat? ResposeFormat { get; set; }

    // Other properties here...

}

​

public readonly partial struct ImageGenerationResponseFormat : IEquatable<ImageGenerationResponseFormat>

{

    private readonly string _value;

​

    /// <summary> Initializes a new instance of <see cref="ImageGenerationResponseFormat"/>. </summary>

    /// <exception cref="ArgumentNullException"> <paramref name="value"/> is null. </exception>

    public ImageGenerationResponseFormat(string value)

    {

        _value = value ?? throw new ArgumentNullException(nameof(value));

    }

​

    private const string UrlValue = "url";

    private const string Base64Value = "b64_json";

​

    /// <summary> Image generation response items should provide a URL from which the image may be retrieved. </summary>

    public static ImageGenerationResponseFormat Url { get; } = new ImageGenerationResponseFormat(UrlValue);

    /// <summary> Image generation response items should provide image data as a base64-encoded string. </summary>

    public static ImageGenerationResponseFormat Base64 { get; } = new ImageGenerationResponseFormat(Base64Value);

    /// <summary> Determines if two <see cref="ImageGenerationResponseFormat"/> values are the same. </summary>

    public static bool operator ==(ImageGenerationResponseFormat left, ImageGenerationResponseFormat right) => left.Equals(right);

    /// <summary> Determines if two <see cref="ImageGenerationResponseFormat"/> values are not the same. </summary>

    public static bool operator !=(ImageGenerationResponseFormat left, ImageGenerationResponseFormat right) => !left.Equals(right);

    /// <summary> Converts a string to a <see cref="ImageGenerationResponseFormat"/>. </summary>

    public static implicit operator ImageGenerationResponseFormat(string value) => new ImageGenerationResponseFormat(value);

​

    /// <inheritdoc />

    [EditorBrowsable(EditorBrowsableState.Never)]

    public override bool Equals(object obj) => obj is ImageGenerationResponseFormat other && Equals(other);

    /// <inheritdoc />

    public bool Equals(ImageGenerationResponseFormat other) => string.Equals(_value, other._value, StringComparison.InvariantCultureIgnoreCase);

​

    /// <inheritdoc />

    [EditorBrowsable(EditorBrowsableState.Never)]

    public override int GetHashCode() => _value?.GetHashCode() ?? 0;

    /// <inheritdoc />

    public override string ToString() => _value;

}

​

# endregion