Skip to content

Commit

Permalink
Adding path to ffmpeg for installed app on windows (#942)
Browse files Browse the repository at this point in the history
  • Loading branch information
raivisdejus authored Oct 12, 2024
1 parent 724fdea commit 03e882e
Showing 1 changed file with 23 additions and 9 deletions.
32 changes: 23 additions & 9 deletions buzz/transcriber/file_transcriber.py
100644 → 100755
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import logging
import os
import subprocess
import shutil
import tempfile
from abc import abstractmethod
Expand All @@ -9,6 +10,7 @@
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,
Expand All @@ -32,31 +34,43 @@ def __init__(self, task: FileTranscriptionTask, parent: Optional["QObject"] = No
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:
Expand Down

0 comments on commit 03e882e

Please sign in to comment.