modusa 0.2.23__py3-none-any.whl → 0.3.1__py3-none-any.whl

modusa/tools/audio_loader.py

@@ -0,0 +1,90 @@
+#!/usr/bin/env python3
+
+
+import soundfile as sf
+from scipy.signal import resample
+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):
+	"""
+	Loads audio file from various sources.
+
+	.. code-block:: python
+		
+		import modusa as ms
+		audio_fp = ms.load(
+			"https://www.youtube.com/watch?v=lIpw9-Y_N0g",
+			sr = None)
+
+	Parameters
+	----------
+	path: str
+		- Path to the audio
+		- Youtube URL
+	sr: int
+		- Sampling rate to load the audio in.
+
+	Return
+	------
+	np.ndarray
+		- Audio signal.
+	int
+		- Sampling rate of the loaded audio signal.
+	title
+		- 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
+		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
+			
+			# 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)
+			
+		# 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
+	
+	return audio_data, audio_sr, title

modusa/tools/audio_player.py

@@ -1,83 +1,66 @@
 #!/usr/bin/env python3
 
-
-from modusa import excp
-from modusa.decorators import validate_args_type
-from modusa.tools.base import ModusaTool
-from IPython.display import display, HTML, Audio
 import numpy as np
+from IPython.display import display, HTML, Audio
 
-class AudioPlayer(ModusaTool):
-	"""
-	Provides audio player in the jupyter notebook environment.
-	"""
-	
-	#--------Meta Information----------
-	_name = "Audio Player"
-	_description = ""
-	_author_name = "Ankit Anand"
-	_author_email = "ankit0.anand0@gmail.com"
-	_created_at = "2025-07-08"
-	#----------------------------------
-	
-	@staticmethod
-	def play(
-		y: np.ndarray,
-		sr: int,
-		regions: list[tuple[float, float]] | None = None,
-		title: str | None = None
-	) -> None:
+def play(y: np.ndarray, sr: float, t0: float = 0.0, regions = None, title = None) -> None:
 		"""
 		Plays audio clips for given regions in Jupyter Notebooks.
 
 		Parameters
 		----------
 		y : np.ndarray
-			Audio time series.
-		sr : int
-			Sampling rate.
-		regions : list of (float, float), optional
-			Regions to extract and play (in seconds).
-		title : str, optional
-			Title to display above audio players.
+			- Audio data.
+			- Mono (1D) numpy array.
+		sr: float
+			- Sampling rate of the audio.
+		t0: float
+			- Starting timestamp, incase the audio is cropped
+			- Default: 0.0 → Starts from 0.0 sec
+		regions : list[tuple[float, float, str]] | tuple[float, float, str] | None
+			- Regions to extract and play (in sec), e.g. [(0, 10.2, "tag")]
+			- If there is only one region, a tuple should also work. e.g. (0, 10.2, "tag")
+			- Default: None → The entire song is selected.
+		title : str | None
+			- Title to display above audio players.
 
 		Returns
 		-------
 		None
 		"""
-		if not AudioPlayer._in_notebook():
-			return
-		
 		if title:
 			display(HTML(f"<h4>{title}</h4>"))
-		
-		clip_numbers = []
+			
+		clip_tags = []
 		timings = []
 		players = []
-		
-		if regions:
-			for i, (start_sec, end_sec) in enumerate(regions):
+	
+		if isinstance(regions, tuple): regions = [regions] # (10, 20, "Region 1") -> [(10, 20, "Region 1")]
+	
+		if regions is not None:
+			for region in regions:
+				assert len(region) == 3
 				
-				if start_sec is None:
-					start_sec = 0.0
-				if end_sec is None:
-					end_sec = y.shape[0] / sr
+				start_sec = region[0] - t0
+				end_sec = region[1] - t0
+				tag = region[2]
 				
-				start_sample = int(start_sec * sr)
-				end_sample = int(end_sec * sr)
-				clip = y[start_sample:end_sample]
-				audio_tag = Audio(data=clip, rate=sr)._repr_html_()
+				start_sample, end_sample = int(start_sec * sr), int(end_sec * sr)
+				clip = y[start_sample: end_sample]
+				audio_player = Audio(data=clip, rate=sr)._repr_html_()
 				
-				clip_numbers.append(f"<td style='text-align:center; border-right:1px solid #ccc; padding:6px;'>{i+1}</td>")
+				clip_tags.append(f"<td style='text-align:center; border-right:1px solid #ccc; padding:6px;'>{tag}</td>")
 				timings.append(f"<td style='text-align:center; border-right:1px solid #ccc; padding:6px;'>{start_sec:.2f}s → {end_sec:.2f}s</td>")
-				players.append(f"<td style='padding:6px;'>{audio_tag}</td>")
+				players.append(f"<td style='padding:6px;'>{audio_player}</td>")
 		else:
-			total_duration = len(y) / sr
-			audio_tag = Audio(data=y, rate=sr)._repr_html_()
+			audio_player = Audio(data=y, rate=sr)._repr_html_()
 			
-			clip_numbers.append(f"<td style='text-align:center; border-right:1px solid #ccc; padding:6px;'>1</td>")
-			timings.append(f"<td style='text-align:center; border-right:1px solid #ccc; padding:6px;'>0.00s → {total_duration:.2f}s</td>")
-			players.append(f"<td style='padding:6px;'>{audio_tag}</td>")
+			start_sec = t0
+			end_sec = t0 + y.shape[0] / sr
+			
+			clip_tags.append(f"<td style='text-align:center; border-right:1px solid #ccc; padding:6px;'>1</td>")
+			timings.append(f"<td style='text-align:center; border-right:1px solid #ccc; padding:6px;'>{start_sec:.2f}s → {end_sec:.2f}s</td>")
+			players.append(f"<td style='padding:6px;'>{audio_player}</td>")
 			
 		# Wrap rows in a table with border
 		table_html = f"""
