subtitle 0.1.7

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: e0908e0ca690672bf633eb1c212c5470aa7316f0e46fb5e224f60e1d17e3b6ee
+  data.tar.gz: 47aebdc5c91dd5ef4a93f3699f3c62042dbaeed01fb31b7ec23c386cc8dfe5ef
+SHA512:
+  metadata.gz: 45fa0b25384d0547ae54bd84f3acd9cfb06d393616e23aea3a986532a141d94ab9f65c3d215b7688c5e964537fd632d5acca949f53e1ec91634ee9f70d9ca61e
+  data.tar.gz: 155b11f1511aeb00d91ba59ea78cf8a432a603ae0f90bcf624828f227eb8abd92e8f162da3b0aae0b5334e75b83d1a18e8bbea71a1abf051866815b7c99f1cdb

data/lib/srt.rb

@@ -0,0 +1,81 @@
+require 'aws-sdk-translate'
+require 'aws-sdk-comprehend'
+
+class SRT
+  def initialize(awskey, awssecret)
+    @translate = Aws::Translate::Client.new(:access_key_id => "#{awskey}", :secret_access_key => "#{awssecret}")
+    @comp = Aws::Comprehend::Client.new(:access_key_id => "#{awskey}", :secret_access_key => "#{awssecret}")
+  end
+
+  def translate_text(srt_file, src_lang, dest_lang, out_file)
+    ccfile = File.open(srt_file, 'r:UTF-8', &:read)
+    outfile = File.open(out_file, "w")
+    text_collection = false
+    text_sample = ""
+    ccfile.each_line do | line |
+      if line =~ /^(\d\d:)\d\d:\d\d[,.]\d\d\d.*-->.*(\d\d:)\d\d:\d\d[,.]\d\d\d/
+        text_collection = true
+        outfile.puts line
+      elsif line.strip.empty? && !text_sample.empty?
+        json_text = JSON.parse(text_sample) rescue nil
+        if json_text.nil?
+          trans_resp = @translate.translate_text({ :text => "#{text_sample}" , :source_language_code => "#{src_lang}", :target_language_code => "#{dest_lang}"})
+          outfile.puts trans_resp.translated_text
+          outfile.puts
+        else
+          outfile.puts text_sample
+          outfile.puts
+        end
+        text_sample = ""
+        text_collection = false
+      elsif text_collection
+        text_sample << line
+      else
+        outfile.puts line
+      end
+      next
+    end
+
+    if !text_sample.empty?
+      trans_resp = @translate.translate_text({ :text => "#{text_sample}" , :source_language_code => "#{src_lang}", :target_language_code => "#{dest_lang}"})
+      outfile.puts trans_resp.translated_text
+      outfile.puts
+      outfile.close
+    end
+  end
+
+
+  def get_text(srt_file, num_chars)
+    ccfile = File.open(srt_file, 'r:UTF-8', &:read)
+    text_collection = false
+    text_sample = ""
+    ccfile.each_line do | line |
+      line = line
+      if line =~ /^(\d\d:)\d\d:\d\d[,.]\d\d\d.*-->.*(\d\d:)\d\d:\d\d[,.]\d\d\d/
+        text_collection = true
+      elsif line.strip.empty?
+        text_collection = false
+      elsif text_collection && text_sample.length < (num_chars+1)
+        text_sample << line
+      end
+      break if text_sample.length > (num_chars+1)
+      next
+    end
+    return text_sample[0,num_chars]
+  end
+
+  def detect_lang(srt_file)
+    lang = nil
+    begin
+    sample_text = get_text(srt_file, 100)
+    response = @comp.detect_dominant_language( {
+        text: "#{sample_text}"
+                                    })
+    lang = response[:languages][0][:language_code] rescue nil
+    rescue => error
+      puts "Error while detecting the language!!"
+    end
+    lang
+  end
+
+end

data/lib/subtitle.rb

@@ -0,0 +1,43 @@
+require "srt"
+
+class Subtitle
+  def initialize(awskey, awssecret, ccfile)
+    if awskey.nil? || awssecret.nil? || ccfile.nil?
+      raise "Invalid Arguments, please check"
+    end
+    @ccfile = ccfile
+    unless file_valid
+      raise "Incorrect File extension"
+    end
+    begin
+      @srt_parser = SRT.new(awskey, awssecret)
+    rescue
+      raise "Could not initialize Parser!!. Check the Keys supplied."
+    end
+  end
+
+  def detect_language
+    detected_lang = @srt_parser.detect_lang(@ccfile)
+    detected_lang
+  end
+
+  def translate_cc( dest_lang, src_lang = nil, outfile = nil)
+    if outfile.nil?
+      outfile = "#{@ccfile}_#{dest_lang}"
+    end
+    if src_lang.nil?
+      src_lang = detect_language
+      raise "could not detect Source Language!!"  if src_lang.nil?
+    end
+    @srt_parser.translate_text(@ccfile, src_lang, dest_lang, outfile)
+    outfile
+  end
+
+  def file_valid
+    valid = false
+    if @ccfile =~ /^.*\.(srt|vtt)$/
+      valid = true
+    end
+    valid
+  end
+end

data/lib/vtt.rb

@@ -0,0 +1,81 @@
+require 'aws-sdk-translate'
+require 'aws-sdk-comprehend'
+
+class VTT
+  def initialize(awskey, awssecret)
+    @translate = Aws::Translate::Client.new(:access_key_id => "#{awskey}", :secret_access_key => "#{awssecret}")
+    @comp = Aws::Comprehend::Client.new(:access_key_id => "#{awskey}", :secret_access_key => "#{awssecret}")
+  end
+
+  def translate_text(srt_file, src_lang, dest_lang, out_file)
+    ccfile = File.open(srt_file, 'r:UTF-8', &:read)
+    outfile = File.open(out_file, "w")
+    text_collection = false
+    text_sample = ""
+    ccfile.each_line do | line |
+      if line =~ /^(\d\d:)\d\d:\d\d[,.]\d\d\d.*-->.*(\d\d:)\d\d:\d\d[,.]\d\d\d/
+        text_collection = true
+        outfile.puts line
+      elsif line.strip.empty? && !text_sample.empty?
+        json_text = JSON.parse(text_sample) rescue nil
+        if json_text.nil?
+          trans_resp = @translate.translate_text({ :text => "#{text_sample}" , :source_language_code => "#{src_lang}", :target_language_code => "#{dest_lang}"})
+          outfile.puts trans_resp.translated_text
+          outfile.puts
+        else
+          outfile.puts text_sample
+          outfile.puts
+        end
+        text_sample = ""
+        text_collection = false
+      elsif text_collection
+        text_sample << line
+      else
+        outfile.puts line
+      end
+      next
+    end
+
+    if !text_sample.empty?
+      trans_resp = @translate.translate_text({ :text => "#{text_sample}" , :source_language_code => "#{src_lang}", :target_language_code => "#{dest_lang}"})
+      outfile.puts trans_resp.translated_text
+      outfile.puts
+      outfile.close
+    end
+  end
+
+
+  def get_text(srt_file, num_chars)
+    ccfile = File.open(srt_file, 'r:UTF-8', &:read)
+    text_collection = false
+    text_sample = ""
+    ccfile.each_line do | line |
+      line = line
+      if line =~ /^(\d\d:)\d\d:\d\d[,.]\d\d\d.*-->.*(\d\d:)\d\d:\d\d[,.]\d\d\d/
+        text_collection = true
+      elsif line.strip.empty?
+        text_collection = false
+      elsif text_collection && text_sample.length < (num_chars+1)
+        text_sample << line
+      end
+      break if text_sample.length > (num_chars+1)
+      next
+    end
+    return text_sample[0,num_chars]
+  end
+
+  def detect_lang(srt_file)
+    lang = nil
+    begin
+      sample_text = get_text(srt_file, 100)
+      response = @comp.detect_dominant_language( {
+                                                     text: "#{sample_text}"
+                                                 })
+      lang = response[:languages][0][:language_code] rescue nil
+    rescue => error
+      puts "Error while detecting the language!!"
+    end
+    lang
+  end
+
+end

metadata

@@ -0,0 +1,91 @@
+--- !ruby/object:Gem::Specification
+name: subtitle
+version: !ruby/object:Gem::Version
+  version: 0.1.7
+platform: ruby
+authors:
+- Maheshwaran G
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2019-10-17 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: aws-sdk-comprehend
+  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: aws-sdk-translate
+  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'
+description: subtitle gem to detect and translate closed caption for SubRip and WebVTT
+email:
+- pgmaheshwaran@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/srt.rb
+- lib/subtitle.rb
+- lib/vtt.rb
+homepage: https://github.com/cloudaffair/subtitle
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/cloudaffair/subtitle
+  source_code_uri: https://github.com/cloudaffair/subtitle
+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: []
+rubyforge_project: 
+rubygems_version: 2.7.3
+signing_key: 
+specification_version: 4
+summary: subtitle gem to detect and translate closed caption
+test_files: []