typecast-ruby 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '07508aef051788fd8993b3c844fd431689193ea006541bc4f2cfc0224de8ed79'
-  data.tar.gz: 0dd2fc6ee5a0c435e5742f89dc8d14cb25feb9a3bc0dad9c6179ba0150617565
+  metadata.gz: e7decb1a088b596c9055b4251ed588af80d3187d997ff0ba337859f479835c64
+  data.tar.gz: 22265da24aa5c1d9a85df19615cfe0e771cb60ebbcadf15f2157cc2ae2da0b9d
 SHA512:
-  metadata.gz: a234192530e80f46f39d8bb4943d796dfad95f4dbe8521e92a52b74ee4b8c393ccd73741dc4f802469ea57e157d096f5a49a5e5ca8f93bfbde87605768a5543c
-  data.tar.gz: ec1f70d384151b04825b6018f9c478fe9c1a301859ddb529ee973128cbf8a44e097765bd9d735612ee9ed110576e1e782f08229cdeaae5d450a8600b540f3926
+  metadata.gz: 1a2024c96530643b47e3760d10a9b07afaf88cb50b7773420970c25e6678f5ac548be1ac15c6776430ab778a2b20370c9c2b03ad3df8938984d35269fe797dd9
+  data.tar.gz: dc4cdeda76890dfd668d519dac0f3c415907f6890c696b7b3fa24e3c255f546800e40246b743c822f56c8598e46fd0e6085987009b8540a26cef197971becbc1

data/lib/typecast/client.rb

@@ -4,6 +4,7 @@ require "securerandom"
 require "uri"
 
 require "typecast/errors"
+require "typecast/composer"
 require "typecast/models"
 require "typecast/timestamps"
 
@@ -30,6 +31,10 @@ module Typecast
       )
     end
 
