Handle base64 images in OpenAiImageGenService

Rename ImageGenDalleService to OpenAiImageGenService and update response handling to support OpenAI base64 (b64_json) image payloads. ProcessOpenAiResponse now returns byte[] and will decode b64_json or download from a URL; callers now receive image bytes directly. Add JsonPropertyName for b64_json, reorder/add necessary usings, and remove unused MaIN.Services.Services.LLMService.Utils imports from Gemini, OpenAi and Xai image services. These changes enable handling both base64 and URL image responses and standardize the OpenAI image service name.

Author: Madzionator

src/MaIN.Services/Services/ImageGenServices/GeminiImageGenService.cs

@@ -8,7 +8,6 @@
 using System.Text.Json.Serialization;
 using MaIN.Domain.Exceptions;
 using MaIN.Domain.Models.Concrete;
-using MaIN.Services.Services.LLMService.Utils;
 
 namespace MaIN.Services.Services.ImageGenServices;
 

…ImageGenServices/ImageGenDalleService.cs …mageGenServices/OpenAiImageGenService.cssrc/MaIN.Services/Services/ImageGenServices/ImageGenDalleService.cs renamed to src/MaIN.Services/Services/ImageGenServices/OpenAiImageGenService.cs src/MaIN.Services/Services/ImageGenServices/ImageGenDalleService.cs renamed to src/MaIN.Services/Services/ImageGenServices/OpenAiImageGenService.cs

@@ -1,13 +1,13 @@
-using System.Net.Http.Headers;
-using System.Net.Http.Json;
 using MaIN.Domain.Configuration;
 using MaIN.Domain.Entities;
 using MaIN.Domain.Exceptions;
 using MaIN.Domain.Models.Concrete;
 using MaIN.Services.Constants;
 using MaIN.Services.Services.Abstract;
-using MaIN.Services.Services.LLMService.Utils;
 using MaIN.Services.Services.Models;
+using System.Net.Http.Headers;
+using System.Net.Http.Json;
+using System.Text.Json.Serialization;
 
 namespace MaIN.Services.Services.ImageGenServices;
 
@@ -34,8 +34,8 @@ public class OpenAiImageGenService(
         };
 
         using var response = await client.PostAsJsonAsync(ServiceConstants.ApiUrls.OpenAiImageGenerations, requestBody);
-        var imageUrl = await ProcessOpenAiResponse(response);
-        byte[] imageBytes = await DownloadImageAsync(imageUrl);
+
+        byte[] imageBytes = await ProcessOpenAiResponse(response);
         return CreateChatResult(imageBytes);
     }
 
@@ -55,13 +55,22 @@ private async Task<byte[]> DownloadImageAsync(string imageUrl)
 
         return await imageResponse.Content.ReadAsByteArrayAsync();
     }
-    
-    private async Task<string> ProcessOpenAiResponse(HttpResponseMessage response)
+
+    private async Task<byte[]> ProcessOpenAiResponse(HttpResponseMessage response)
     {
         response.EnsureSuccessStatusCode();
         var responseData = await response.Content.ReadFromJsonAsync<OpenAiImageResponse>();
-        return responseData?.Data.FirstOrDefault()?.Url 
-               ?? throw new InvalidOperationException("No image URL returned from OpenAI");
+
+        var imageData = responseData?.Data.FirstOrDefault()
+                        ?? throw new InvalidOperationException("No image data returned from OpenAI");
+
+        if (!string.IsNullOrEmpty(imageData.B64Json))
+            return Convert.FromBase64String(imageData.B64Json);
+
+        if (!string.IsNullOrEmpty(imageData.Url))
+            return await DownloadImageAsync(imageData.Url);
+
+        throw new InvalidOperationException("No image URL or base64 data returned from OpenAI");
     }
 
     private static ChatResult CreateChatResult(byte[] imageBytes)
@@ -95,4 +104,6 @@ file class OpenAiImageResponse
 file class ImageData
 {
     public string Url { get; set; } = string.Empty;
+    [JsonPropertyName("b64_json")]
+    public string B64Json { get; set; } = string.Empty;
 }

src/MaIN.Services/Services/ImageGenServices/XaiImageGenService.cs

@@ -8,7 +8,6 @@
 using System.Text.Json;
 using MaIN.Domain.Exceptions;
 using MaIN.Domain.Models.Concrete;
-using MaIN.Services.Services.LLMService.Utils;
 
 namespace MaIN.Services.Services.ImageGenServices;