eleben 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: a5d77822e9bc4d7bca50fa017c35c4e422290206fd5df247d86447beb16752b6
+  data.tar.gz: 4c5d6fe4af81bb496544576c69d44f9692d15bd1a2edc493faf301ae5e4e6d1c
+SHA512:
+  metadata.gz: 5457b8a28bcd0ec0c42de7ab221a51d7b94dbd18febbb1341347cde772b20e02f37e10725b300f954055fcceca1d08856ff798f721448390bbbbd1beea7047fa
+  data.tar.gz: 78f91dc711a8facedf68949920dd99c5dd7399b5447e407bc8180a8acb7041ddb289a6facf0a4495e1dde8bf053abc1e0b970a115d8e774610004a4502f7e916

data/lib/eleben.rb

@@ -0,0 +1,155 @@
+require 'net/http'
+require 'json'
+require 'base64'
+require 'tempfile'
+require 'fileutils'
+
+class Eleben
+    DEFAULT_TIMEOUT = 300 # Default timeout in seconds
+
+    # Initializes a new instance of the StableDiffusion class.
+    #
+    # @param base_url [String] The base URL for the API.
+    # @param timeout [Integer] The timeout value in seconds (optional, default is 30).
+    def initialize(base_url, timeout: DEFAULT_TIMEOUT)
+        @base_url = base_url
+        @timeout = timeout
+    end
+
+    # Retrieves the available Stable Diffusion models.
+    #
+    # @return [Array] An array of Stable Diffusion models.
+    def get_sd_models
+        uri = URI("#{@base_url}/sdapi/v1/sd-models")
+        response = Net::HTTP.get(uri)
+        JSON.parse(response)
+    end
+
+    # Retrieves the Stable Diffusion Variational Autoencoder (VAE).
+    #something 
+    # @return [Hash] The Stable Diffusion VAE.
+    def get_sd_vae
+        uri = URI("#{@base_url}/sdapi/v1/sd-vae")
+        response = Net::HTTP.get(uri)
+        JSON.parse(response)
+    end
+
+    # Retrieves the available samplers.
+    #
+    # @return [Array] An array of samplers.
+    def get_samplers
+        uri = URI("#{@base_url}/sdapi/v1/samplers")
+        response = Net::HTTP.get(uri)
+        JSON.parse(response)
+    end
+
+    # Retrieves the available embeddings.
+    #
+    # @return [Array] An array of embeddings.
+    def get_embeddings
+        uri = URI("#{@base_url}/sdapi/v1/embeddings")
+        response = Net::HTTP.get(uri)
+        JSON.parse(response)
+    end
+
+    # Retrieves the available options.
+    #
+    # @return [Hash] The available options.
+    def get_options
+        uri = URI("#{@base_url}/sdapi/v1/options")
+        response = Net::HTTP.get(uri)
+        JSON.parse(response)
+    end
+
+    # Updates the options with new values.
+    #
+    # @param new_options [Hash] The new options to be updated.
+    # @return [Hash] The updated options.
+    def update_options(new_options)
+        uri = URI("#{@base_url}/sdapi/v1/options")
+        http = Net::HTTP.new(uri.host, uri.port)
+
+        http.read_timeout = @timeout
+        http.open_timeout = @timeout
+
+        request = Net::HTTP::Post.new(uri.path, 'Content-Type' => 'application/json')
+        request.body = new_options.to_json
+
+        response = http.request(request)
+
+        if response.code.to_i == 200
+            JSON.parse(response.body)
+        else
+            raise "Error #{response.code}: #{response.message}"
+        end
+    end
+
+    # Changes the model used for stable diffusion.
+    #
+    # @param model_name [String] The name of the model to be used.
+    # @return [void]
+    def change_model(model_name)
+        options = get_options
+        options['sd_model_checkpoint'] = model_name
+        update_options(options)
+    end
+
+    # Converts text to image using Stable Diffusion.
+    #
+    # @param request_body [Hash] The request body containing the text.
+    # @return [Hash] The generated image.
+    def txt2img(request_body = {})
+        uri = URI("#{@base_url}/sdapi/v1/txt2img")
+        http = Net::HTTP.new(uri.host, uri.port)
+
+        http.read_timeout = @timeout
+        http.open_timeout = @timeout
+
+        request = Net::HTTP::Post.new(uri.path, 'Content-Type' => 'application/json')
+        request.body = request_body.to_json
+
+        response = http.request(request)
+
+        if response.code.to_i == 200
+            JSON.parse(response.body)
+        else
+            raise "Error #{response.code}: #{response.message}"
+        end
+    end
+
+    # Saves the image data to disk at the specified output path.
+    #
+    # @param image_data [String] The base64-encoded image data.
+    # @param output_path [String] The path where the image will be saved.
+    # @return [void]
+    def save_image_to_disk(image_data, output_path)
+        image_binary_data = Base64.decode64(image_data)
+
+        # Use Tempfile to generate a unique temporary file path
+        temp_file = Tempfile.new(['sd_image', '.png'])
+        temp_file.binmode
+        temp_file.write(image_binary_data)
+        temp_file.close
+
+        # Move the temporary file to the specified output path
+        FileUtils.mv(temp_file.path, output_path)
+
+        puts "Image saved to #{output_path}"
+    end
+
+    # Converts text to image using the txt2img API and saves the resulting image to disk.
+    #
+    # @param params [Hash] The parameters for the txt2img request.
+    # @param output_path [String] The path where the image will be saved.
+    # @return [void]
+    def txt2img_and_save(params, output_path)
+        # Make txt2img request
+        txt2img_response = txt2img(params)
+        puts "txt2img Response:"
+        puts txt2img_response
+
+        # Save the image to disk
+        image_data = txt2img_response['images'][0]
+        save_image_to_disk(image_data, output_path)
+    end
+end

metadata

@@ -0,0 +1,71 @@
+--- !ruby/object:Gem::Specification
+name: eleben
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- lleetllama
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2024-01-18 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'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: This gem provides a Ruby client for the Automatic1111 API.
+email:
+- lleetllama@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/eleben.rb
+homepage: https://github.com/lleetllama/eleben
+licenses: []
+metadata: {}
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.2.32
+signing_key:
+specification_version: 4
+summary: A gem for interacting with the Automatic1111 API
+test_files: []