gen-ai 0.2.0 → 0.2.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 20e16166f4ff21fb5130cf198be040b3d89019ba18cebd55e25f02720b32de62
-  data.tar.gz: 0eb3c0b7aace3f1dc97cc6f28635b45dae67aac3304d9fc9926a2f6a517d42cf
+  metadata.gz: 2b25d130c100818a73d652eaf9b585a4d024f6134a66de3f20eb5dd6e3d2212a
+  data.tar.gz: 7d971d676915f71edf74fec57cd7cb59b2d2c6607d7ed818d05d07b19f8cdecc
 SHA512:
-  metadata.gz: '02380f33e49b058b40c62c07b2470d5cebb36e8301f4d985077fbd27c9279948cf981227b6f8293bcac74a7b455f17e0ce64310601afcc7cfcd4bf81e3293fee'
-  data.tar.gz: 51e6bd0e3950fa62ae26f7f4103fbdd457e262212a840a9dab3abf84ed8337a6f50b60f011fbc717b891b4ac78483cbc60224784f4dfc78cdc7b938ac29f22a6
+  metadata.gz: 98554f5f32925f6ea70ea78e8c462d3d24949fb265fed69ae5f73fc722206c455e81bec75a879aa21bf841b372518f6ffce5497ec3c82bf60260b4d4c8ed2525
+  data.tar.gz: 905b2aa2b483711de399aa166dc5c83f9e7a482a1649f77281c995adbcf0f3dc9847b3ecbc0bdc21028a0f79f4ffb770c0a09dbad6a02d2a191d206dfeb43712

data/CHANGELOG.md

@@ -1,4 +1,5 @@
-## [Unreleased]
+## [0.2.0] - 2023-10-22
+- Add image generation abilities
 
 ## [0.1.0] - 2023-10-13
 

data/README.md

@@ -2,7 +2,7 @@
 
 ✨ Generative AI toolset for Ruby ✨
 
