generate_image 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2de351fd26b27b162f1cadf99e8bebec19eab834a7326add766aa99494ef72d1
-  data.tar.gz: 7fd43c333077f86d8fef116b80b6868ea7da18b1d22ba82b7c43c77a211ea14a
+  metadata.gz: e0cf6f92bb8a3dc9f268736bc22cae4cfec8cd0af5a4e193e383feb156a1d2ac
+  data.tar.gz: 78735582f957597a69428939cc39f9a857583443c6f903927b4b449a3e7427b5
 SHA512:
-  metadata.gz: 20050a7726574a61d88718d29b0f1fb3dde18968a07cd4191f0f4bfaef9bc41b294378da6464e8456008d1804a0960061457dac6baf1e83f32e4d2be3dcf0fb7
-  data.tar.gz: 4ccd64ccd20b3393f471a0df141427e302a0b59d50e051739624ba7df16c81485845dc07d9cb8c8ea67a90a9890a830ee4e5961e35ba5848874bd2357af9fedf
+  metadata.gz: 25a102011cffbda0cec595234e51bccfc9593022362ae0d9d1cd44f0744ac1aed1c0d591921abe8270f7e655899e289fc2c7dae2cb0c03a548db35223fcc9ec2
+  data.tar.gz: 2bab3bbcd46182d4f2487a6a13f7fe5d74537b512b036ab958ac1d648a2190219c35129f1bc3df005ed9e56506bf345083c3f7925a6781116174a9a1d3977ae1

data/lib/generate_image/version.rb

@@ -1,3 +1,3 @@
 module GenerateImage
-  VERSION = "1.0.0"
+  VERSION = "1.1.0"
 end

data/lib/generate_image.rb

@@ -1,36 +1,56 @@
 require 'net/http'
-require 'json'
 
 module GenerateImage
-  class RequestFailed < StandardError; end
+  class RequestFailed < StandardError
+    def initialize(message = "Request failed")
+      super(message)
+    end
+  end
+
+  class MissingApiKey < RequestFailed
+    def initialize(message = "API key not set")
+      super(message)
+    end
+  end
 
   class << self
     API_ENDPOINT = 'https://api.openai.com/v1/images/generations'
-    IMAGE_MODEL_NAME = 'image-alpha-001'
-    TEXT_MODEL_NAME = 'text-davinci-002'
     API_KEY = ENV['DALL_E_API_KEY']
+    DEFAULT_OPTIONS = {
+      model: 'image-alpha-001',
+      num_images: 1,
+      size: '1024x1024',
+      response_format: 'url',
+      style: nil,
+      scale: 1,
+      seed: nil,
+      quality: 100,
+      text_model: 'text-davinci-002',
+      text_prompt: nil,
+      text_length: nil
+    }
 
     def generate_image(text, options = {})
-      unless API_KEY
-        raise StandardError, "API Key not set"
-      end
+      options = DEFAULT_OPTIONS.merge(options)
+      raise MissingApiKey if API_KEY.nil?
+
       uri = URI(API_ENDPOINT)
       request = Net::HTTP::Post.new(uri)
       request['Content-Type'] = 'application/json'
       request['Authorization'] = "Bearer #{API_KEY}"
       request.body = {
-        model: options[:model] || IMAGE_MODEL_NAME,
+        model: options[:model],
         prompt: text,
-        num_images: options[:num_images] || 1,
-        size: options[:size] || '512x512',
-        response_format: options[:response_format] || 'url',
-        style: options[:style] || nil,
-        scale: options[:scale] || 1,
-        seed: options[:seed] || nil,
-        quality: options[:quality] || 80,
-        text_model: options[:text_model] || TEXT_MODEL_NAME,
-        text_prompt: options[:text_prompt] || nil,
-        text_length: options[:text_length] || nil
+        num_images: options[:num_images],
+        size: options[:size],
+        response_format: options[:response_format],
+        style: options[:style],
+        scale: options[:scale],
+        seed: options[:seed],
+        quality: options[:quality],
+        text_model: options[:text_model],
+        text_prompt: options[:text_prompt],
+        text_length: options[:text_length]
       }.to_json
 
       response = Net::HTTP.start(uri.hostname, uri.port, use_ssl: true) do |http|
@@ -40,13 +60,13 @@ module GenerateImage
       if response.code == '200'
         if options[:response_format] == 'base64'
           image_base64 = JSON.parse(response.body)['data'][0]['base64']
-          return { image_base64: image_base64 }
+          { image_base64: image_base64 }
         else
           image_url = JSON.parse(response.body)['data'][0]['url']
-          return { image_url: image_url }
+          { image_url: image_url }
         end
       else
-        raise RequestFailed, "Failed to generate image. Response code: #{response.code}. Response body: #{response.body}"
+        raise RequestFailed.new("Failed to generate image. Response code: #{response.code}. Response body: #{response.body}")
       end
     end
   end

metadata

@@ -1,77 +1,78 @@
 --- !ruby/object:Gem::Specification
 name: generate_image
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - AMQOR Merouane
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-02-23 00:00:00.000000000 Z
+date: 2023-03-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: net-http
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.3.2
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.3.2
 - !ruby/object:Gem::Dependency
   name: sinatra
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 3.0.5
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 3.0.5
 - !ruby/object:Gem::Dependency
   name: json
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 2.6.3
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 2.6.3
 - !ruby/object:Gem::Dependency
   name: openai
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.3.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
-description: Unleash your imagination and create stunning images with the 'generate_image'
-  gem. This Ruby gem provides a seamless interface for accessing the powerful DALL-E
-  API and generates images based on the text you provide. Perfect for Ruby on Rails
-  projects, but can also be used in other Ruby projects.
+        version: 0.3.0
+description: The 'generate_image' gem provides a simple and easy-to-use interface
+  for generating images using the powerful DALL-E API from OpenAI. This Ruby gem can
+  be used in Ruby on Rails projects or any other Ruby projects to create stunning
+  images based on the text you provide. Unleash your imagination and generate images
+  for any use case, from social media posts to marketing materials and beyond.
 email:
-- marouane.amqor6@gmail.com
+- marouane.amqor@gmail.com
 executables:
 - console
 - setup