generate_image 1.1.1 → 1.1.2
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/lib/generate_image/version.rb +1 -1
- data/lib/generate_image.rb +5 -8
- metadata +1 -1
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: f8b5a6646b862d0c9a5146b85f1db0e698f26d26c68452d6c72cb02860a2f944
|
4
|
+
data.tar.gz: ffd5491fd39c2f99434387f152b0a3540976055c5ada23dfee387cbb1353c3a0
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: eecd22dc2fbf78e6ff4d370597f76364286ae963a4100312eccce67ec744e719a155c4b362ce428f9f2b4f612fdbd5d27df5e5f6e69f563ecfacd23f6e96e7a4
|
7
|
+
data.tar.gz: f80686589c0f26f986fa22bc8eb69bcd3bc38090989db26783ccbc2dc2d89de689d4dfa214a765d659e79c899f6aad9ba9efd29d225fc83d5cd86243a232e625
|
data/lib/generate_image.rb
CHANGED
@@ -17,20 +17,18 @@ module GenerateImage
|
|
17
17
|
end
|
18
18
|
|
19
19
|
def generate_image(text, options = {})
|
20
|
+
unless API_KEY
|
21
|
+
raise StandardError, "API Key not set"
|
22
|
+
end
|
20
23
|
uri = URI(API_ENDPOINT)
|
21
24
|
request = Net::HTTP::Post.new(uri)
|
22
25
|
request['Content-Type'] = 'application/json'
|
23
|
-
request['Authorization'] = "Bearer #{
|
26
|
+
request['Authorization'] = "Bearer #{API_KEY}"
|
24
27
|
request.body = {
|
25
|
-
model: options[:model] || IMAGE_MODEL_NAME,
|
26
28
|
prompt: text,
|
27
|
-
|
29
|
+
n: options[:num_images] || 1,
|
28
30
|
size: options[:size] || '512x512',
|
29
31
|
response_format: options[:response_format] || 'url',
|
30
|
-
style: options[:style] || nil,
|
31
|
-
scale: options[:scale] || 1,
|
32
|
-
seed: options[:seed] || nil,
|
33
|
-
quality: options[:quality] || 80
|
34
32
|
}.to_json
|
35
33
|
|
36
34
|
response = Net::HTTP.start(uri.hostname, uri.port, use_ssl: true) do |http|
|
@@ -49,6 +47,5 @@ module GenerateImage
|
|
49
47
|
raise RequestFailed, "Failed to generate image. Response code: #{response.code}. Response body: #{response.body}"
|
50
48
|
end
|
51
49
|
end
|
52
|
-
|
53
50
|
end
|
54
51
|
end
|