modusa 0.4.21__py3-none-any.whl → 0.4.23__py3-none-any.whl

modusa/__init__.py

@@ -8,4 +8,4 @@ from modusa.tools import play, convert, record, save
 from modusa.tools import download
 from modusa.tools import load, load_ann
 
-__version__ = "0.4.21"
+__version__ = "0.4.23"

modusa/tools/audio_loader.py

@@ -1,109 +1,155 @@
 #!/usr/bin/env python3
 
 
-import soundfile as sf
-from scipy.signal import resample
+import subprocess
 import numpy as np
+import imageio_ffmpeg as ffmpeg
 from pathlib import Path
-import tempfile
-from scipy.signal import resample
-from .youtube_downloader import download
-from .audio_converter import convert
+import re
 
-def load(path, sr=None, trim=None, mono=True):
+def _get_audio_info_ffmpeg(path: Path):
 	"""
-	Loads audio file from various sources.
+	To get the original sampling rate and number of
+	channels of a given audio file by parsing the
+	metadata. (No extra tool required).
 
-	.. code-block:: python
-		
-		import modusa as ms
-		audio_fp = ms.load(
-			"https://www.youtube.com/watch?v=lIpw9-Y_N0g",
-			sr = None, trim=(5, 10))
+	Parameters
+	----------
+	audiofp: PathLike
+		- Audio filepath
+	
+	Returns
+	-------
+	int
+		- Original sampling rate (hz)
+	int
+		- Number of channels
+	"""
+	ffmpeg_exe = ffmpeg.get_ffmpeg_exe()
+	cmd = [ffmpeg_exe, "-i", str(path)]
+	proc = subprocess.run(cmd, stderr=subprocess.PIPE, text=True)
+	text = proc.stderr
+	
+	# Example parse: "Stream #0:0: Audio: mp3, 44100 Hz, stereo, ..."
+	m = re.search(r'Audio:.*?(\d+)\s*Hz.*?(mono|stereo)', text)
+	if not m:
+		raise RuntimeError("Could not parse audio info")
+	sr = int(m.group(1))
+	channels = 1 if m.group(2) == "mono" else 2
+	return sr, channels
+
+def _load_audio_from_youtube(url: str):
+	"""
+	Download audio from a YouTube URL, convert it to WAV, and return the path.
 
 	Parameters
 	----------
-	path: str
-		- Path to the audio file.
-		- YouTube URL.
-	sr: int | None
+	url : str
+		YouTube video URL.
+
+	Returns
+	-------
+	Path
+		Path to the converted WAV file (you can delete it later).
+	"""
+	from modusa.tools.youtube_downloader import download
+	from modusa.tools.audio_converter import convert
+	import tempfile
+	
+	# Temporary directory to hold files (auto-created, not auto-deleted)
+	tmpdir = Path(tempfile.mkdtemp())
+	
+	# Download YouTube audio (e.g. .m4a or .webm)
+	audio_fp: Path = download(url=url, content_type="audio", output_dir=tmpdir)
+	
+	# Convert downloaded file to .wav
+	wav_audio_fp: Path = convert(inp_audio_fp=audio_fp, output_audio_fp=audio_fp.with_suffix(".wav"))
+	
+	# Return path to the WAV file
+	return wav_audio_fp
+
+#---------------------
+# Main Function
+#---------------------
+def load(path, sr=None, trim=None, ch=None):
+	"""
+	Lightweight audio loader using imageio-ffmpeg.
+
+	Parameters
+	----------
+	path: PathLike/str/URL
+		- Path to the audio file / YouTube video
+	sr: int
 		- Sampling rate to load the audio in.
-	trim: number | tuple[number, number] | None
-		- Segment of the audio to load.
-		- Example: 10 => First 10 seconds, (5, 10) => 5 to 10 seconds.
-		- Default: None => Entire audio.
-	mono: bool
-		- If True, loads the signal in mono.
+		- Default: None => Use the original sampling rate
+	trim: tuple[number, number]
+		- (start, end) in seconds to trim the audio clip.
+		- Default: None => No trimming
+	ch: int
+		- 1 for mono and 2 for stereo
+		- Default: None => Use the original number of channels.
 
-	Return
-	------
+	Returns
+	-------
 	np.ndarray
-		- Audio signal.
-	int
-		- Sampling rate of the loaded audio signal.
-	title
-		- Title of the loaded audio.
-		- Filename without extension or YouTube title.
+		- Audio signal Float32 waveform in [-1, 1].
+	int:
+		Sampling rate.
+	str:
+		File name stem.
 	"""
-	# Check if the path is YouTube
-	if ".youtube." in str(path):
-		# Download the audio in temp directory using tempfile module
-		with tempfile.TemporaryDirectory() as tmpdir:
-			# Download
-			audio_fp: Path = download(url=path, content_type="audio", output_dir=Path(tmpdir))
-			
-			# Convert the audio to ".wav" form for loading
-			wav_audio_fp: Path = convert(inp_audio_fp=audio_fp, output_audio_fp=audio_fp.with_suffix(".wav"))
-			
-			# Load the audio in memory
-			audio_data, audio_sr = sf.read(wav_audio_fp)
-			title = audio_fp.stem
-	else:
-		# Check if the file exists
-		fp = Path(path)
-		
-		if not fp.exists():
-			raise FileNotFoundError(f"{path} does not exist.")
-			
-		# Load the audio in memory
-		audio_data, audio_sr = sf.read(fp)
-		title = fp.stem
+	path = Path(path)
+	ffmpeg_exe = ffmpeg.get_ffmpeg_exe()
+	
+	yt = False # Is the path a youtube URL
+	
+	if ".youtube" in str(path):
+		yt = True
+		try:
+			path: Path = _load_audio_from_youtube(url=str(path))
+		except Exception as e:
+			raise ConnectionRefusedError("unable to download from YouTube")
+	
+	# Find the real sample rate from the file
+	if sr is None:
+		sr, _ = _get_audio_info_ffmpeg(path)
+		if not (sr > 100 and sr < 80000):
+			raise Exception(f"Error reading the metadata for original sampling rate {sr}, please set `sr` explicitly")
+	
+	# Find the real number of channels from the file
+	if ch is None:
+		_, ch = _get_audio_info_ffmpeg(path)
 		
-	# Convert to mono if requested and it's multi-channel
-	if mono and audio_data.ndim > 1:
-		audio_data = audio_data.mean(axis=1)
+		if ch not in [1, 2]:
+			raise Exception(f"Error reading the metadata for number of channels {ch}, please set `ch` explicitly")
 		
-	# Resample if needed
-	if sr is not None and audio_sr != sr:
-		n_samples = int(len(audio_data) * sr / audio_sr)
+	cmd = [ffmpeg_exe]
+	
+	# Optional trimming
+	if trim is not None:
+		start, end = trim
+		duration = end - start
+		cmd += ["-ss", str(start), "-t", str(duration)]
 		
-		if audio_data.ndim == 1:
-			# Mono
-			audio_data = resample(audio_data, n_samples)
-		else:
-			# Stereo or multi-channel: resample each channel independently
-			audio_data = np.stack([
-				resample(audio_data[:, ch], n_samples)
-				for ch in range(audio_data.shape[1])
-			], axis=1)
-			
-		audio_sr = sr
+	cmd += ["-i", str(path), "-f", "s16le", "-acodec", "pcm_s16le"]
+	cmd += ["-ar", str(sr)]
+	cmd += ["-ac", str(ch)]
 		
-	# Trim if requested
-	if trim is not None:
-		if isinstance(trim, (int, float)):
-			trim = (0, trim)
-		elif isinstance(trim, tuple) and len(trim) > 1:
-			trim = (trim[0], trim[1])
-		else:
-			raise ValueError(f"Invalid trim type or length: {type(trim)}, len={len(trim)}")
-			
-		start = int(trim[0] * audio_sr)
-		end = int(trim[1] * audio_sr)
-		audio_data = audio_data[start:end]
+	cmd += ["-"]
+	
+	proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.DEVNULL)
+	raw = proc.stdout.read()
+	proc.wait()
+	
+	audio = np.frombuffer(raw, np.int16).astype(np.float32) / 32768.0
+	
+	# Stereo reshaping if forced
+	if ch == 2:
+		audio = audio.reshape(-1, 2).T
 		
-	# Clip to avoid out-of-range playback issues
-	if np.issubdtype(audio_data.dtype, np.floating):
-		audio_data = np.clip(audio_data, -1.0, 1.0)
+	# Delete the file if downloaded from youtube
+	if yt:
+		path.unlink(missing_ok=True)
+		path.parent.rmdir()
 		
-	return audio_data.T, audio_sr, title
+	return audio, sr, path.stem

