From 03e882eebfb35218bba7e866cb0d1c44d474ccaa Mon Sep 17 00:00:00 2001
From: Raivis Dejus <raivisd@scandiweb.com>
Date: Sat, 12 Oct 2024 12:20:14 +0300
Subject: [PATCH] Adding path to ffmpeg for installed app on windows (#942)

---
 buzz/transcriber/file_transcriber.py | 32 ++++++++++++++++++++--------
 1 file changed, 23 insertions(+), 9 deletions(-)
 mode change 100644 => 100755 buzz/transcriber/file_transcriber.py

diff --git a/buzz/transcriber/file_transcriber.py b/buzz/transcriber/file_transcriber.py
old mode 100644
new mode 100755
index 4aae5bd6..38ff393f
--- a/buzz/transcriber/file_transcriber.py
+++ b/buzz/transcriber/file_transcriber.py
@@ -1,5 +1,6 @@
 import logging
 import os
+import subprocess
 import shutil
 import tempfile
 from abc import abstractmethod
@@ -9,6 +10,7 @@ from PyQt6.QtCore import QObject, pyqtSignal, pyqtSlot
 from yt_dlp import YoutubeDL
 from yt_dlp.utils import DownloadError
 
+from buzz.whisper_audio import SAMPLE_RATE
 from buzz.transcriber.transcriber import (
     FileTranscriptionTask,
     get_output_file_path,
@@ -32,31 +34,43 @@ class FileTranscriber(QObject):
     def run(self):
         if self.transcription_task.source == FileTranscriptionTask.Source.URL_IMPORT:
             temp_output_path = tempfile.mktemp()
+            wav_file = temp_output_path + ".wav"
 
             ydl = YoutubeDL(
                 {
                     "format": "wav/bestaudio/best",
                     "progress_hooks": [self.on_download_progress],
                     "outtmpl": temp_output_path,
-                    "logger": logging.getLogger(),
-                    "postprocessors": [
-                        {
-                            "key": "FFmpegExtractAudio",
-                            "preferredcodec": "wav",
-                        }
-                    ],
+                    "logger": logging.getLogger()
                 }
             )
 
             try:
                 logging.debug(f"Downloading audio file from URL: {self.transcription_task.url}")
                 ydl.download([self.transcription_task.url])
-            except DownloadError as exc:
+            except Exception as exc:
                 logging.debug(f"Error downloading audio: {exc.msg}")
                 self.error.emit(exc.msg)
                 return
 
-            self.transcription_task.file_path = temp_output_path + ".wav"
+            cmd = [
+                "ffmpeg",
+                "-nostdin",
+                "-threads", "0",
+                "-f", "s16le",
+                "-ac", "1",
+                "-acodec", "pcm_s16le",
+                "-ar", str(SAMPLE_RATE),
+                "-loglevel", "error",
+                "-i", temp_output_path, wav_file]
+
+            try:
+                subprocess.run(cmd, capture_output=True, check=True)
+            except subprocess.CalledProcessError as exc:
+                logging.debug(f"Error processing downloaded audio: {exc.msg}")
+                raise Exception(exc.stderr.decode("utf-8"))
+
+            self.transcription_task.file_path = wav_file
             logging.debug(f"Downloaded audio to file: {self.transcription_task.file_path}")
 
         try: