audio_analysis 0.0.4 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0a1c982a664f5cea0335d76a00b15057db67c92bbdf3a25f4972b201defb7409
-  data.tar.gz: 98884a8ab4ebc3b098e961b9c58f1261fa1d8f2df1a541d5491d1114765c0f61
+  metadata.gz: 29e393fe23792ab42e8be688b502b74ac6302b9bed789d38de45c3b9fef00ba1
+  data.tar.gz: 4b1ddf61484442ad4e5e2a5f40c841b64710d1d5a1cd7a6fe7b80da7cd744edc
 SHA512:
-  metadata.gz: 8fb0f096e78c10d32bb822e51ddd6bccd9e12874580c6a875056836f0f55d116551da7799bb69f4b8bc2b6cdc709b2e3bdd963ca233a60fca010d9858210dc6a
-  data.tar.gz: 80df53e6a53b1d4a74d3ae23469774994caaa01f2ff5d52e5104175a39c9ee1973b731f838ea64f580c6dddca419d3f5ee09f2718f38a452f66f718ca26e885d
+  metadata.gz: b8ad08d3049aa0a33e6c93d0c425ae486aa947fd9e2bc118bdba7ce5abfaa64b8b88627f8e72a52cb8e53bb10f779bbaea137ecb7911cd6fb4e57c4c48aa60ae
+  data.tar.gz: 6a4f9904f2187aad58cae81e5134c5fc7a3fbb6b58bde93c976900ef9913369bd1bd65910fb417fa6bf4df472b50414309a119e889b5490f2dcca4360c6cf198

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.6
 platform: ruby
 authors:
 - 'Indigo Tech Tutorials '
@@ -17,10 +17,14 @@ 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
-metadata: {}
+metadata:
+  source_code_uri: https://github.com/enochtamulonis/audio-analysis-gem
 post_install_message: 
 rdoc_options: []
 require_paths: