whisper.cpp 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0e8d71af28ba0c9af9fce2d70c86c36b6ebfb2a14d0b93720c87df0102358cc1
-  data.tar.gz: af1711e65ce109501e96aff8ce986764997033e15bc9eb97f9252d7cee9c074f
+  metadata.gz: 9233f65df607fba80d30ff4ae4a86dcbfaa1a750c88de487a2f98ace0a15de75
+  data.tar.gz: 157dcee2a04c9b514c3b36a92500532521aa964eed35ef9ffefd6f2d6b5fd40b
 SHA512:
-  metadata.gz: 98d417a79f60594c4a903aaaf2661e2b358113013dec48f8afd3d0bc2bf753a0091021cf912f59a87291fcf2fb65c87225807f7315366c89d169ee1b9f711efe
-  data.tar.gz: 54584e200cd1132ed924e9172a558bd5c3831ef4e7a63bf27a7b97e6726e9d74191a7db261dbd9c21ca2debb85e5fca9178727bb2285b4abe04d0d83b58ede2e
+  metadata.gz: 5a1b01fb0ae991764d318f054369f1740d9f7d5324aa8a0deedf7e5bad9f1c8ed84573061c3a68dfda1b73c60bc5761dd2f4f21931c22414098ebe3b1cc9b17d
+  data.tar.gz: bae0d8a234b23f88fca2a363438a43989ad01d6e732b95987f740b999b6fa9c74f300de1033e338a141d85e29d5dff2827763b408f599c34bbed50fe58f5d441

data/.gitignore

@@ -1,35 +1,7 @@
-# Ignore bundler config
-/.bundle
-
+/ext/Makefile
 /ext/whisper.cpp
-/lib/libwhispercpp.so
-Makefile
+libwhispercpp.so
 
-# Ignore Gem artifacts
 *.gem
