Fix overlap and unrecognized audio format error (#25)

* Fix overlap and unrecognized audio format error.

* Simplify channel

* Update pyproject.toml

pyproject.toml

@@ -4,10 +4,10 @@ version = "0.5.3"
 description = "CLI app to manage, search, and play collections of mp3 tracks."
 authors = ["Kinuax <[email protected]>"]
 license = "MIT"
-readme = "README.rst"
-homepage = "https://github.com/kinuax/rolabesti"
-repository = "https://github.com/kinuax/rolabesti"
-documentation = "https://github.com/kinuax/rolabesti"
+readme = "README.md"
+homepage = "https://github.com/kinuax/rolabesti/"
+repository = "https://github.com/kinuax/rolabesti/"
+documentation = "https://github.com/kinuax/rolabesti/"
 keywords = ["mp3", "id3", "vlc", "mongo"]
 classifiers = [
     "Development Status :: 4 - Beta",
@@ -34,6 +34,7 @@ mutagen = "1.47.0"
 platformdirs = "4.2.0"
 pydantic = "2.7.0"
 pydantic-settings = "2.2.1"
+pydub = "0.25.1"
 pygame = "2.5.2"
 pymongo = "4.6.2"
 tinydb = "4.8.0"

rolabesti/views/tracklist.py

@@ -5,7 +5,7 @@
 
 import typer
 
-from .utils import play_audio
+from .utils import play_mp3
 from rolabesti.models import FIELD_FILTERS, Track
 from rolabesti.logger import Logger
 from rolabesti.utils import length_to_string
@@ -37,7 +37,7 @@ def play(self, cli: bool, overlap_length: int) -> None:
         if cli:
             for i, track in enumerate(self.tracks):
                 self.logger.log(f"[green]playing --> [/green]{track}")
-                play_audio(track.path)
+                play_mp3(track.path, i % 2)
 
                 # Adjust waiting_length. Filter out last track and too short tracks.
                 if i < self.count - 1 and overlap_length < track.length:

rolabesti/views/utils.py

@@ -1,6 +1,10 @@
+import tempfile
 from contextlib import contextmanager, redirect_stdout
 from pathlib import Path
 
+from pydub import AudioSegment
+
+
 try:
     from wurlitzer import pipes
 except ModuleNotFoundError:
@@ -15,10 +19,17 @@ def pipes():
     import pygame
 
 
-def play_audio(trackpath: Path) -> None:
-    """Play the audio file located at trackpath."""
+def play_mp3(trackpath: Path, channel: int) -> None:
+    """Play the track located at trackpath on the given channel."""
     pygame.mixer.init()
+    channel = pygame.mixer.Channel(channel)
     # Avoid messages from C libraries.
     with pipes():
-        pygame.mixer.music.load(trackpath)
-    pygame.mixer.music.play()
+        try:
+            sound = pygame.mixer.Sound(trackpath)
+        except pygame.error:
+            # Convert mp3 to wav if "Unrecognized audio format" error.
+            with tempfile.TemporaryFile() as wav_file:
+                AudioSegment.from_mp3(trackpath).export(wav_file, format="wav")
+                sound = pygame.mixer.Sound(wav_file)
+        channel.play(sound)