Module: SmartPrompt::ImagePersistence

Included in:
SenseNovaAdapter, SiliconFlowAdapter, ZhipuAIAdapter
Defined in:
lib/smart_prompt/concerns/image_persistence.rb

Overview

Shared image-saving logic for adapters that produce generated images (ZhipuAI, SenseNova, SiliconFlow). Each previously carried a byte-identical copy of save_image / save_single_image; this concern is the single source.

Adapters override two hooks:

* default_image_prefix — filename prefix when the caller passes none
  (e.g. "zhipu_image", "sensenova_image", "siliconflow_image")
* provider_label       — human label for the "Saved N <label> image(s)" log line

Instance Method Summary collapse

Instance Method Details

#default_image_prefix ⇒ Object

—- hooks (override in adapter) —————————————–



54
55
56
 
# File 'lib/smart_prompt/concerns/image_persistence.rb', line 54

def default_image_prefix
  "image"
end

#provider_label ⇒ Object 



58
59
60
 
# File 'lib/smart_prompt/concerns/image_persistence.rb', line 58

def provider_label
  "Adapter"
end

#save_image(image_data, output_dir = "./output", filename_prefix = nil) ⇒ Object

Save one or many generated images to disk. Accepts the Array returned by generate_image/edit_image or a single image hash. Returns the written paths.



18
19
20
21
22
23
24
25
26
 
# File 'lib/smart_prompt/concerns/image_persistence.rb', line 18

def save_image(image_data, output_dir = "./output", filename_prefix = nil)
  FileUtils.mkdir_p(output_dir)
  images = image_data.is_a?(Array) ? image_data : [image_data]
  saved = images.each_with_index.map do |img, index|
    save_single_image(img, output_dir, "#{filename_prefix || default_image_prefix}_#{index + 1}")
  end
  SmartPrompt.logger.info "Saved #{saved.size} #{provider_label} image(s) to #{output_dir}"
  saved
end

#save_single_image(image_data, output_dir, filename) ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
 
# File 'lib/smart_prompt/concerns/image_persistence.rb', line 28

def save_single_image(image_data, output_dir, filename)
  if image_data[:b64_json]
    file_path = File.join(output_dir, "#{filename}.png")
    File.binwrite(file_path, Base64.decode64(image_data[:b64_json]))
  elsif image_data[:url]
    uri = URI.parse(image_data[:url])
    response = Net::HTTP.get_response(uri)
    raise Error, "Failed to download image from URL: #{response.code}" unless response.is_a?(Net::HTTPSuccess)

    ext = case response["content-type"]
          when "image/jpeg", "image/jpg" then "jpg"
          when "image/png"               then "png"
          when "image/gif"               then "gif"
          when "image/webp"              then "webp"
          else "png"
          end
    file_path = File.join(output_dir, "#{filename}.#{ext}")
    File.binwrite(file_path, response.body)
  else
    raise Error, "No image data available to save"
  end
  file_path
end