modusa 0.4.20__py3-none-any.whl → 0.4.21__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.20"
+__version__ = "0.4.21"

modusa/tools/ann_loader.py

@@ -6,7 +6,7 @@
 # Email: ankit0.anand0@gmail.com
 #---------------------------------
 
-def load_ann(path, clip=None):
+def load_ann(path, trim=None):
 	"""
 	Load annotation from audatity label
 	text file and also ctm file.
@@ -15,9 +15,9 @@ def load_ann(path, clip=None):
 	----------
 	path: str
 		- label text/ctm file path.
-	clip: tuple[number, number] | number | None
-		- Incase you clipped the audio signal, this parameter will help clip the annotation.
-		- If you clip the audio, say from (10, 20), set the clip to (10, 20).
+	trim: tuple[number, number] | number | None
+		- Incase you trimmed the audio signal, this parameter will help clip the annotation making sure that the timings are aligned to the trimmed audio.
+		- If you trimmed the audio, say from (10, 20), set the trim to (10, 20).
 		- Default: None
 
 	Returns
@@ -42,14 +42,14 @@ def load_ann(path, clip=None):
 	ann = [] # This will store the annotation
 	
 	# Clipping the annotation to match with the clipped audio
-	if clip is not None:
+	if trim is not None:
 		# Map clip input to the right format
-		if isinstance(clip, int or float):
-			clip = (0, clip)
-		elif isinstance(clip, tuple) and len(clip) > 1:
-			clip = (clip[0], clip[1])
+		if isinstance(trim, int or float):
+			trim = (0, trim)
+		elif isinstance(trim, tuple) and len(trim) > 1:
+			trim = (trim[0], trim[1])
 		else:
-			raise ValueError(f"Invalid clip type or length: {type(clip)}, len={len(clip)}")
+			raise ValueError(f"Invalid clip type or length: {type(trim)}, len={len(trim)}")
 	
 	if path.suffix == ".txt":
 		with open(str(path), "r") as f:
@@ -60,11 +60,11 @@ def load_ann(path, clip=None):
 				
 				# Incase user has clipped the audio signal, we adjust the annotation
 				# to match the clipped audio
-				if clip is not None:
-					offset = clip[0]
+				if trim is not None:
+					offset = trim[0]
 					# Clamp annotation to clip boundaries
-					new_start = max(start, clip[0]) - offset
-					new_end   = min(end, clip[1]) - offset
+					new_start = max(start, trim[0]) - offset
+					new_end   = min(end, trim[1]) - offset
 					
 					# only keep if there's still overlap
 					if new_start < new_end:
@@ -89,11 +89,11 @@ def load_ann(path, clip=None):
 				
 				# Incase user has clipped the audio signal, we adjust the annotation
 				# to match the clipped audio
-				if clip is not None:
-					offset = clip[0]
+				if trim is not None:
+					offset = trim[0]
 					# Clamp annotation to clip boundaries
-					new_start = max(start, clip[0]) - offset
-					new_end   = min(end, clip[1]) - offset
+					new_start = max(start, trim[0]) - offset
+					new_end   = min(end, trim[1]) - offset
 					
 					# only keep if there's still overlap
 					if new_start < new_end:

modusa/tools/audio_loader.py

@@ -3,14 +3,14 @@
 
 import soundfile as sf
 from scipy.signal import resample
+import numpy as np
 from pathlib import Path
 import tempfile
 from scipy.signal import resample
 from .youtube_downloader import download
 from .audio_converter import convert
 
-
-def load(path, sr=None, clip=None):
+def load(path, sr=None, trim=None, mono=True):
 	"""
 	Loads audio file from various sources.
 
@@ -19,19 +19,21 @@ def load(path, sr=None, clip=None):
 		import modusa as ms
 		audio_fp = ms.load(
 			"https://www.youtube.com/watch?v=lIpw9-Y_N0g",
-			sr = None, clip=(5, 10))
+			sr = None, trim=(5, 10))
 
 	Parameters
 	----------
 	path: str
-		- Path to the audio
-		- Youtube URL
+		- Path to the audio file.
+		- YouTube URL.
 	sr: int | None
 		- Sampling rate to load the audio in.
