final

Author: dkalinowski

src/image_gen/http_image_gen_calculator.cc

@@ -30,22 +30,11 @@
 
 #include "pipelines.hpp"
 
-static void save_png_to_disk(const std::string& png_data, const std::string& filename) {
-    std::ofstream out_file(filename, std::ios::binary);
-    if (!out_file) {
-        throw std::runtime_error("Failed to open file for writing: " + filename);
-    }
-    
-    out_file.write(png_data.data(), png_data.size());
-    
-    if (!out_file.good()) {
-        out_file.close();
-        std::remove(filename.c_str()); // Clean up partial file
-        throw std::runtime_error("Failed to write data to file: " + filename);
-    }
-    
-    out_file.close();
-}
+#pragma warning(push)
+#pragma warning(disable : 6001 4324 6385 6386)
+#include "absl/strings/escaping.h"
+#include "absl/strings/str_cat.h"
+#pragma warning(pop)
 
 using namespace ovms;
 
@@ -115,7 +104,17 @@ class ImageGenCalculator : public CalculatorBase {
                 ov::genai::num_images_per_prompt(1)});  // todo: get from req
 
         std::string res = save_image_stbi(image);
-        save_png_to_disk(res, "output.png");
+
+        // Convert the image to a base64 string
+        std::string base64_image;
+        absl::Base64Escape(res, &base64_image);
+
+        // Create the JSON response
+        std::string json_response = absl::StrCat("{\"data\":[{\"b64_json\":\"", base64_image, "\"}]}");
+        // Produce std::string packet
+        auto output = absl::make_unique<std::string>(json_response);
+        cc->Outputs().Tag(OUTPUT_TAG_NAME).Add(output.release(), cc->InputTimestamp());
+
         SPDLOG_LOGGER_DEBUG(llm_calculator_logger, "ImageGenCalculator  [Node: {}] Process end", cc->NodeName());
         return absl::OkStatus();
     }

src/llm/apis/openai_completions.cpp

@@ -89,43 +89,6 @@ absl::Status OpenAIChatCompletionsHandler::parseCompletionsPart() {
     return absl::OkStatus();
 }
 
-// ov::Tensor load_image_stbi(const std::string& imageBytes) {
-//     int x = 0, y = 0, channelsInFile = 0;
-//     constexpr int desiredChannels = 3;
-//     unsigned char* image = stbi_load_from_memory(
-//         (const unsigned char*)imageBytes.data(), imageBytes.size(),
-//         &x, &y, &channelsInFile, desiredChannels);
-//     if (!image) {
-//         std::stringstream errorMessage;
-//         errorMessage << "Failed to load the image";
-//         throw std::runtime_error{errorMessage.str()};
-//     }
-//     struct SharedImageAllocator {
-//         unsigned char* image;
-//         int channels, height, width;
-//         void* allocate(size_t bytes, size_t) const {
-//             if (image && channels * height * width == bytes) {
-//                 return image;
-//             }
-//             throw std::runtime_error{"Unexpected number of bytes was requested to allocate."};
-//         }
-//         void deallocate(void*, size_t bytes, size_t) {
-//             if (channels * height * width != bytes) {
-//                 throw std::runtime_error{"Unexpected number of bytes was requested to deallocate."};
-//             }
-//             if (image != nullptr) {
-//                 stbi_image_free(image);
-//                 image = nullptr;
-//             }
-//         }
-//         bool is_equal(const SharedImageAllocator& other) const noexcept { return this == &other; }
-//     };
-//     return ov::Tensor(
-//         ov::element::u8,
-//         ov::Shape{1, size_t(y), size_t(x), size_t(desiredChannels)},
-//         SharedImageAllocator{image, desiredChannels, y, x});
-// }
-
 absl::Status OpenAIChatCompletionsHandler::parseMessages() {
     auto it = doc.FindMember("messages");
     if (it == doc.MemberEnd())