generate_image 1.1.1 → 1.1.2.1
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 +7 -15
- 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: 9b49d016f61d888fd44be3bf340bba2613f7df0587922edf96ace80ef3d0c183
|
4
|
+
data.tar.gz: b05ca85ee8e2c925e1a19fbbaf493fc38e5804d62a26ec81409ed58f21e06f23
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 635b4d2702eef6145cd4f9be6ca51e8a3bff5a2d2cdfbf0993f7617a259c50dd71713477647b2de318abc7e8ce4cbf2038b7d4a3efecd5d628491fcb32141cff
|
7
|
+
data.tar.gz: ccf51aeffc3cc9571935710af9693e9e0ee5ec62c8591705244dbd31a51440412a1cf5b25d5367f4fa74778d93c2713b6768fb47ee4ef6d37debad3c133d6538
|
data/lib/generate_image.rb
CHANGED
@@ -6,31 +6,24 @@ module GenerateImage
|
|
6
6
|
|
7
7
|
class Client
|
8
8
|
API_ENDPOINT = 'https://api.openai.com/v1/images/generations'
|
9
|
-
IMAGE_MODEL_NAME = 'image-alpha-001'
|
10
|
-
TEXT_MODEL_NAME = 'text-davinci-002'
|
11
9
|
|
12
|
-
|
13
|
-
|
14
|
-
def initialize(api_key = nil)
|
15
|
-
@api_key = api_key || ENV['DALL_E_API_KEY']
|
16
|
-
raise StandardError, "API Key not set" unless @api_key
|
10
|
+
def initialize(api_key)
|
11
|
+
@api_key = api_key
|
17
12
|
end
|
18
13
|
|
19
14
|
def generate_image(text, options = {})
|
15
|
+
unless @api_key
|
16
|
+
raise StandardError, "API Key not set"
|
17
|
+
end
|
20
18
|
uri = URI(API_ENDPOINT)
|
21
19
|
request = Net::HTTP::Post.new(uri)
|
22
20
|
request['Content-Type'] = 'application/json'
|
23
|
-
request['Authorization'] = "Bearer #{
|
21
|
+
request['Authorization'] = "Bearer #{API_KEY}"
|
24
22
|
request.body = {
|
25
|
-
model: options[:model] || IMAGE_MODEL_NAME,
|
26
23
|
prompt: text,
|
27
|
-
|
24
|
+
n: options[:num_images] || 1,
|
28
25
|
size: options[:size] || '512x512',
|
29
26
|
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
27
|
}.to_json
|
35
28
|
|
36
29
|
response = Net::HTTP.start(uri.hostname, uri.port, use_ssl: true) do |http|
|
@@ -49,6 +42,5 @@ module GenerateImage
|
|
49
42
|
raise RequestFailed, "Failed to generate image. Response code: #{response.code}. Response body: #{response.body}"
|
50
43
|
end
|
51
44
|
end
|
52
|
-
|
53
45
|
end
|
54
46
|
end
|