-	clip: number | tuple[number, number] | None
-		- Which segment of the audio you want.
-		- Eg., 10 => First 10 sec, (5, 10) => 5 to 10 second
+	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.
 
 	Return
 	------
@@ -43,7 +45,6 @@ def load(path, sr=None, clip=None):
 		- Title of the loaded audio.
 		- Filename without extension or YouTube title.
 	"""
-	
 	# Check if the path is YouTube
 	if ".youtube." in str(path):
 		# Download the audio in temp directory using tempfile module
@@ -57,50 +58,52 @@ def load(path, sr=None, clip=None):
 			# Load the audio in memory
 			audio_data, audio_sr = sf.read(wav_audio_fp)
 			title = audio_fp.stem
-			
-			# Convert to mono if it's multi-channel
-			if audio_data.ndim > 1:
-				audio_data = audio_data.mean(axis=1)
-				
-			# Resample if needed
-			if sr is not None:
-				if audio_sr != sr:
-					n_samples = int(len(audio_data) * sr / audio_sr)
-					audio_data = resample(audio_data, n_samples)
-					audio_sr = sr
-		
 	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
 		
-		# Convert to mono if it's multi-channel
-		if audio_data.ndim > 1:
-			audio_data = audio_data.mean(axis=1)
+	# Convert to mono if requested and it's multi-channel
+	if mono and audio_data.ndim > 1:
+		audio_data = audio_data.mean(axis=1)
+		
+	# Resample if needed
+	if sr is not None and audio_sr != sr:
+		n_samples = int(len(audio_data) * sr / audio_sr)
+		
+		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)
 			
-		# Resample if needed
-		if sr is not None:
-			if audio_sr != sr:
-				n_samples = int(len(audio_data) * sr / audio_sr)
-				audio_data = resample(audio_data, n_samples)
-				audio_sr = sr
-				
-	# Clip the audio signal as per needed
-	if clip is not None:
-		# Map clip input to the right format
-		if isinstance(clip, int or float):
-			clip = (0, clip)
-		elif isinstance(clip, tuple) and len(clip) > 1:
-			clip = (clip[0], clip[1])
+		audio_sr = sr
+		
+	# 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 clip type or length: {type(clip)}, len={len(clip)}")
+			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]
+		
+	# 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)
 		
-		audio_data = audio_data[int(clip[0]*audio_sr):int(clip[1]*audio_sr)]
-	
-	return audio_data, audio_sr, title
+	return audio_data.T, audio_sr, title

modusa/tools/audio_player.py

@@ -5,17 +5,27 @@ from pathlib import Path
 import numpy as np
 import base64
 
-def play(
-	y: np.ndarray,
-	sr: float,
-	clip: tuple[float, float] | None = None,
-	label: str | None = None,
-) -> None:
+def play(y, sr: float, clip=None, label=None):
 	"""
