Add image links to Blueprint output file

Author: saddam213

OnnxStack.WebUI/Hubs/StableDiffusionHub.cs

@@ -14,7 +14,6 @@ public class StableDiffusionHub : Hub<IStableDiffusionClient>
         private readonly ILogger<StableDiffusionHub> _logger;
         private readonly IStableDiffusionService _stableDiffusionService;
 
-
         /// <summary>
         /// Initializes a new instance of the <see cref="StableDiffusionHub"/> class.
         /// </summary>
@@ -122,7 +121,10 @@ private async Task<StableDiffusionResult> GenerateImageToImageResult(PromptOptio
                 return new StableDiffusionResult("Failed to copy input image");
 
             //3. Generate blueprint
-            var blueprint = new ImageBlueprint(promptOptions, schedulerOptions);
+            var inputImageLink = await _fileService.CreateOutputUrl(inputImage, false);
+            var outputImageLink = await _fileService.CreateOutputUrl(outputImage, false);
+            promptOptions.InputImage = await _fileService.CreateOutputUrl(inputImageFile.Filename, false);
+            var blueprint = new ImageBlueprint(promptOptions, schedulerOptions, outputImageLink, inputImageLink);
             var bluprintFile = await _fileService.SaveBlueprintFile(blueprint, outputBlueprint);
             if (bluprintFile is null)
                 return new StableDiffusionResult("Failed to save blueprint");
@@ -159,7 +161,8 @@ private async Task<StableDiffusionResult> GenerateTextToImageResult(PromptOption
             var outputImageFile = await _fileService.UrlToPhysicalPath(outputImageUrl);
 
             //2. Generate blueprint
-            var blueprint = new ImageBlueprint(promptOptions, schedulerOptions);
+            var outputImageLink = await _fileService.CreateOutputUrl(outputImage, false);
+            var blueprint = new ImageBlueprint(promptOptions, schedulerOptions, outputImageLink);
             var bluprintFile = await _fileService.SaveBlueprintFile(blueprint, outputBlueprint);
             if (bluprintFile is null)
                 return new StableDiffusionResult("Failed to save blueprint");

OnnxStack.WebUI/Models/ImageBlueprint.cs

@@ -2,7 +2,7 @@
 
 namespace OnnxStack.WebUI.Models
 {
-    public record ImageBlueprint(PromptOptions Prompt, SchedulerOptions Options)
+    public record ImageBlueprint(PromptOptions Prompt, SchedulerOptions Options, string OutputImageUrl, string InputImageUrl = null)
     {
         public DateTime Timestamp { get; } = DateTime.UtcNow;
     }

OnnxStack.WebUI/Program.cs

@@ -1,4 +1,6 @@
 using AspNetCore.Unobtrusive.Ajax;
+using Microsoft.AspNetCore.Hosting.Server;
+using Microsoft.AspNetCore.Hosting.Server.Features;
 using OnnxStack.Core;
 using OnnxStack.Web.Hubs;
 using Services;
@@ -23,6 +25,7 @@ public static void Main(string[] args)
 
             builder.Services.AddOnnxStackStableDiffusion();
             builder.Services.AddSingleton<IFileService, FileService>();
+            builder.Services.AddSingleton((s) => s.GetService<IServer>().Features.Get<IServerAddressesFeature>());
 
             var app = builder.Build();
 

OnnxStack.WebUI/Services/FileService.cs

@@ -1,4 +1,5 @@
-using OnnxStack.Web.Models;
+using Microsoft.AspNetCore.Hosting.Server.Features;
+using OnnxStack.Web.Models;
 using OnnxStack.WebUI.Models;
 using System.Text.Json;
 using System.Text.Json.Serialization;
@@ -10,16 +11,18 @@ public class FileService : IFileService
         private readonly ILogger<FileService> _logger;
         private readonly IWebHostEnvironment _webHostEnvironment;
         private readonly JsonSerializerOptions _serializerOptions;
+        private readonly IServerAddressesFeature _serverAddressesFeature;
 
         /// <summary>
         /// Initializes a new instance of the <see cref="FileService"/> class.
         /// </summary>
         /// <param name="logger">The logger.</param>
         /// <param name="webHostEnvironment">The web host environment.</param>
-        public FileService(ILogger<FileService> logger, IWebHostEnvironment webHostEnvironment)
+        public FileService(ILogger<FileService> logger, IWebHostEnvironment webHostEnvironment, IServerAddressesFeature serverAddressesFeature)
         {
             _logger = logger;
             _webHostEnvironment = webHostEnvironment;
+            _serverAddressesFeature = serverAddressesFeature;
             _serializerOptions = new JsonSerializerOptions { WriteIndented = true, Converters = { new JsonStringEnumConverter() } };
         }
 
@@ -155,9 +158,11 @@ public Task<string> UrlToPhysicalPath(string url)
         /// <param name="folder">The folder.</param>
         /// <param name="file">The file.</param>
         /// <returns></returns>
-        public Task<string> CreateOutputUrl(string file)
+        public Task<string> CreateOutputUrl(string file, bool relative = true)
         {
-            return Task.FromResult($"/images/results/{file}");
+            return relative
+                ? Task.FromResult($"/images/results/{file}")
+                : Task.FromResult($"{GetServerUrl()}/images/results/{file}");
         }
 
 
@@ -169,6 +174,17 @@ public Task<string> CreateRandomName()
         {
             return Task.FromResult(Path.GetFileNameWithoutExtension(Path.GetRandomFileName()));
         }
+
+
+        private string GetServerUrl()
+        {
+            var address = _serverAddressesFeature.Addresses.FirstOrDefault(x=> x.StartsWith("https"))
+                       ?? _serverAddressesFeature.Addresses.FirstOrDefault();
+            if (string.IsNullOrEmpty(address))
+                return string.Empty;
+
+            return address;
+        }
     }
 
     public record FileServiceResult(string Filename, string FileUrl, string FilePath);

OnnxStack.WebUI/Services/IFileService.cs

@@ -53,6 +53,6 @@ public interface IFileService
         /// </summary>
         /// <param name="file">The file.</param>
         /// <returns></returns>
-        Task<string> CreateOutputUrl(string file);
+        Task<string> CreateOutputUrl(string file, bool relative = true);
     }
 }