stability_ai 0.1.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ea1e45d07d4a8dddedcfc4f8e433e51f243022f0e55cb96b50448951e7dd6a64
-  data.tar.gz: ea94f24e054cece698cb4616e3d3dd8495afba6eb2176a71082412887a52648a
+  metadata.gz: fb17addfc822fe3363f6d07a04a1055abc3bf7cc00d39e4d498bfcb0a10ee41c
+  data.tar.gz: 19dc57a7b3b13e09906c3c356931c253749d47c6412083f0ac690f93c5b5e32a
 SHA512:
-  metadata.gz: 3415165709e357e8d1d291f21a62f7340714a3187eabd2ffd231754364961d559d04e92ce7c34b0d6b7246e9c575b9ec3e9f14f105aac5bb1039e23703b76505
-  data.tar.gz: d2452910befdddf078445b45595e6b1961669a6d6febb705ec3f5ca10d81392ab2229431acf6972df6161f94dd7386f8082a5d77ebc6623f27df96f216cc8e27
+  metadata.gz: 186e15ecb08fa19b21e0cc51529541193dc5900f5b6c8b06156cdf866b35202c4937eda49ad62835aa4f93b573d864b37875fe9f2ed3ebf7eae8ccdb7d439403
+  data.tar.gz: 47f8aff5d068b97b19153be5b17bf210ad582f9806af8a80b8214f2def367666c48c914fc81a23072d56b63cb40e3706808c01b8ccdee3ccde998fddab096259

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 ## [Unreleased]
 
-## [0.1.0] - 2023-04-20
+## [0.1.0] - 2023-04-22
 
 - Initial release
+
+## [0.1.1] - 2023-04-22
+
+- Hotfix

data/Gemfile

@@ -10,3 +10,9 @@ gem "rake", "~> 13.0"
 gem "rspec", "~> 3.0"
 
 gem "rubocop", "~> 1.21"
+gem "httparty", "~> 0.20.0"
+gem "rmagick", "~> 5.2.0"
+
+group :development do
+  gem "pry-byebug"
+end

data/Gemfile.lock

@@ -0,0 +1,79 @@
+PATH
+  remote: .
+  specs:
+    stability_ai (0.1.1)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    ast (2.4.2)
+    byebug (11.1.3)
+    coderay (1.1.3)
+    diff-lcs (1.5.0)
+    httparty (0.20.0)
+      mime-types (~> 3.0)
+      multi_xml (>= 0.5.2)
+    json (2.6.3)
+    method_source (1.0.0)
+    mime-types (3.4.1)
+      mime-types-data (~> 3.2015)
+    mime-types-data (3.2023.0218.1)
+    multi_xml (0.6.0)
+    parallel (1.23.0)
+    parser (3.2.2.0)
+      ast (~> 2.4.1)
+    pkg-config (1.5.1)
+    pry (0.14.2)
+      coderay (~> 1.1)
+      method_source (~> 1.0)
+    pry-byebug (3.10.1)
+      byebug (~> 11.0)
+      pry (>= 0.13, < 0.15)
+    rainbow (3.1.1)
+    rake (13.0.6)
+    regexp_parser (2.8.0)
+    rexml (3.2.5)
+    rmagick (5.2.0)
+      pkg-config (~> 1.4)
+    rspec (3.12.0)
+      rspec-core (~> 3.12.0)
+      rspec-expectations (~> 3.12.0)
+      rspec-mocks (~> 3.12.0)
+    rspec-core (3.12.2)
+      rspec-support (~> 3.12.0)
+    rspec-expectations (3.12.3)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.12.0)
+    rspec-mocks (3.12.5)
+      diff-lcs (>= 1.2.0, < 2.0)
+      rspec-support (~> 3.12.0)
+    rspec-support (3.12.0)
+    rubocop (1.50.2)
+      json (~> 2.3)
+      parallel (~> 1.10)
+      parser (>= 3.2.0.0)
+      rainbow (>= 2.2.2, < 4.0)
+      regexp_parser (>= 1.8, < 3.0)
+      rexml (>= 3.2.5, < 4.0)
+      rubocop-ast (>= 1.28.0, < 2.0)
+      ruby-progressbar (~> 1.7)
+      unicode-display_width (>= 2.4.0, < 3.0)
+    rubocop-ast (1.28.0)
+      parser (>= 3.2.1.0)
+    ruby-progressbar (1.13.0)
+    unicode-display_width (2.4.2)
+
+PLATFORMS
+  x86_64-darwin-22
+
+DEPENDENCIES
+  httparty (~> 0.20.0)
+  pry-byebug
+  rake (~> 13.0)
+  rmagick (~> 5.2.0)
+  rspec (~> 3.0)
+  rubocop (~> 1.21)
+  stability_ai!
+
+BUNDLED WITH
+   2.4.12