{modusa-0.4.21.dist-info → modusa-0.4.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: modusa
-Version: 0.4.21
+Version: 0.4.23
 Summary: A modular signal analysis python library.
 Author-Email: Ankit Anand <ankit0.anand0@gmail.com>
 License: MIT
@@ -9,10 +9,8 @@ Requires-Dist: numpy>=2.2.6
 Requires-Dist: matplotlib>=3.10.3
 Requires-Dist: yt-dlp==2025.9.23
 Requires-Dist: IPython>=9.5.0
-Requires-Dist: sounddevice>=0.5.2
 Requires-Dist: ipywidgets>=8.1.7
-Requires-Dist: soundfile>=0.13.1
-Requires-Dist: scipy>=1.16.2
+Requires-Dist: imageio-ffmpeg>=0.6.0
 Description-Content-Type: text/markdown
 
 # modusa

{modusa-0.4.21.dist-info → modusa-0.4.23.dist-info}/RECORD

@@ -1,8 +1,9 @@
-modusa-0.4.21.dist-info/METADATA,sha256=Rlr19wBffotWSOhku0JBPBSn91tUW7mRKSCxVsm7pMg,1467
-modusa-0.4.21.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-modusa-0.4.21.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
-modusa-0.4.21.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
-modusa/__init__.py,sha256=PwHhnV8JyHP8unMnnTXjJmqQtd1E-b4vUbuo6m6n_tw,324
+modusa-0.4.23.dist-info/METADATA,sha256=VcG0sWHvNgl2n0jMCiKiWM2BAeGHUJorSp0pFogxx90,1408
+modusa-0.4.23.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
+modusa-0.4.23.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
+modusa-0.4.23.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
+modusa/.DS_Store,sha256=hHecruWmea5u7ITMSdq62sVVdvT2I5lTN20gTfD3Btk,8196
+modusa/__init__.py,sha256=90isCiArafICPkaTgChbljh2F5o_iDKhqMz1gArlDOE,324
 modusa/config.py,sha256=bTqK4t00FZqERVITrxW_q284aDDJAa9aMSfFknfR-oU,280
 modusa/decorators.py,sha256=8zeNX_wE37O6Vp0ysR4-WCZaEL8mq8dyCF_I5DHOzks,5905
 modusa/devtools/generate_docs_source.py,sha256=UDflHsk-Yh9-3YJTVBzKL32y8hcxiRgAlFEBTMiDqwM,3301
@@ -44,7 +45,7 @@ modusa/tools/__init__.py,sha256=S7P1uYckyUdkha2UX9uj4P7mqpF6cc5SHqiW6NEupgs,342
 modusa/tools/_plotter_old.py,sha256=KGow7mihA2H1WNq7s5bpivhCgGo2qVIeDaO6iabpsrg,19294
 modusa/tools/ann_loader.py,sha256=m6Qu6jXnQ8LfUhKItoHSaHlGxUyzUJlGEyu4_50qJ8w,3099
 modusa/tools/audio_converter.py,sha256=415qBoPm2sBIuBSI7m1XBKm0AbmVmPydIPPr-uO8D3c,1778
-modusa/tools/audio_loader.py,sha256=vzoEIhuWedHcMbqxMXQkjiUAvkpXh5vTsR-L-NnFjHM,2928
+modusa/tools/audio_loader.py,sha256=nQl-E8xM1wdoYzZ3yGw25FJY8EogbqIZzPSMFt4Fv1E,3899
 modusa/tools/audio_player.py,sha256=kyBUnodkOE9Ox-hKHkfPeGAQ1RPTddbZYXO1ezz6-9w,2494
 modusa/tools/audio_recorder.py,sha256=K_LGqsPdjTdf3figEZTSQLmgMzYWgz18HTO8C1j5fE4,2788
 modusa/tools/audio_saver.py,sha256=ldzfr_AydsHTnKbxmBLJblN-hLzTmOlppOm306xI4Ug,510
@@ -58,4 +59,4 @@ modusa/utils/excp.py,sha256=L9vhaGjKpv9viJYdmC9n5ndmk2GVbUBuFyZyhAQZmWY,906
 modusa/utils/logger.py,sha256=K0rsnObeNKCxlNeSnVnJeRhgfmob6riB2uyU7h3dDmA,571
 modusa/utils/np_func_cat.py,sha256=TyIFgRc6bARRMDnZxlVURO5Z0I-GWhxRONYyIv-Vwxs,1007
 modusa/utils/plot.py,sha256=s_vNdxvKfwxEngvJPgrF1PcmxZNnNaaXPViHWjyjJ-c,5335
-modusa-0.4.21.dist-info/RECORD,,
+modusa-0.4.23.dist-info/RECORD,,