-	Audio player with optional clip selection, transcription-style label,
-	and an embedded bottom-right logo (../images/icon.png).
+	Audio player with optional clip selection, transcription-style label.
+
+	Parameters
+	----------
+	y: ndarray
+		- Audio signal.
+	sr: float
+		- Sampling rate.
+	clip: tuple[float, float] | None
+		- The portion from the audio signal to be played.
+	label: str | None
+		- Could be transcription/labels attached to the audio.
+	
+	Returns
+	-------
+	None
 	"""
-	start_time, end_time = 0.0, len(y) / sr
+	start_time = 0.0
+	end_time = len(y) / sr if y.ndim < 2 else y[0].size / sr
 	
 	# Optional clip selection
 	if clip is not None:
@@ -84,7 +94,7 @@ def play(
 		box-shadow:0 1px 3px rgba(0,0,0,0.05);
 	">
 		{label_html}
-		<div style="margin-top:10px;">
+		<div style="margin-top:10px; margin-bottom:10px">
 			{audio_html}
 		</div>
 		{logo_html}

modusa/tools/plotter.py

@@ -209,6 +209,10 @@ class Fig:
 		
 		axs[-1, 0].tick_params(bottom=True, labelbottom=True)
 		
+		# Add the figure title on top-left (if any)
+		if self._fig_num is not None:
+			fig.suptitle(f'fig - {self._fig_num}', fontsize=12, fontweight='bold', x=0.01, ha='left', va='top', y=0.98)
+			
 		# xlim should be applied on reference subplot, rest all subplots will automatically adjust
 		if xlim is not None:
 			axs[0, 0].set_xlim(xlim)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: modusa
-Version: 0.4.20
+Version: 0.4.21
 Summary: A modular signal analysis python library.
 Author-Email: Ankit Anand <ankit0.anand0@gmail.com>
 License: MIT

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

@@ -1,8 +1,8 @@
-modusa-0.4.20.dist-info/METADATA,sha256=pJ932mbsVsJ8XTbIHPX3h1RZa5k-GUBk0i5zgsqGMKY,1467
-modusa-0.4.20.dist-info/WHEEL,sha256=9P2ygRxDrTJz3gsagc0Z96ukrxjr-LFBGOgv3AuKlCA,90
-modusa-0.4.20.dist-info/entry_points.txt,sha256=fmKpleVXj6CdaBVL14WoEy6xx7JQCs85jvzwTi3lePM,73
-modusa-0.4.20.dist-info/licenses/LICENSE.md,sha256=JTaXAjx5awk76VArKCx5dUW8vmLEWsL_ZlR7-umaHbA,1078
-modusa/__init__.py,sha256=x95HzUy8e7c6XeEIZsuy6QmE82fcfJsnWvTUm8u6WyE,324
+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/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
@@ -42,15 +42,15 @@ modusa/plugins/__init__.py,sha256=r1Bf5mnrVKRIwxboutY1iGzDy4EPQhqpk1kSW7iJj_Q,54
 modusa/plugins/base.py,sha256=Bh_1Bja7fOymFsCgwhXDbV6ys3D8muNrPwrfDrG_G_A,2382
 modusa/tools/__init__.py,sha256=S7P1uYckyUdkha2UX9uj4P7mqpF6cc5SHqiW6NEupgs,342
 modusa/tools/_plotter_old.py,sha256=KGow7mihA2H1WNq7s5bpivhCgGo2qVIeDaO6iabpsrg,19294
-modusa/tools/ann_loader.py,sha256=BEdwAh_lccx_SnAO3bNMeY3O5zGiJlH2o4snWmXj8eQ,3034
+modusa/tools/ann_loader.py,sha256=m6Qu6jXnQ8LfUhKItoHSaHlGxUyzUJlGEyu4_50qJ8w,3099
 modusa/tools/audio_converter.py,sha256=415qBoPm2sBIuBSI7m1XBKm0AbmVmPydIPPr-uO8D3c,1778
-modusa/tools/audio_loader.py,sha256=n9Q9t_GmlE8AtioVwRcXX3rnd6PkbGTx-hAoNgUnNOQ,2780
-modusa/tools/audio_player.py,sha256=BI1ZmETxnmJACDHZMsbplzspwAx-_oKe3SpaHz-MFds,2295
+modusa/tools/audio_loader.py,sha256=vzoEIhuWedHcMbqxMXQkjiUAvkpXh5vTsR-L-NnFjHM,2928
+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
 modusa/tools/base.py,sha256=C0ESJ0mIfjjRlAkRbSetNtMoOfS6IrHBjexRp3l_Mh4,1293
 modusa/tools/math_ops.py,sha256=ZZ7U4DgqT7cOeE7_Lzi_Qq-48WYfwR9_osbZwTmE9eg,8690
-modusa/tools/plotter.py,sha256=V0CcRnXkE4tkTZQe6v0LI0ucEagqeudOZ-xoQw2clTc,30667
+modusa/tools/plotter.py,sha256=RrELZYgnyUVhGzcNfpTvyZvVTmZ6dvQmN9YSQoUjmaM,30860
 modusa/tools/youtube_downloader.py,sha256=hB_X8-7nOHXOlxg6vv3wyhBLoAsWyomrULP6_uCQL7s,1698
 modusa/utils/__init__.py,sha256=1oLL20yLB1GL9IbFiZD8OReDqiCpFr-yetIR6x1cNkI,23
 modusa/utils/config.py,sha256=cuGbqbovx5WDQq5rw3hIKcv3CnE5NttjacSOWnP1yu4,576
@@ -58,4 +58,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.20.dist-info/RECORD,,
+modusa-0.4.21.dist-info/RECORD,,