audio_analysis 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0a1c982a664f5cea0335d76a00b15057db67c92bbdf3a25f4972b201defb7409
-  data.tar.gz: 98884a8ab4ebc3b098e961b9c58f1261fa1d8f2df1a541d5491d1114765c0f61
+  metadata.gz: 77f3e6d28dab05d54e885f1d430c3e1034ad56c1ef318ffc7fcc01584022bdd3
+  data.tar.gz: 24a283dcf4579c575fcb82f3b661423a476fcd9df0ff9ab5163421fb9210f517
 SHA512:
-  metadata.gz: 8fb0f096e78c10d32bb822e51ddd6bccd9e12874580c6a875056836f0f55d116551da7799bb69f4b8bc2b6cdc709b2e3bdd963ca233a60fca010d9858210dc6a
-  data.tar.gz: 80df53e6a53b1d4a74d3ae23469774994caaa01f2ff5d52e5104175a39c9ee1973b731f838ea64f580c6dddca419d3f5ee09f2718f38a452f66f718ca26e885d
+  metadata.gz: 78c530c25407cc5975649e213043cecb66e8262e1c7b86c7a8fce6a8dbb6064ed2056ac0b5ab3aade7ed9d5f70faa4417d98921b21f07c42f5b801076c7c1733
+  data.tar.gz: d8caae585bd56c1266fe705f3c5538386dc59db3b45d9862238f3488c244edaa1c8b7c6e83991ab6007ed80687b60afb16fe27cc2256080053c39f4190e77856

data/lib/audio_analysis.rb

@@ -1,19 +1,25 @@
 class AudioAnalysis
   def self.find_bpm(file_path)
-    `pip install -r lib/python_analysis/requirements.txt`
-    bpm = `python3 lib/python_analysis/bpm_analysis.py #{file_path}`
+    `pip install -r #{gem_path}/lib/python_analysis/requirements.txt`
+    bpm = `python3 #{gem_path}/lib/python_analysis/bpm_analysis.py #{file_path}`
     bpm.to_f
   end
 
   def self.find_key(file_path)
-    `pip install -r lib/python_analysis/requirements.txt`
-    `python3 lib/python_analysis/key_analysis.py #{file_path}`
+    `pip install -r #{gem_path}/lib/python_analysis/requirements.txt`
+    `python3 #{gem_path}/lib/python_analysis/key_analysis.py #{file_path}`
   end
 
   def self.analyze(file_path)
-    `pip install -r lib/python_analysis/requirements.txt`
-    bpm = `python3 lib/python_analysis/bpm_analysis.py #{file_path}`
-    key =`python3 lib/python_analysis/key_analysis.py #{file_path}`
+    `pip install -r #{gem_path}/lib/python_analysis/requirements.txt`
+    bpm = `python3 #{gem_path}/lib/python_analysis/bpm_analysis.py #{file_path}`
+    key =`python3 #{gem_path}/lib/python_analysis/key_analysis.py #{file_path}`
     { bpm: bpm, key: key }
   end
+
+  private
+
+  def self.gem_path
+    Gem.loaded_specs["audio_analysis"].gem_dir
+  end
 end

data/lib/python_analysis/bpm_analysis.py

@@ -0,0 +1,14 @@
+import librosa
+import sys
+
+if len(sys.argv) != 2:
+    print("You have to pass in a file path `python3 analysis.py path_to_file.mp3`")
+    sys.exit(1)
+
+audio_file = sys.argv[1]
+y, sr = librosa.load(audio_file)
+
+# Estimate tempo (BPM)
+tempo, _ = librosa.beat.beat_track(y=y, sr=sr)
+
+print(tempo, end='')

data/lib/python_analysis/key_analysis.py

@@ -0,0 +1,27 @@
+import librosa
+import numpy as np
+import sys
+
+if len(sys.argv) != 2:
+    print("You have to pass in a file path `python3 analysis.py path_to_file.mp3`")
+    sys.exit(1)
+
+audio_file = sys.argv[1]
+
+y, sr = librosa.load(audio_file)
+
+# Compute the Chroma Short-Time Fourier Transform (chroma_stft)
+chromagram = librosa.feature.chroma_stft(y=y, sr=sr)
+
+# Calculate the mean chroma feature across time
+mean_chroma = np.mean(chromagram, axis=1)
+
+# Define the mapping of chroma features to keys
+chroma_to_key = ['C', 'C#', 'D', 'D#', 'E', 'F', 'F#', 'G', 'G#', 'A', 'A#', 'B']
+
+# Find the key by selecting the maximum chroma feature
+estimated_key_index = np.argmax(mean_chroma)
+estimated_key = chroma_to_key[estimated_key_index]
+
+# Print the detected key
+print(estimated_key, end='')

data/lib/python_analysis/requirements.txt

@@ -0,0 +1 @@
+librosa==0.10.1

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: audio_analysis
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.5
 platform: ruby
 authors:
 - 'Indigo Tech Tutorials '
@@ -17,6 +17,9 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/audio_analysis.rb
+- lib/python_analysis/bpm_analysis.py
+- lib/python_analysis/key_analysis.py
+- lib/python_analysis/requirements.txt
 homepage: https://rubygems.org/gems/audio_analysis
 licenses:
 - MIT