discourse-ai/spec/lib/modules/ai_helper/painter_spec.rb

# frozen_string_literal: true

require_relative "../../../support/openai_completions_inference_stubs"
require_relative "../../../support/stable_difussion_stubs"

RSpec.describe DiscourseAi::AiHelper::Painter do
  subject(:painter) { described_class.new }

  fab!(:user) { Fabricate(:user) }

  before do
    SiteSetting.ai_stability_api_url = "https://api.stability.dev"
    SiteSetting.ai_stability_api_key = "abc"
  end

  describe "#commission_thumbnails" do
    let(:artifacts) do
      %w[
        iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mP8z8BQDwAEhQGAhKmMIQAAAABJRU5ErkJggg==
        iVBORw0KGgoAAAANSUhEUgAAAAoAAAAKCAYAAACNMs+9AAAAFUlEQVR42mP8z8BQz0AEYBxVSF+FABJADveWkH6oAAAAAElFTkSuQmCC
        iVBORw0KGgoAAAANSUhEUgAAAAoAAAAKCAYAAACNMs+9AAAAFUlEQVR42mNk+M9Qz0AEYBxVSF+FAAhKDveksOjmAAAAAElFTkSuQmCC
        iVBORw0KGgoAAAANSUhEUgAAAAoAAAAKCAYAAACNMs+9AAAAFUlEQVR42mNkYPhfz0AEYBxVSF+FAP5FDvcfRYWgAAAAAElFTkSuQmCC
      ]
    end

    let(:raw_content) do
      "Poetry is a form of artistic expression that uses language aesthetically and rhythmically to evoke emotions and ideas."
    end

    let(:expected_image_prompt) { <<~TEXT.strip }
        Visualize a vibrant scene of an inkwell bursting, spreading colors across a blank canvas,
        embodying words in tangible forms, symbolizing the rhythm and emotion evoked by poetry, 
        under the soft glow of a full moon.
        TEXT

    it "returns 4 samples" do
      expected_prompt = [
        { role: "system", content: <<~TEXT },
      Provide me a StableDiffusion prompt to generate an image that illustrates the following post in 40 words or less, be creative.
      TEXT
        { role: "user", content: raw_content },
      ]

      OpenAiCompletionsInferenceStubs.stub_response(expected_prompt, expected_image_prompt)
      StableDiffusionStubs.new.stub_response(expected_image_prompt, artifacts)

      thumbnails = subject.commission_thumbnails(raw_content, user)
      thumbnail_urls = Upload.last(4).map(&:short_url)

      expect(thumbnails).to contain_exactly(*thumbnail_urls)
    end
  end
end