getimg 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 31aeec5edbd705b9a3b5a16013c4de04680d870a6a202508e0d62dc97857877d
-  data.tar.gz: 33ff6d08ccc747e744bac524adba125f508d0d94f97502797ee40049d0aa7e2e
+  metadata.gz: f1afb7594cb526ab50ac78d45fe06296e3dddb7f0c405c46379ab01486c9391b
+  data.tar.gz: 998563e15a69918d5838a16d8246c5390367ca0abbd509942830b89b2c759763
 SHA512:
-  metadata.gz: 917f8c12b40080ece7177d2e5293da5cb0fa73733af6c76d7f5e0eae1b2f670976db71731babad610522c408a29ed145304437e209f5b8c70fd8e5106e92a228
-  data.tar.gz: 8b761f119d83aad973e4784600d3267dccd237a73b76cddbc1793b78be698142a482d66ca473f6442e6fd5954bca80022936ccfa07d346a256ada18b71362818
+  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.1"
+  VERSION = "0.1.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: getimg
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  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
@@ -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