-GenAI allows you to easily integrate Generative AI model providers like OpenAI, Google Vertex AI, Stability AI, etc. Easily add Large Language Models, Stable Diffusion image generation, and other AI models into your application!
+GenAI allows you to easily integrate Generative AI model providers like OpenAI, Google Vertex AI, Stability AI, etc. Easily add Large Language Models, Stable Diffusion image generation, and other AI model integrations into your application!
 
 ![Tests](https://github.com/alchaplinsky/gen-ai/actions/workflows/main.yml/badge.svg?branch=main)
 [![Gem Version](https://badge.fury.io/rb/gen-ai.svg)](https://badge.fury.io/rb/gen-ai)
@@ -31,12 +31,12 @@ Language models capabilities
 | **Google Palm2** | ✅        | ✅         | ✅            | 🛠️        | 🛠️             |
 
 
-Image generation model capabilities 
+Image generation model capabilities
 
 | Provider         | Generate  | Variations | Edit         | Upscale   |
 | ---------------- | --------- | :--------: | :----------: | :-------: |
 | **OpenAI**       | ✅        | ✅         | ✅            | ❌        |
-| **StabilityAI**  | ✅        | ❌         | ✅            | 🛠️        |
+| **StabilityAI**  | ✅        | ❌         | ✅            | ✅        |
 
 ### Language
 
@@ -123,6 +123,8 @@ File.open('dog.jpg', 'wb') do |f|
   f.write(Base64.decode64(result.value))
 end
 ```
+![dog](https://github.com/alchaplinsky/gen-ai/assets/695947/27a2af5d-530b-4966-94e8-6cdf628b6cac)
+
 
 Get more **variations** of the same image
 
@@ -139,6 +141,8 @@ File.open('dog_variation.jpg', 'wb') do |f|
 end
 
 ```
+![dog_variation](https://github.com/alchaplinsky/gen-ai/assets/695947/977f5238-0114-4085-8e61-8f8b356ce308)
+
 
 **Editing** existing images with additional prompt
 
@@ -153,9 +157,12 @@ result.value
 File.open('dog_edited.jpg', 'wb') do |f|
   f.write(Base64.decode64(result.value))
 end
-
 ```
 
+![llama](https://github.com/alchaplinsky/gen-ai/assets/695947/9c862c6c-428e-463c-b935-ca749a6a33df)
+![llama_edited](https://github.com/alchaplinsky/gen-ai/assets/695947/070d8e6a-07a0-4ed2-826f-8b9aabd183ae)
+
+
 ## Development
 
 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.

data/gen-ai.gemspec

@@ -9,8 +9,9 @@ Gem::Specification.new do |spec|
   spec.email = ['alchaplinsky@gmail.com']
 
   spec.summary = 'Generative AI toolset for Ruby.'
-  spec.description = 'Generative AI toolset for Ruby.'
+  spec.description = 'GenAI allows you to easily integrate Generative AI model providers like OpenAI, Google Vertex AI, Stability AI, etc'
   spec.homepage = 'https://github.com/alchaplinsky/gen-ai'
+  spec.licenses = ['MIT']
   spec.required_ruby_version = '>= 2.7.0'
 
   # spec.metadata["allowed_push_host"] = "TODO: Set to your gem server 'https://example.com'"
@@ -33,7 +34,7 @@ Gem::Specification.new do |spec|
 
   # Uncomment to register a new dependency of your gem
   spec.add_dependency 'faraday', '~> 2.7'
-  spec.add_dependency 'faraday-multipart'
+  spec.add_dependency 'faraday-multipart', '~> 1.0'
   spec.add_dependency 'zeitwerk', '~> 2.6'
 
   spec.add_development_dependency 'google_palm_api', '~> 0.1'

data/lib/gen_ai/image/base.rb

@@ -10,6 +10,14 @@ module GenAI
       def variations(...)
         raise NotImplementedError, "#{self.class.name} does not support variations"
       end
+
+      def edit(...)
+        raise NotImplementedError, "#{self.class.name} does not support editing"
+      end
+
+      def upscale(...)
+        raise NotImplementedError, "#{self.class.name} does not support upscaling"
+      end
     end
   end
 end

data/lib/gen_ai/image/stability_ai.rb

@@ -8,6 +8,7 @@ module GenAI
       DEFAULT_SIZE = '256x256'
       API_BASE_URL = 'https://api.stability.ai'
       DEFAULT_MODEL = 'stable-diffusion-xl-beta-v2-2-2'
+      UPSCALE_MODEL = 'stable-diffusion-x4-latent-upscaler'
 
       def initialize(token:, options: {})
         build_client(token)
@@ -39,6 +40,19 @@ module GenAI
         )
       end
 
+      def upscale(image, options = {})
+        model = options[:model] || UPSCALE_MODEL
+        url = "/v1/generation/#{model}/image-to-image/upscale"
+
+        response = client.post url, build_upscale_body(image, options), multipart: true
+
+        build_result(
+          raw: response,
+          model: model,
+          parsed: response['artifacts'].map { |artifact| artifact['base64'] }
+        )
+      end
+
       private
 
       def build_client(token)
@@ -56,10 +70,20 @@ module GenAI
       end
 
       def build_edit_body(image, prompt, options)
-        {
+        params = {
           init_image: File.binread(image),
           'text_prompts[0][text]' => prompt,
           'text_prompts[0][weight]' => 1.0
+        }
+        params.merge!(mask: File.binread(options.delete(:mask))) if options[:mask]
+        params.merge(options)
+      end
+
+      def build_upscale_body(image, options)
+        w, = size(options)
+        {
+          image: File.binread(image),
+          width: w
         }.merge(options)
       end
 

data/lib/gen_ai/image.rb

@@ -18,7 +18,9 @@ module GenAI
       client.edit(image, prompt, options)
     end
 
-    # def upscale; end
+    def upscale(image, options = {})
+      client.upscale(image, options)
+    end
 
     private
 

data/lib/gen_ai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module GenAI
-  VERSION = '0.2.0'
+  VERSION = '0.2.2'
 end

data/lib/gen_ai.rb

@@ -1,7 +1,6 @@
 # frozen_string_literal: true
 
 require 'zeitwerk'
-require 'pry'
 
 loader = Zeitwerk::Loader.for_gem
 loader.inflector.inflect(

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gen-ai
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.2
 platform: ruby
 authors:
 - Alex Chaplinsky
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-10-22 00:00:00.000000000 Z
+date: 2023-10-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -28,16 +28,16 @@ dependencies:
   name: faraday-multipart
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.0'
 - !ruby/object:Gem::Dependency
   name: zeitwerk
   requirement: !ruby/object:Gem::Requirement
@@ -80,7 +80,8 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '5.1'
-description: Generative AI toolset for Ruby.
+description: GenAI allows you to easily integrate Generative AI model providers like
+  OpenAI, Google Vertex AI, Stability AI, etc
 email:
 - alchaplinsky@gmail.com
 executables: []
@@ -111,7 +112,8 @@ files:
 - lib/gen_ai/version.rb
 - sig/gen_ai.rbs
 homepage: https://github.com/alchaplinsky/gen-ai
-licenses: []
+licenses:
+- MIT
 metadata:
   homepage_uri: https://github.com/alchaplinsky/gen-ai
   source_code_uri: https://github.com/alchaplinsky/gen-ai