+    def compose_speech
+      SpeechComposer.new(method(:text_to_speech))
+    end
+
     # Browse available API voices at https://typecast.ai/developers/api/voices.
     def generate_to_file(path, text:, voice_id:, model: Models::TTS_MODEL_V30, language: nil, prompt: nil, output: nil, seed: nil)
       request = Models::TTSRequest.new(

data/lib/typecast/composer.rb

@@ -0,0 +1,249 @@
+require "stringio"
+
+require "typecast/models"
+
+module Typecast
+  PausePart = Struct.new(:kind, :seconds, keyword_init: true)
+  TextPart = Struct.new(:kind, :text, keyword_init: true)
+
+  PAUSE_TOKEN = /<\|(\d+(?:\.\d+)?)s\|>/.freeze
+
+  def self.parse_pause_markup(text)
+    parts = []
+    last_index = 0
+    text.to_s.scan(PAUSE_TOKEN) do |match|
+      match_data = Regexp.last_match
+      if match_data.begin(0) > last_index
+        parts << TextPart.new(kind: "text", text: text[last_index...match_data.begin(0)])
+      end
+      parts << PausePart.new(kind: "pause", seconds: match[0].to_f)
+      last_index = match_data.end(0)
+    end
+    if last_index < text.length
+      parts << TextPart.new(kind: "text", text: text[last_index..-1])
+    end
+    parts
+  end
+
+  class SpeechComposer
+    def initialize(text_to_speech)
+      @text_to_speech = text_to_speech
+      @defaults = {}
+      @parts = []
+    end
+
+    def defaults(voice_id: nil, model: nil, language: nil, prompt: nil, output: nil, seed: nil)
+      @defaults = merge_settings(@defaults, settings_hash(
+        voice_id: voice_id,
+        model: model,
+        language: language,
+        prompt: prompt,
+        output: output,
+        seed: seed
+      ))
+      self
+    end
+
+    def say(text, voice_id: nil, model: nil, language: nil, prompt: nil, output: nil, seed: nil)
+      @parts << {
+        kind: "speech",
+        text: text.to_s,
+        settings: merge_settings(@defaults, settings_hash(
+          voice_id: voice_id,
+          model: model,
+          language: language,
+          prompt: prompt,
+          output: output,
+          seed: seed
+        ))
+      }
+      self
+    end
+
+    # Inserts silence between speech segments.
+    #
+    # seconds is a duration in seconds. Use 0.3 for 300 ms, 3 for 3 seconds.
+    def pause(seconds)
+      unless seconds.is_a?(Numeric) && seconds.finite? && seconds.positive?
+        raise ArgumentError, "pause seconds must be greater than 0"
+      end
+
+      @parts << PausePart.new(kind: "pause", seconds: seconds.to_f)
+      self
+    end
+
+    def generate
+      plan = build_plan
+      unless plan.any? { |part| part.is_a?(Hash) && part[:kind] == "speech" }
+        raise ArgumentError, "at least one speech segment is required"
+      end
+
+      output_format = @defaults.dig(:output, :audio_format) || Models::AUDIO_WAV
+      unless [Models::AUDIO_WAV, Models::AUDIO_MP3].include?(output_format)
+        raise ArgumentError, "unsupported composed speech output format: #{output_format}"
+      end
+
+      wav_spec = nil
+      output_samples = []
+      plan.each do |part|
+        if part.is_a?(PausePart)
+          raise ArgumentError, "pause cannot be the first composed part" if wav_spec.nil?
+
+          output_samples.concat(Array.new(seconds_to_samples(part.seconds, wav_spec[:sample_rate]), 0))
+          next
+        end
+
+        response = @text_to_speech.call(request_from_settings(part[:text], part[:settings]))
+        wav = parse_wav(response.audio_data)
+        if wav_spec && wav[:spec] != wav_spec
+          raise ArgumentError, "all composed WAV segments must use the same PCM format"
+        end
+
+        wav_spec = wav[:spec]
+        output_samples.concat(trim_silence(wav[:samples]))
+      end
+
+      wav_data = encode_wav(output_samples, wav_spec)
+      raise ArgumentError, "ffmpeg is required to encode composed speech as mp3" if output_format == Models::AUDIO_MP3
+
+      Models::TTSResponse.new(
+        audio_data: wav_data,
+        duration: output_samples.length.to_f / wav_spec[:sample_rate],
+        format: Models::AUDIO_WAV
+      )
+    end
+
+    private
+
+    def build_plan
+      plan = []
+      @parts.each do |part|
+        if part.is_a?(PausePart)
+          plan << part
+          next
+        end
+
+        Typecast.parse_pause_markup(part[:text]).each do |parsed|
+          if parsed.is_a?(PausePart)
+            plan << parsed
+            next
+          end
+          next if parsed.text.strip.empty?
+
+          raise ArgumentError, "voice_id is required for composed speech segments" if part[:settings][:voice_id].to_s.empty?
+          raise ArgumentError, "model is required for composed speech segments" if part[:settings][:model].to_s.empty?
+
+          plan << { kind: "speech", text: parsed.text, settings: part[:settings] }
+        end
+      end
+      plan
+    end
+
+    def settings_hash(voice_id:, model:, language:, prompt:, output:, seed:)
+      {
+        voice_id: voice_id,
+        model: model,
+        language: language,
+        prompt: prompt,
+        output: output_hash(output),
+        seed: seed
+      }.reject { |_key, value| value.nil? }
+    end
+
+    def merge_settings(base, override)
+      merged = base.merge(override)
+      merged[:output] = merge_output(base[:output], override[:output])
+      merged.reject { |_key, value| value.nil? }
+    end
+
+    def merge_output(base, override)
+      return nil if base.nil? && override.nil?
+
+      (base || {}).merge(override || {})
+    end
+
+    def output_hash(output)
+      return nil if output.nil?
+      return output.to_h if output.respond_to?(:to_h)
+
+      output
+    end
+
+    def request_from_settings(text, settings)
+      output = merge_output(settings[:output], audio_format: Models::AUDIO_WAV)
+      Models::TTSRequest.new(
+        voice_id: settings[:voice_id],
+        text: text,
+        model: settings[:model],
+        language: settings[:language],
+        prompt: settings[:prompt],
+        output: Models::Output.new(**output),
+        seed: settings[:seed]
+      )
+    end
+
+    def parse_wav(data)
+      io = StringIO.new(data)
+      raise ArgumentError, "unsupported WAV data" unless io.read(4) == "RIFF"
+
+      io.read(4)
+      raise ArgumentError, "unsupported WAV data" unless io.read(4) == "WAVE"
+
+      spec = nil
+      samples = nil
+      until io.eof?
+        chunk_id = io.read(4)
+        break if chunk_id.nil? || chunk_id.bytesize < 4
+
+        chunk_size_bytes = io.read(4)
+        raise ArgumentError, "unsupported WAV data" if chunk_size_bytes.nil? || chunk_size_bytes.bytesize < 4
+
+        chunk_size = chunk_size_bytes.unpack1("V")
+        chunk_data = io.read(chunk_size)
+        io.read(1) if chunk_size.odd?
+        raise ArgumentError, "unsupported WAV data" if chunk_data.nil? || chunk_data.bytesize < chunk_size
+
+        case chunk_id
+        when "fmt "
+          audio_format, channels, sample_rate, _byte_rate, _block_align, bits_per_sample = chunk_data.unpack("vvVVvv")
+          if audio_format != 1 || channels != 1 || bits_per_sample != 16
+            raise ArgumentError, "only mono 16-bit PCM WAV is supported for composed speech"
+          end
+          spec = { sample_rate: sample_rate, channels: channels, bits_per_sample: bits_per_sample }
+        when "data"
+          samples = chunk_data.unpack("s<*")
+        end
+      end
+
+      raise ArgumentError, "unsupported WAV data" if spec.nil? || samples.nil?
+
+      { spec: spec, samples: samples }
+    end
+
+    def encode_wav(samples, spec)
+      payload = samples.pack("s<*")
+      [
+        "RIFF",
+        [36 + payload.bytesize].pack("V"),
+        "WAVE",
+        "fmt ",
+        [16, 1, spec[:channels], spec[:sample_rate], spec[:sample_rate] * spec[:channels] * 2, spec[:channels] * 2, spec[:bits_per_sample]].pack("VvvVVvv"),
+        "data",
+        [payload.bytesize].pack("V"),
+        payload
+      ].join
+    end
+
+    def trim_silence(samples)
+      start_index = 0
+      end_index = samples.length
+      start_index += 1 while start_index < end_index && samples[start_index].abs <= 0
+      end_index -= 1 while end_index > start_index && samples[end_index - 1].abs <= 0
+      samples[start_index...end_index] || []
+    end
+
+    def seconds_to_samples(seconds, sample_rate)
+      (seconds * sample_rate).round
+    end
+  end
+end

data/lib/typecast.rb

@@ -1,4 +1,5 @@
 require "typecast/client"
+require "typecast/composer"
 require "typecast/errors"
 require "typecast/models"
 require "typecast/timestamps"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: typecast-ruby
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Neosapience
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2026-06-16 00:00:00.000000000 Z
+date: 2026-06-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: minitest
@@ -51,6 +51,7 @@ files:
 - THIRD-PARTY-LICENSES.md
 - lib/typecast.rb
 - lib/typecast/client.rb
+- lib/typecast/composer.rb
 - lib/typecast/errors.rb
 - lib/typecast/models.rb
 - lib/typecast/timestamps.rb