@@ -85,7 +68,7 @@ class AudioPlayer(ModusaTool):
 			<table style="border-collapse:collapse;">
 				<tr style="background-color:#f2f2f2;">
 					<th style="text-align:left; padding:6px 12px;">Clip</th>
-					{''.join(clip_numbers)}
+					{''.join(clip_tags)}
 				</tr>
 				<tr style="background-color:#fcfcfc;">
 					<th style="text-align:left; padding:6px 12px;">Timing</th>
@@ -98,17 +81,9 @@ class AudioPlayer(ModusaTool):
 			</table>
 		</div>
 		"""
-		
+	
 		return HTML(table_html)
-		
-			
-	@staticmethod
-	def _in_notebook() -> bool:
-		try:
-			from IPython import get_ipython
-			shell = get_ipython()
-			return shell and shell.__class__.__name__ == "ZMQInteractiveShell"
-		except ImportError:
-			return False
+
 		
 		
+		

modusa/tools/math_ops.py

@@ -217,6 +217,30 @@ class MathOps(ModusaTool):
 			raise excp.InputError(f"can't find abs for `a`") from e
 		return result
 	
+	@staticmethod
+	def floor(a: Any) -> np.generic | np.ndarray:
+		try:
+			result = np.floor(a)
+		except Exception as e:
+			raise excp.InputError(f"can't find floor for `a`") from e
+		return result
+	
+	@staticmethod
+	def ceil(a: Any) -> np.generic | np.ndarray:
+		try:
+			result = np.ceil(a)
+		except Exception as e:
+			raise excp.InputError(f"can't find ceil for `a`") from e
+		return result
+	
+	@staticmethod
+	def round(a: Any) -> np.generic | np.ndarray:
+		try:
+			result = np.round(a)
+		except Exception as e:
+			raise excp.InputError(f"can't find round for `a`") from e
+		return result
+	
 	#------------------------------------
 	# TODO: Add shape-changing ops like 
 	# reshape, transpose, squeeze later
@@ -229,4 +253,83 @@ class MathOps(ModusaTool):
 		except Exception as e:
 			raise excp.InputError(f"can't reshape `a`") from e
 		return result
-	
+	
+	#------------------------------------
+	# Complex numbers operations
+	#------------------------------------
+	
+	@staticmethod
+	def real(a: Any) -> np.ndarray:
+		try:
+			result = np.real(a)
+		except Exception as e:
+			raise excp.InputError(f"can't find real for `a`") from e
+		return result
+	
+	@staticmethod
+	def imag(a: Any) -> np.ndarray:
+		try:
+			result = np.imag(a)
+		except Exception as e:
+			raise excp.InputError(f"can't find imag for `a`") from e
+		return result
+	
+	@staticmethod
+	def angle(a: Any) -> np.ndarray:
+		try:
+			result = np.angle(a)
+		except Exception as e:
+			raise excp.InputError(f"can't find angle for `a`") from e
+		return result
+	
+	#------------------------------------
+	# Comparison
+	#------------------------------------
+	
+	@staticmethod
+	def lt(a: Any, b: Any) -> np.ndarray:
+		try:
+			mask = a < b
+		except Exception as e:
+			raise excp.InputError(f"`a` and `b` can't be compared") from e
+		return mask
+	
+	@staticmethod
+	def le(a: Any, b: Any) -> np.ndarray:
+		try:
+			mask = a <= b
+		except Exception as e:
+			raise excp.InputError(f"`a` and `b` can't be compared") from e
+		return mask
+	
+	@staticmethod
+	def gt(a: Any, b: Any) -> np.ndarray:
+		try:
+			mask = a > b
+		except Exception as e:
+			raise excp.InputError(f"`a` and `b` can't be compared") from e
+		return mask
+	
+	@staticmethod
+	def ge(a: Any, b: Any) -> np.ndarray:
+		try:
+			mask = a >= b
+		except Exception as e:
+			raise excp.InputError(f"`a` and `b` can't be compared") from e
+		return mask
+	
+	@staticmethod
+	def eq(a: Any, b: Any) -> np.ndarray:
+		try:
+			mask = a == b
+		except Exception as e:
+			raise excp.InputError(f"`a` and `b` can't be compared") from e
+		return mask
+	
+	@staticmethod
+	def ne(a: Any, b: Any) -> np.ndarray:
+		try:
+			mask = a != b
+		except Exception as e:
+			raise excp.InputError(f"`a` and `b` can't be compared") from e
+		return mask