generate_image 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 462b48a0798e5e6d50bffb2386d428b69d43b2c547a0e1bcc7ef01e1ada2fdb3
-  data.tar.gz: 9f6bf6aac0c92744f538aee95c82a64dab88200370891918a6406c77cb047e93
+  metadata.gz: a729f8fda3aa05792e49f221a8d94d471185f0fba92dbb6d76fc5ea92767a6b5
+  data.tar.gz: c9df8fe206dd5e467085640f253b6bae608e89ec20c5c6ccfdbda2ff60d44684
 SHA512:
-  metadata.gz: 3682a51b886ecf2808b6f826bb30243e0b08a2628e93b6011c241fa201ab8abaaecc2195f9f5e46423a347330dbab8286e28fd3795417fb2661c0cf23c939f46
-  data.tar.gz: 5426a9c7093f85c6078f9d6d3deaab45253ca5d538dabe5ffd01c53317e8c452e623f0b071f693367c29702990b313499023ec047c62513bd599996100f5ded7
+  metadata.gz: 92969cfef020fc0c2c3c3c50b9c62489d12206dfaf6f68fd1d565388580866cf9f732751b74b93c0eacb6453e6e35ee1c82b79ca4a3847fc05b28482d079f699
+  data.tar.gz: c0d3971e217207be1043dd081a979b2cd669813e1b217bd83214dc8a1dcec29be8bda115562b05005bc3db915a2622fa7e5c26c3811c9a44859585980e334195

data/lib/generate_image/version.rb

@@ -1,3 +1,3 @@
 module GenerateImage
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

data/lib/generate_image.rb

@@ -2,22 +2,24 @@ require 'net/http'
 require 'json'
 
 module GenerateImage
+  class RequestFailed < StandardError; end
+
   class << self
-    API_ENDPOINT = 'https://api.openai.com/v1/images/generations'
-    MODEL_NAME = 'image-alpha-001'
+    IMAGE_API_ENDPOINT = 'https://api.openai.com/v1/images/generations'
+    IMAGE_MODEL_NAME = 'image-alpha-001'
+    REWRITE_MODEL_NAME = 'text-davinci-002'
+    API_KEY = ENV['DALL_E_API_KEY']
 
     def generate_image(text)
-      api_key = ENV['DALL_E_API_KEY']
-      unless api_key
+      unless API_KEY
         raise StandardError, "API Key not set"
       end
-
-      uri = URI(API_ENDPOINT)
+      uri = URI(IMAGE_API_ENDPOINT)
       request = Net::HTTP::Post.new(uri)
       request['Content-Type'] = 'application/json'
-      request['Authorization'] = "Bearer #{api_key}"
+      request['Authorization'] = "Bearer #{API_KEY}"
       request.body = {
-        model: MODEL_NAME,
+        model: IMAGE_MODEL_NAME,
         prompt: text,
         num_images: 1
       }.to_json
@@ -30,9 +32,7 @@ module GenerateImage
         image_url = JSON.parse(response.body)['data'][0]['url']
         return { image_url: image_url }
       else
-        message = "Failed to generate image. Response code: #{response.code}. Response body: #{response.body}"
-        Rails.logger.error(message) if defined?(Rails)
-        return { error: message }, 500
+        raise RequestFailed, "Failed to generate image. Response code: #{response.code}. Response body: #{response.body}"
       end
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: generate_image
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - AMQOR Merouane