tulpa-prompter 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 60722d5f03571bd1d01dab440f16e903127cc2548615aa88a184204ff680f40c
-  data.tar.gz: 433e3440317d8ea32987e591150bce1b03019128581d3ed69ef298240084f9d1
+  metadata.gz: 867ab87efc3a3655fae1be03b6f5b9e953a9edfaf2566ac191c2bc3b52d773c9
+  data.tar.gz: ce8697e0098318da3245ebf5486475bb7187f9574aa0b9ea50b448c48c20b0d0
 SHA512:
-  metadata.gz: f735c4fa0ac6b95487af971f6d5b147a83575e527447d68328fe6e289d7e360f1c3181d9a93f70bdc9d504ea5161bbfbc3b6c54a4a62afac2a1cff84b90c99a3
-  data.tar.gz: 3cd9110f3e1a659bb2e7a18db9926d181f75d3f5cbd8bcf968cf6fab00b66562c456bc7f416a05320740ab64908c6ba609892f5b9028656f41446629b5a25d23
+  metadata.gz: 5274be0e1de701e0f5528ae45b82d238fb9b93f09685d4763d58c3b9b346d1b2fe6575c0167ed3c22734271ab1bcb425495224422eba2aefa73a0125c2e2ff09
+  data.tar.gz: 886511a804ddae7eb83f2d6308a2b18d4e7f09ad469dca74d144cf13dc9d3cd5d8e7207bb04800ed1f8a2208d7fe9e30bbea98e3e63c7a55f35742143db02a61

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    tulpa-prompter (0.1.1)
+    tulpa-prompter (0.1.3)
 
 GEM
   remote: https://rubygems.org/

data/README.md

@@ -26,7 +26,8 @@ json_string = TulpaPrompter.call(
   prefix: "portrait of ",
   suffix: " by greg rutkowski",
   interval: 15,
-  shuffle: false
+  shuffle: false,
+  format: :deforum_colab # default
 )
 ```
 

data/lib/tulpa-prompter.rb

@@ -4,19 +4,44 @@ module TulpaPrompter
   class Error < StandardError; end
 
   def self.call(options)
+    options[:format] ||= :deforum_colab
+    options[:format] = parse_format_options(options[:format]) if options[:format].is_a?(Symbol)
+
     options[:interval] ||= 15
     options[:interval] = options[:interval].to_i
-    options[:lines].shuffle! if options[:shuffle]
+    options[:lines].shuffle! if options[:shuffle].to_s.downcase == "true" || options[:shuffle].to_s == "1"
 
-    animation_prompts = "animation_prompts = {\n"
-    options[:lines].each_with_index do |line, index|
+    prompts = options[:lines].map.with_index do |line, index|
       frame = index * options[:interval]
-      prompt = line.strip
+      frame = "\"#{frame}\"" if options[:format][:frame] == :string
 
+      prompt = line.strip
       prompt = options[:prefix].to_s + prompt + options[:suffix].to_s
       prompt = prompt.gsub("\"", "\\\"")
-      animation_prompts << "  #{frame}: \"#{prompt}\",\n"
+
+      "  #{frame}: \"#{prompt}\""
     end
-    animation_prompts << "}"
+
+    output = ""
+    output << "#{options[:format][:assign]} = " if options[:format][:assign]
+    output << "{\n"
+    output << prompts.join(",\n")
+    output << "\n}"
+
+    output
+  end
+
+  def self.parse_format_options(format)
+    format_options = {}
+
+    case format
+    when :deforum_webui
+      format_options[:frame] = :string
+    else
+      format_options[:assign] = "animation_prompts"
+      format_options[:frame] = :integer
+    end
+
+    format_options
   end
 end

data/lib/tulpa_prompter/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tulpa-prompter
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Dreaming Tulpa
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2022-10-17 00:00:00.000000000 Z
+date: 2022-10-31 00:00:00.000000000 Z
 dependencies: []
 description:
 email: