Add batch console example

Author: saddam213

OnnxStack.Console/Examples/StableDiffusionBatch.cs

@@ -0,0 +1,92 @@
+using OnnxStack.Core;
+using OnnxStack.StableDiffusion.Common;
+using OnnxStack.StableDiffusion.Config;
+using OnnxStack.StableDiffusion.Enums;
+using OnnxStack.StableDiffusion.Helpers;
+using SixLabors.ImageSharp;
+
+namespace OnnxStack.Console.Runner
+{
+    public sealed class StableDiffusionBatch : IExampleRunner
+    {
+        private readonly string _outputDirectory;
+        private readonly IStableDiffusionService _stableDiffusionService;
+
+        public StableDiffusionBatch(IStableDiffusionService stableDiffusionService)
+        {
+            _stableDiffusionService = stableDiffusionService;
+            _outputDirectory = Path.Combine(Directory.GetCurrentDirectory(), "Examples", nameof(StableDiffusionBatch));
+        }
+
+        public string Name => "Stable Diffusion Batch Demo";
+
+        public string Description => "Creates a batch of images from the text prompt provided using all Scheduler types";
+
+        /// <summary>
+        /// Stable Diffusion Demo
+        /// </summary>
+        public async Task RunAsync()
+        {
+            Directory.CreateDirectory(_outputDirectory);
+
+            var model = _stableDiffusionService.Models.First();
+            await _stableDiffusionService.LoadModel(model);
+
+            while (true)
+            {
+                OutputHelpers.WriteConsole("Please type a prompt and press ENTER", ConsoleColor.Yellow);
+                var prompt = OutputHelpers.ReadConsole(ConsoleColor.Cyan);
+
+                OutputHelpers.WriteConsole("Please type a negative prompt and press ENTER (optional)", ConsoleColor.Yellow);
+                var negativePrompt = OutputHelpers.ReadConsole(ConsoleColor.Cyan);
+
+                OutputHelpers.WriteConsole("Please enter a batch count and press ENTER", ConsoleColor.Yellow);
+                var batch = OutputHelpers.ReadConsole(ConsoleColor.Cyan);
+                int.TryParse(batch, out var batchCount);
+                batchCount = Math.Max(1, batchCount);
+
+                var promptOptions = new PromptOptions
+                {
+                    Prompt = prompt,
+                    NegativePrompt = negativePrompt,
+                    BatchCount = batchCount
+                };
+
+                var schedulerOptions = new SchedulerOptions
+                {
+                    Seed = Random.Shared.Next(),
+
+                    GuidanceScale = 8,
+                    InferenceSteps = 22,
+                    Strength = 0.6f
+                };
+                foreach (var schedulerType in Enum.GetValues<SchedulerType>())
+                {
+                    promptOptions.SchedulerType = schedulerType;
+
+                    OutputHelpers.WriteConsole("Generating Image...", ConsoleColor.Green);
+                    await GenerateImage(model, promptOptions, schedulerOptions);
+                }
+            }
+        }
+
+        private async Task<bool> GenerateImage(ModelOptions model, PromptOptions prompt, SchedulerOptions options)
+        {
+
+            var result = await _stableDiffusionService.GenerateAsync(model, prompt, options);
+            if (result == null)
+                return false;
+
+            var imageTensors = result.Split(prompt.BatchCount);
+            for (int i = 0; i < imageTensors.Length; i++)
+            {
+                var outputFilename = Path.Combine(_outputDirectory, $"{options.Seed}_{prompt.SchedulerType}_{i}.png");
+                var image = imageTensors[i].ToImage();
+                await image.SaveAsPngAsync(outputFilename);
+                OutputHelpers.WriteConsole($"{prompt.SchedulerType} Image Created: {Path.GetFileName(outputFilename)}", ConsoleColor.Green);
+            }
+
+            return true;
+        }
+    }
+}

OnnxStack.StableDiffusion/Helpers/ImageHelpers.cs

@@ -8,7 +8,7 @@
 
 namespace OnnxStack.StableDiffusion.Helpers
 {
-    internal static class ImageHelpers
+    public static class ImageHelpers
     {
         /// <summary>
         /// Converts to image.