getimg 0.1.0 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9278befbee725204753bb3f9bedb08ae51dba0782b35aa17149dd5ea49754070
-  data.tar.gz: 350c69db91451ca82f21a597bcd624f5e19edd807797d761f1ae0946af42a5d9
+  metadata.gz: f1afb7594cb526ab50ac78d45fe06296e3dddb7f0c405c46379ab01486c9391b
+  data.tar.gz: 998563e15a69918d5838a16d8246c5390367ca0abbd509942830b89b2c759763
 SHA512:
-  metadata.gz: e9b3f3489eb956d699b47011ab22a082de2b7be077de98279602ee8317b1ded2c74fec18edbc8722d67987576dc73a8dfec005a4088ccfb64a554cdf5f370114
-  data.tar.gz: 1d1a56adc56e52034fb051afa762e620873d537c9b48d51823b5d250c5343506ee0e8b528afc7108e02f280e52dc2f0b8169c17669971f22468d2a1bc8b27702
+  metadata.gz: 45cf177d08c98576490f2ac4b6945efd0557b1e8368f193f0c25d634228aa8d14a4a63dfa78b6f4ba02e5490d267d97790a03d9e68859fb6a205b912fc85a351
+  data.tar.gz: b1343f3d6204d16ceafed56e158a6ad77dfb298a5d2502c145ab44679d3dbb911902eb5fe664abb50cbc28a3784459051f70323b49270da4e9309a930ef5b756

data/lib/getimg/client/enhancements.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module Getimg
+  class Client
+    class Enhancements
+      def initialize(client)
+        @client = client
+      end
+
+      # Body Params
+      # model string
+      # image string
+      # scale numbe
+      # output_format string
+
+      # gen = client.enhancements.upscale({model: "real-esrgan-4x", image: Getimg::Image.new.file_to_base64("xl-test.PNG"), scale: 4, output_format: "png"})
+      # Getimg::Image.new.base64_to_file(gen["image"])
+      def upscale(options = {})
+        @client.post("/enhancements/upscale", options)
+      end
+
+      # Body Params
+      # model string
+      # image string
+      # output_format string
+
+      # gen = client.enhancements.face_fix({model: "gfpgan-v1-3", image: Getimg::Image.new.file_to_base64("test.PNG"), output_format: "png"})
+      # Getimg::Image.new.base64_to_file(gen["image"])
+      def face_fix(options = {})
+        @client.post("/enhancements/face-fix", options)
+      end
+    end
+  end
+end

data/lib/getimg/client/stable_diffusion.rb

@@ -19,10 +19,39 @@ module Getimg
       # scheduler string
       # output_format string
 
-      # client.stable_diffusion.text_to_image({model: "stable-diffusion-v1-5", prompt: "a photo of an astronaut riding a horse on mars", guidance: 7.5, width: 256, height: 256, steps: 25, output_format: "png"})
+      # gen = client.stable_diffusion.text_to_image({model: "stable-diffusion-v1-5", prompt: "a photo of an astronaut riding a horse", guidance: 7.5, width: 256, height: 256, steps: 25, output_format: "png"})
+      # Getimg::Image.new.base64_to_file(gen["image"])
       def text_to_image(options = {})
         @client.post("/stable-diffusion/text-to-image", options)
       end
+
+      # Body Params
+      # model string
+      # prompt string, required
+      # negative_prompt string
+      # image string, required
+      # strength number
+      # steps integer
+      # guidance number
+      # seed integer
+      # scheduler string
+      # output_format string
+
+      # client.stable_diffusion.image_to_image(
+      #   {
+      #     model: "stable-diffusion-v1-5", 
+      #     prompt: "mars", 
+      #     image: Getimg::Image.new.file_to_base64("test.PNG"),
+      #     strength: 0.5, 
+      #     steps: 25,
+      #     guidance: 7.5, 
+      #     output_format: "png"
+      #   }
+      # )
+      # Getimg::Image.new.base64_to_file(gen["image"])
+      def image_to_image(options = {})
+        @client.post("/stable-diffusion/image-to-image", options)
+      end
     end
   end
 end

data/lib/getimg/client/stable_diffusion_xl.rb

@@ -21,10 +21,31 @@ module Getimg
       # scheduler string
       # output_format string
 
-      # client.stable_diffusion_xl.text_to_image({model: "stable-diffusion-xl-v1-0", prompt: "a photo of an astronaut riding a horse on mars", guidance: 7.5, width: 1024, height: 1024, steps: 30, output_format: "png"})
+      # gen = client.stable_diffusion_xl.text_to_image({model: "stable-diffusion-xl-v1-0", prompt: "a photo of an astronaut riding a horse", guidance: 7.5, width: 1024, height: 1024, steps: 30, output_format: "png"})
+      # Getimg::Image.new.base64_to_file(gen["image"])
       def text_to_image(options = {})
         @client.post("/stable-diffusion-xl/text-to-image", options)
       end
+
+      # Body Params
+      # model string
+      # prompt string, required
+      # negative_prompt string
+      # prompt_2 string
+      # negative_prompt_2 string
+      # image string, required
+      # strength number
+      # steps integer
+      # guidance number
+      # seed integer
+      # scheduler string
+      # output_format string
+
+      # gen = client.stable_diffusion_xl.image_to_image({model: "stable-diffusion-xl-v1-0", prompt: "mars background", image: Getimg::Image.new.file_to_base64("xl-test.PNG"), strength: 0.5, steps: 50, guidance: 7.5, output_format: "png"})
+      # Getimg::Image.new.base64_to_file(gen["image"])
+      def image_to_image(options = {})
+        @client.post("/stable-diffusion-xl/image-to-image", options)
+      end
     end
   end
 end

data/lib/getimg/client.rb

@@ -3,6 +3,7 @@
 require "getimg/client/models"
 require "getimg/client/stable_diffusion"
 require "getimg/client/stable_diffusion_xl"
+require "getimg/client/enhancements"
 require "getimg/client/account"
 
 module Getimg
@@ -34,6 +35,10 @@ module Getimg
       @stable_diffusion_xl ||= StableDiffusionXl.new(self)
     end
 
+    def enhancements
+      @enhancements ||= Enhancements.new(self)
+    end
+
     def account
       @account ||= Account.new(self)
     end

data/lib/getimg/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Getimg
-  VERSION = "0.1.0"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: getimg
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.2
 platform: ruby
 authors:
 - acidtib
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2023-11-09 00:00:00.000000000 Z
+date: 2023-11-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty
@@ -44,7 +44,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 3.16.1
-description: Client library for getimg API.
+description: 'Ruby bindings for getimg.ai, APIs can be found here: https://docs.getimg.ai'
 email:
 - arubinofaux@gmail.com
 executables: []
@@ -60,6 +60,7 @@ files:
 - lib/getimg.rb
 - lib/getimg/client.rb
 - lib/getimg/client/account.rb
+- lib/getimg/client/enhancements.rb
 - lib/getimg/client/models.rb
 - lib/getimg/client/stable_diffusion.rb
 - lib/getimg/client/stable_diffusion_xl.rb
@@ -90,5 +91,5 @@ requirements: []
 rubygems_version: 3.4.21
 signing_key: 
 specification_version: 4
-summary: Client library for getimg API.
+summary: Ruby bindings for getimg.ai
 test_files: []