data/README.md

@@ -1,39 +1,94 @@
-# StabilityAi
+# StabilityAI
 
-TODO: Delete this and the text below, and describe your gem
+StabilityAI is a Ruby gem that simplifies interactions with the Stability AI API. It supports image generation, image-to-image manipulation, upscaling, and masking.
 
-Welcome to your new gem! In this directory, you'll find the files you need to be able to package up your Ruby library into a gem. Put your Ruby code in the file `lib/stability_ai`. To experiment with that code, run `bin/console` for an interactive prompt.
+## Demo
+
+I made this `gem` part of a personal project: [ciel.chat](https://ciel.chat) - an AI supercharged WhatsApp bot, using ChatGPT, Bard, StableDiffusion, Dalle, GoogleSpeech, ElevenLabs.
+You can try it there for free ✌️
+
+## Disclaimer
+
+It's my first gem ever. There are plenty of room for improvements there - feel free to contribute!
 
 ## Installation
 
-TODO: Replace `UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG` with your gem name right after releasing it to RubyGems.org. Please do not do it earlier due to security reasons. Alternatively, replace this section with instructions to install your gem from git if you don't plan to release to RubyGems.org.
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'stability_ai'
+```
 
-Install the gem and add to the application's Gemfile by executing:
+And then execute:
 
-    $ bundle add UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+```ruby
+$ bundle install
+```
 
-If bundler is not being used to manage dependencies, install the gem by executing:
+Or install it yourself as:
 
-    $ gem install UPDATE_WITH_YOUR_GEM_NAME_PRIOR_TO_RELEASE_TO_RUBYGEMS_ORG
+```ruby
+gem install stability_ai
+```
+
+## Configuration
+
+Create an initializer in your Rails project or any Ruby application (e.g., `config/initializers/stability_ai.rb`) to configure your API keys and credentials:
+
+```ruby
+StabilityAI.configure do |config|
+  config.stability_api_key = 'your_stability_api_key'
+  config.path_prefix = './'
+end
+```
 
 ## Usage
 
-TODO: Write usage instructions here
+Here's an example of how to use the gem with the Stability AI API:
+Check [StabilityAI API](https://api.stability.ai/docs) for all the options available. If there are not passed in the `options` hash, the default value is used.
 
-## Development
+```ruby
+require 'stability_ai'
 
-After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.
 
-To install this gem onto your local machine, run `bundle exec rake install`. To release a new version, update the version number in `version.rb`, and then run `bundle exec rake release`, which will create a git tag for the version, push git commits and the created tag, and push the `.gem` file to [rubygems.org](https://rubygems.org).
+client = StabilityAI::Client.new
 
-## Contributing
+# Text to image
+text_to_image_response = client.text_to_image('engine_id', text_prompts: [{ text: 'A lighthouse on a cliff' }])
+uri_image_1 = text_to_image_response.image_uris.first           # ->  <img src="#{uri_image_1}"/>  or image_tag(uri_image_1)
+text_to_image_response.save_images('your_image_name_prefix')  # -> returns ["your_image_name_prefix_1.png", "your_image_name_prefix_2.png", ...]
 
-Bug reports and pull requests are welcome on GitHub at https://github.com/[USERNAME]/stability_ai. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [code of conduct](https://github.com/[USERNAME]/stability_ai/blob/master/CODE_OF_CONDUCT.md).
+# Image to image
 
-## License
+options =     {
+      text_prompts: [
+        {
+          text: "Snow",
+        }
+      ],
+      image_strength: 0.35, # Default: 0.35 How much influence the init_image has on the diffusion process. Values close to 1 will yield images very similar to the init_image while values close to 0 will yield images wildly different than the init_image
+      cfg_scale: 7, # DEFAULT: [0..35] How strictly the diffusion process adheres to the prompt text (higher values keep your image closer to your prompt),
+      # clip_guidance_preset: "FAST_BLUE", # DEFAULT: NONE, WTF IS THAT?? FAST_BLUE FAST_GREEN NONE SIMPLE SLOW SLOWER SLOWEST
+      # sampler: "DDIM",    # DEFAULT: Automatically choose by StableAI. DDIM DDPM K_DPMPP_2M K_DPMPP_2S_ANCESTRAL K_DPM_2 K_DPM_2_ANCESTRAL K_EULER K_EULER_ANCESTRAL K_HEUN K_LMS
+      sytle_preset: "neon-punk", # check
+    }
+response = client.image_to_image(image_path: image_path, )
+
+# Image to image upscale
+response = client.image_to_image_upscale(engine_id: "esrgan-v1-x2plus", image_path: image_path, options: { width: 1024 })
+response = client.image_to_image_upscale(image_path: image_path, use_maximum_resolution: true) # default engine: esrgan-v1-x2plus / use `use_maximum_resolution: true` is to use 2048px
+response = client.image_to_image_upscale(image_binary: image_binary, )
 
-The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).
+# Image to image masking
+# TODO
+```
 
-## Code of Conduct
+Replace `'engine_id'` with the appropriate engine ID for each method call.
+
+## Contributing
+
+Bug reports and pull requests are welcome on GitHub at https://github.com/mael-ha/stability_ai.
+
+## License
 
-Everyone interacting in the StabilityAi project's codebases, issue trackers, chat rooms and mailing lists is expected to follow the [code of conduct](https://github.com/[USERNAME]/stability_ai/blob/master/CODE_OF_CONDUCT.md).
+The gem is available as open source under the terms of the MIT License.

data/lib/stability_ai/api/engines.rb

@@ -0,0 +1,16 @@
+module StabilityAI
+    module API
+      module Engines
+        def engines_list
+          response = self.class.get('/v1/engines/list')
+          if response.code == 200
+            StabilityAI::Response::EnginesListResponse.new(response)
+          else
+            error_data = response.parsed_response
+            raise StabilityAI::StabilityAIError.new(response.code, error_data["id"], error_data["name"], error_data["message"])
+          end
+        end
+      end
+    end
+end
+  

data/lib/stability_ai/api/generation.rb

@@ -0,0 +1,176 @@
+require "uri"
+require "rmagick"
+
+module StabilityAI
+  module API
+    module Generation
+      def text_to_image(engine_id: nil, options: {})
+        response = self.class.post("/v1/generation/#{get_engine_id(engine_id)}/text-to-image", body: options.to_json,
+                                                                                               headers: { "Content-Type" => "application/json" })
+        handle_response(response)
+      end
+
+      def image_to_image(engine_id: nil, image_binary: nil, image_base64: nil, image_path: nil, options: {})
+        raise "No image file path provided." unless image_path || image_binary || image_base64
+        raise "No prompt provided." if options[:text_prompts].nil?
+
+        image_payload = create_payload(image_binary, image_base64, image_path)
+        image =         convert_and_resize_image(image_payload)
+        temp_file =     create_temp_file_from_image(image)
+        form_data =     set_form_data(image:, endpoint: :image_to_image, options:, temp_file:)
+
+        default_weight = options[:text_prompts].count > 1 ? (1 / options[:text_prompts].count).round(2) : 1
+        options[:text_prompts].each_with_index do |text_prompt, i|
+          form_data.merge!("text_prompts[#{i}][text]" => text_prompt[:text]) unless text_prompt[:text].nil?
+          form_data.merge!("text_prompts[#{i}][weight]" => text_prompt[:weight] || default_weight) unless text_prompt[:weight].nil?
+        end
+
+        headers = { "Content-Type" => "multipart/form-data" }
+        response = self.class.post("/v1/generation/#{get_engine_id(engine_id)}/image-to-image",
+                                   headers: headers, multipart: true, body: form_data)
+        handle_response(response)
+      ensure
+        if temp_file
+          temp_file.close
+          temp_file.unlink
+        end
+      end
+
+      def image_to_image_upscale(engine_id: "esrgan-v1-x2plus", image_binary: nil, image_base64: nil, image_path: nil, use_maximum_resolution: false, options: {})
+        raise "No image file path provided." unless image_path || image_binary || image_base64
+
+        image_payload = create_payload(image_binary, image_base64, image_path)
+        image =         convert_and_resize_image(image_payload)
+        temp_file =     create_temp_file_from_image(image)
+        form_data =     set_form_data(image:, endpoint: :image_to_image_upscale, options: options, temp_file:)
+
+        headers = { "Content-Type" => "multipart/form-data" }
+        response = self.class.post("/v1/generation/#{engine_id}/image-to-image/upscale", headers: headers, multipart: true, body: form_data)
+        handle_response(response)
+      ensure
+        if temp_file
+          temp_file.close
+          temp_file.unlink
+        end
+      end
+
+      # def text_to_animation(engine_id: nil, options: {})
+      #   response = self.class.post("/v1/generation/#{get_engine_id(engine_id)}/text-to-animation", body: options.to_json,
+      #                                                                                          headers: { "Content-Type" => "application/json" })
+      #   handle_response(response)
+      # end
+
+      private
+
+      def get_engine_id(engine_id)
+        engine_id || StabilityAI.configuration.default_engine_id
+      end
+
+      def handle_response(response)
+        if response.code == 200
+          StabilityAI::Response::ImageResponse.new(response)
+        else
+          error_data = response.parsed_response
+          raise StabilityAI::StabilityAIError.new(response.code, error_data["id"], error_data["name"],
+                                                  error_data["message"])
+        end
+      end
+
+      def create_payload(image_binary, image_base64, image_path)
+        if image_binary
+          StringIO.new(image_binary).read
+        elsif image_base64
+          StringIO.new(Base64.decode64(image_base64)).read
+        elsif image_path
+          File.read(File.open(image_path, 'rb'))
+        else
+          raise "No image binary, base64, or file path provided."
+        end
+      end
+
+      def create_temp_file_from_image(image)
+        temp_file = Tempfile.new(['image', '.png'])
+        temp_file.binmode
+        image.format = 'PNG' # Set the format explicitly before calling to_blob
+        temp_file.write(image.to_blob)
+        temp_file.flush
+        temp_file.rewind
+        temp_file
+      end
+
+      def get_image_dimensions(image)
+        [image.columns, image.rows]
+      end
+
+      def set_max_dimension(image, options, use_maximum_resolution)
+        if use_maximum_resolution
+          width, height = get_image_dimensions(image)
+          (width > height) ? { 'width' => 2048 } : { 'height' => 2048 }
+       else
+         case
+         when !options[:width].nil?
+           { 'width' => options[:width] }
+         when !options[:height].nil?
+           { 'height' => options[:height] }
+         else
+           nil
+         end
+       end
+      end
+
+      def set_form_data(image:, endpoint:, options:, temp_file:, use_maximum_resolution: true)
+        raise "No image file provided." unless temp_file
+
+        form_data = {}
+        case endpoint
+        when :image_to_image
+          form_data.merge!("init_image" => temp_file)
+          parameters = %w[image_strength, init_image_mode, cfg_scale, clip_guidance_preset, samples, steps]
+          parameters.each do |parameter|
+            next if options[parameter.to_sym].nil?
+            form_data.merge!(parameter => options[parameter.to_sym])
+          end
+          form_data
+        when :image_to_image_upscale
+          form_data.merge!("image" => temp_file)
+          max_dimension = set_max_dimension(image, options, use_maximum_resolution)
+          form_data.merge!(max_dimension) if max_dimension
+        end
+      end
+
+      def get_new_dimensions(width, height, step = 64)
+        new_width = (width / step) * step
+        new_height = (height / step) * step
+        [new_width, new_height]
+      end
+
+      def convert_and_resize_image(image_payload)
+        # Convert the input image to PNG format if it's a JPG or JPEG file
+        image = Magick::Image.from_blob(image_payload).first
+        image.format = 'PNG' if %w[JPG JPEG].include?(image.format)
+
+        if image.columns % 64 != 0 || image.rows % 64 != 0
+          # Calculate new dimensions
+          new_width, new_height = get_new_dimensions(image.columns, image.rows)
+
+          # Resize the image while maintaining aspect ratio
+          image.change_geometry("#{new_width}x#{new_height}") do |cols, rows, img|
+            img.resize!(cols, rows)
+          end
+
+          # Find the nearest multiples of 64 for both width and height
+          width_multiple = (image.columns / 64.0).floor * 64
+          height_multiple = (image.rows / 64.0).floor * 64
+
+          # Calculate new offsets for cropping
+          x_offset = (image.columns - width_multiple) / 2
+          y_offset = (image.rows - height_multiple) / 2
+
+          # Crop the centered part to ensure dimensions are multiples of 64
+          image.crop!(x_offset, y_offset, width_multiple, height_multiple)
+        end
+        image
+      end
+    end
+  end
+end

data/lib/stability_ai/api/user.rb

@@ -0,0 +1,16 @@
+module StabilityAI
+    module API
+      module User
+        def balance
+          response = self.class.get('/v1/user/balance')
+          if response.code == 200
+              StabilityAI::Response::UserResponse.new(response)
+          else
+              error_data = response.parsed_response
+              raise StabilityAI::StabilityAIError.new(response.code, error_data["id"], error_data["name"], error_data["message"])
+          end
+        end
+      end
+    end
+end
+  

data/lib/stability_ai/api.rb

@@ -0,0 +1,4 @@
+module StabilityAI
+    module API
+    end
+end

data/lib/stability_ai/client.rb

@@ -0,0 +1,39 @@
+module StabilityAI
+  class Client
+    include HTTParty
+    include StabilityAI::API::User
+    include StabilityAI::API::Engines
+    include StabilityAI::API::Generation
+
+    base_uri "https://api.stability.ai"
+    format :json
+
+    def initialize(api_key = nil)
+      @api_key = api_key || StabilityAI.configuration.stability_api_key || ENV["STABILITY_API_KEY"]
+      raise ArgumentError, "Missing Stability API key." unless @api_key
+
+      self.class.default_options.merge!(
+        headers: {
+          "Authorization" => "Bearer #{@api_key}",
+          "Content-Type" => "application/json",
+          "Accept" => "application/json"
+        }
+      )
+
+      set_default_engine_id
+    end
+
+    private
+
+    def set_default_engine_id
+      # Setting stable-diffusion-xl as default engine
+      return if StabilityAI.configuration.default_engine_id
+
+      engines_list_response = engines_list
+      engines = engines_list_response.engines
+
+      default_engine = engines.find { |engine| engine.id.include?("stable-diffusion-xl") }
+      StabilityAI.configuration.default_engine_id = default_engine.id if default_engine
+    end
+  end
+end

data/lib/stability_ai/configuration.rb

@@ -0,0 +1,12 @@
+module StabilityAI
+    class Configuration
+      attr_accessor :stability_api_key, :default_engine_id, :path_prefix
+  
+      def initialize
+        @stability_api_key = nil
+        @default_engine_id = nil
+        @path_prefix = './'
+      end
+    end
+  end
+  

data/lib/stability_ai/helpers/image_helper.rb

@@ -0,0 +1,16 @@
+module StabilityAI
+    module Helpers
+      class ImageHelper
+        def self.save_artifacts(artifacts, prefix = "output")
+          artifacts.each_with_index do |artifact, i|
+            save_base64_image("#{prefix}_#{i}.png", artifact.base64)
+          end
+        end
+  
+        def self.save_base64_image(file_path, base64_image)
+          File.binwrite(file_path, Base64.decode64(base64_image))
+        end
+      end
+    end
+end
+  

data/lib/stability_ai/response/base_response.rb

@@ -0,0 +1,14 @@
+module StabilityAI
+    module Response
+      class BaseResponse
+        attr_reader :raw_response, :http_status
+  
+        def initialize(raw_response)
+          @raw_response = raw_response
+          @parsed_response = @raw_response.parsed_response
+          @http_status = raw_response.code
+        end
+      end
+    end
+end
+  

data/lib/stability_ai/response/engines_response.rb

@@ -0,0 +1,9 @@
+module StabilityAI
+    module Response
+        class EnginesListResponse < BaseResponse
+            def engines
+                @parsed_response.map { |engine| OpenStruct.new(engine) }
+            end
+        end
+    end
+end

data/lib/stability_ai/response/image_response.rb

@@ -0,0 +1,39 @@
+module StabilityAI
+  module Response
+    class ImageResponse < BaseResponse
+      attr_reader :artifacts
+
+      def initialize(response)
+        super(response)
+        @artifacts = response.parsed_response["artifacts"]
+        convert_artifacts_to_openstruct
+      end
+
+      def convert_artifacts_to_openstruct
+        @artifacts.map! do |artifact|
+          image_binary_data = Base64.decode64(artifact["base64"])
+          artifact["image_binary"] = image_binary_data
+          artifact.delete("base64")
+          OpenStruct.new(artifact)
+        end
+      end
+
+      def save_images(filename_prefix: nil)
+        filename_prefix = Time.now.utc.strftime("%Y%m%d%H%M%S") if filename_prefix.nil?
+        downloaded_images = []
+        path_prefix = StabilityAI.configuration.path_prefix
+        @artifacts.each_with_index do |artifact, i|
+          image_name = "#{filename_prefix}_#{i}.png"
+          image_path = path_prefix + image_name
+          File.binwrite(image_path, artifact["image_binary"])
+          downloaded_images << {
+            file_name: image_name,
+            seed: artifact["seed"],
+            finish_reason: artifact["finishReason"],
+          }
+        end
+        downloaded_images
+      end
+    end
+  end
+end

data/lib/stability_ai/response/user_response.rb

@@ -0,0 +1,10 @@
+module StabilityAI
+    module Response
+      class UserResponse < BaseResponse
+        def credits
+          @parsed_response['credits']
+        end
+      end
+    end
+end
+  

data/lib/stability_ai/stability_ai_error.rb

@@ -0,0 +1,14 @@
+module StabilityAI
+    class StabilityAIError < StandardError
+      attr_reader :id, :name, :message
+  
+      def initialize(code, id, name, message)
+        @code = code
+        @id = id
+        @name = name
+        @message= message
+        super("#{code}# #{name}: #{message} (ID: #{id})")
+      end
+    end
+end
+  

data/lib/stability_ai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
-module StabilityAi
-  VERSION = "0.1.0"
+module StabilityAI
+  VERSION = "0.1.1"
 end

data/lib/stability_ai.rb

@@ -1,8 +1,32 @@
 # frozen_string_literal: true
 
+require "httparty"
+require "pry-byebug"
+require "dotenv/load"
+require "base64"
+require "uri"
 require_relative "stability_ai/version"
+require_relative "stability_ai/api"
+require_relative "stability_ai/api/user"
+require_relative "stability_ai/api/engines"
+require_relative "stability_ai/api/generation"
+require_relative "stability_ai/client"
+require_relative "stability_ai/response/base_response"
+require_relative "stability_ai/response/user_response"
+require_relative "stability_ai/response/engines_response"
+require_relative "stability_ai/response/image_response"
+require_relative "stability_ai/stability_ai_error"
+require_relative "stability_ai/configuration"
+
+module StabilityAI
+  class << self
+    attr_accessor :configuration
+  end
+
+  def self.configure
+    self.configuration ||= Configuration.new
+    yield(configuration)
+  end
 
-module StabilityAi
   class Error < StandardError; end
-  # Your code goes here...
 end

data/stability_ai.gemspec

@@ -4,7 +4,7 @@ require_relative "lib/stability_ai/version"
 
 Gem::Specification.new do |spec|
   spec.name = "stability_ai"
-  spec.version = StabilityAi::VERSION
+  spec.version = StabilityAI::VERSION
   spec.authors = ["Maël Harnois"]
   spec.email = ["m@maelus.com"]
 
@@ -13,7 +13,6 @@ Gem::Specification.new do |spec|
   spec.license = "MIT"
   spec.required_ruby_version = ">= 2.6.0"
 
-
   spec.metadata["homepage_uri"] = spec.homepage
   spec.metadata["source_code_uri"] = "https://github.com/mael-ha/stability_ai"
   spec.metadata["changelog_uri"] = "https://github.com/mael-ha/stability_ai/blob/main/CHANGELOG.md"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: stability_ai
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Maël Harnois
@@ -22,10 +22,23 @@ files:
 - CHANGELOG.md
 - CODE_OF_CONDUCT.md
 - Gemfile
+- Gemfile.lock
 - LICENSE.txt
 - README.md
 - Rakefile
 - lib/stability_ai.rb
+- lib/stability_ai/api.rb
+- lib/stability_ai/api/engines.rb
+- lib/stability_ai/api/generation.rb
+- lib/stability_ai/api/user.rb
+- lib/stability_ai/client.rb
+- lib/stability_ai/configuration.rb
+- lib/stability_ai/helpers/image_helper.rb
+- lib/stability_ai/response/base_response.rb
+- lib/stability_ai/response/engines_response.rb
+- lib/stability_ai/response/image_response.rb
+- lib/stability_ai/response/user_response.rb
+- lib/stability_ai/stability_ai_error.rb
 - lib/stability_ai/version.rb
 - sig/stability_ai.rbs
 - stability_ai.gemspec