-Gemfile.lock
-
-# Ignore log files
-/log/*.log
-
-# Ignore temp files
-/tmp
-/coverage/
-/doc/
-/.yardoc
-
-# Ignore OS-specific files
-.DS_Store
-Thumbs.db
-
-# Ignore editor-specific files
-.idea/
-/.vscode/
-/*.swp
-
-# Ignore pry history
-.pry_history
 
-# Ignore test coverage reports
-/coverage
 

data/.ruby-version

@@ -0,0 +1 @@
+3.3.0

data/Gemfile.lock

@@ -0,0 +1,34 @@
+PATH
+  remote: .
+  specs:
+    whisper.cpp (0.1.0)
+      ffi (~> 1.15)
+
+GEM
+  remote: https://rubygems.org/
+  specs:
+    coderay (1.1.3)
+    ffi (1.17.0)
+    ffi (1.17.0-x86_64-linux-gnu)
+    method_source (1.1.0)
+    pry (0.14.2)
+      coderay (~> 1.1)
+      method_source (~> 1.0)
+    rake (13.2.1)
+    rake-compiler (1.2.7)
+      rake
+
+PLATFORMS
+  ruby
+  x86_64-linux
+
+DEPENDENCIES
+  pry
+  rake-compiler
+  whisper.cpp!
+
+RUBY VERSION
+   ruby 3.3.0p0
+
+BUNDLED WITH
+   2.5.11

data/lib/whisper/model.rb

@@ -3,7 +3,10 @@ require_relative 'audio_processor'
 
 module Whisper
   class Model
-    def initialize(model_path)
+
+    TranscriptionResult = Struct.new :language, :output
+
+    def initialize model_path
       params = Whisper.whisper_context_default_params
       # Modify params as needed
       params[:use_gpu] = true
@@ -32,30 +35,33 @@ module Whisper
       result = Whisper.whisper_full @ctx, params, samples_ptr, n_samples
       raise 'Transcription failed' if result != 0
 
+      # Retrieve detected language
+      lang_id = Whisper.whisper_full_lang_id(@ctx)
+      language = Whisper.whisper_lang_str(lang_id)
+
       n_segments = Whisper.whisper_full_n_segments @ctx
+      output = ''
       case format.downcase
       when 'plaintext'
-        transcript = ''
         n_segments.times do |i|
           segment_text = Whisper.whisper_full_get_segment_text @ctx, i
-          transcript += segment_text
+          output += segment_text
         end
-        transcript
       when 'srt'
-        srt_content = ''
         n_segments.times do |i|
-          start_time = Whisper.whisper_full_get_segment_t0(@ctx, i) / 100.0
-          end_time = Whisper.whisper_full_get_segment_t1(@ctx, i) / 100.0
+          start_time   = Whisper.whisper_full_get_segment_t0(@ctx, i) / 100.0
+          end_time     = Whisper.whisper_full_get_segment_t1(@ctx, i) / 100.0
           segment_text = Whisper.whisper_full_get_segment_text @ctx, i
 
-          srt_content += "#{i + 1}\n"
-          srt_content += "#{format_time_srt start_time} --> #{format_time_srt end_time}\n"
-          srt_content += "#{segment_text.strip}\n\n"
+          output += "#{i + 1}\n"
+          output += "#{format_time_srt start_time} --> #{format_time_srt end_time}\n"
+          output += "#{segment_text.strip}\n\n"
         end
-        srt_content
       else
         raise "Unsupported format: #{format}"
       end
+
+      TranscriptionResult.new language, output
     end
 
     def close
@@ -64,13 +70,14 @@ module Whisper
 
     private
 
-    def format_time_srt(seconds)
-      hours = (seconds / 3600).to_i
+    def format_time_srt seconds
+      hours   = (seconds / 3600).to_i
       minutes = ((seconds % 3600) / 60).to_i
-      secs = (seconds % 60).to_i
-      millis = ((seconds - seconds.to_i) * 1000).to_i
+      secs    = (seconds % 60).to_i
+      millis  = ((seconds - seconds.to_i) * 1000).to_i
       format '%02d:%02d:%02d,%03d', hours, minutes, secs, millis
     end
+
   end
 end
 

data/lib/whisper.cpp.rb

@@ -3,7 +3,6 @@ require_relative 'whisper/audio_processor'
 require_relative 'whisper/model'
 
 module Whisper
-  # The module is intentionally left empty here.
-  # All classes and modules are defined in their respective files.
+
 end
 

data/lib/whisper.rb

@@ -177,5 +177,25 @@ module Whisper
   # Get segment start and end times
   attach_function :whisper_full_get_segment_t0, [:pointer, :int], :int64
   attach_function :whisper_full_get_segment_t1, [:pointer, :int], :int64
+
+    # Get detected language ID
+  attach_function :whisper_full_lang_id, [:pointer], :int
+
+  # Convert language ID to string
+  attach_function :whisper_lang_str, [:int], :string
+
+  # void log_callback(int level, const char *msg, void *user_data);
+  callback :ggml_log_callback, [:int, :string, :pointer], :void
+
+  # Set the log callback
+  attach_function :whisper_log_set, [:ggml_log_callback, :pointer], :void
+
+  # Define a no-op log callback to suppress debug messages
+  NOOP_LOG_CALLBACK = FFI::Function.new(:void, [:int, :string, :pointer]) do |level, msg, user_data|
+    # Intentionally do nothing to suppress logs
+  end
+  # Set the no-op log callback to suppress logging
+  Whisper.whisper_log_set NOOP_LOG_CALLBACK, FFI::Pointer::NULL unless ENV['WHISPER_DEBUG']
+
 end
 

data/whisper.cpp.gemspec

@@ -1,6 +1,8 @@
+require_relative 'lib/whisper/version'
+
 Gem::Specification.new do |spec|
   spec.name          = 'whisper.cpp'
-  spec.version       = '0.1.0'
+  spec.version       = Whisper::VERSION
   spec.authors       = ['Braulio Oliveira']
   spec.email         = ['brauliobo@gmail.com']
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: whisper.cpp
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Braulio Oliveira
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-09-27 00:00:00.000000000 Z
+date: 2024-09-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: ffi
@@ -62,7 +62,9 @@ extensions:
 extra_rdoc_files: []
 files:
 - ".gitignore"
+- ".ruby-version"
 - Gemfile
+- Gemfile.lock
 - LICENSE.md
 - README.md
 - Rakefile