Trying libav

2 files changed, 31 insertions, 2 deletions

diff --git a/src/model/youload_playlist.py b/src/model/youload_playlist.py
index 1856c64..9189522 100644
--- a/src/model/youload_playlist.py
+++ b/src/model/youload_playlist.py
@@ -1,6 +1,6 @@
 from pytube import Playlist
-import ffmpeg
 import os
+import util
 
 def make_alpha_numeric(string: str) -> str:
     return "".join(char for char in string if char.isalnum())
@@ -54,7 +54,7 @@ class YouLoadPlayList:
 
             # To mp3.
             file_path_mp3 = file_path[:-4] + ".mp3"
-            ffmpeg.input(file_path).output(file_path_mp3).run()
+            util.convert_mp4_to_mp3(file_path, file_path_mp3)
 
             # Remove mp4 file.
             os.remove(file_path)
diff --git a/src/util.py b/src/util.py
index 4b82c77..fd14e2e 100644
--- a/src/util.py
+++ b/src/util.py
@@ -1,4 +1,5 @@
 import os
+import av
 from kivy.utils import platform
 
 def get_default_download_dir() -> str:
@@ -7,3 +8,31 @@ def get_default_download_dir() -> str:
     else:
         from pathlib import Path
         return os.path.join(str(Path.home()), "Downloads")
+
+# Chatgpt code (:
+# I use av instead of ffmpeg so I can run it on android.
+def convert_mp4_to_mp3(input_file: str, output_file: str) -> None:
+    # Open the input video file
+    with av.open(input_file) as container:
+        # Find the audio stream
+        audio_stream = next(
+            (stream for stream in container.streams if stream.type == 'audio'),
+            None
+        )
+
+        if audio_stream is None:
+            raise ValueError("No audio stream found in the input file")
+
+        # Open an output file for writing
+        with av.open(output_file, 'w') as output_container:
+            # Create an audio stream in the output container
+            output_audio_stream = output_container.add_stream('mp3')
+
+            # Iterate through frames, decode audio, and encode into the output stream
+            for frame in container.decode(audio_stream):
+                # Encode the frame into the output stream
+                output_packet = output_audio_stream.encode(frame)
+
+                # Write the encoded packet to the output file
+                if output_packet:
+                    output_container.mux(output_packet)