audio_analysis 0.0.3 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7c6ee2bf1738a0fefba9106383ac99014910171a22d0e34f86b72a201a3d3ae5
-  data.tar.gz: e5d37d7b4215d3772ce5c46b35e3424134bcdf1f6cf59896eb0380364a7031f0
+  metadata.gz: 77f3e6d28dab05d54e885f1d430c3e1034ad56c1ef318ffc7fcc01584022bdd3
+  data.tar.gz: 24a283dcf4579c575fcb82f3b661423a476fcd9df0ff9ab5163421fb9210f517
 SHA512:
-  metadata.gz: fce4fab5dc03e927721825c46999cb0d1e181ad4cd82086d917a69aeb577ce491bbaf4c3310d9849d8665c581059a355e3746ba3095a23b438489da233f3ec56
-  data.tar.gz: 3494f50f35eb17713fc752760a3625d58bdb0d48a94e350a3bfbc1cc939f667da73c4b0ac45fa02f37841371d5ab7e93d4486e721fbbdb2ccdaaf6a9dcef1f11
+  metadata.gz: 78c530c25407cc5975649e213043cecb66e8262e1c7b86c7a8fce6a8dbb6064ed2056ac0b5ab3aade7ed9d5f70faa4417d98921b21f07c42f5b801076c7c1733
+  data.tar.gz: d8caae585bd56c1266fe705f3c5538386dc59db3b45d9862238f3488c244edaa1c8b7c6e83991ab6007ed80687b60afb16fe27cc2256080053c39f4190e77856

data/lib/audio_analysis.rb

@@ -20,6 +20,6 @@ class AudioAnalysis
   private
 
   def self.gem_path
-    Gem.datadir("audio_analysis")
+    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.3
+  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