whisper-webui / src /source.py
aadnk's picture
Add support for multiple input files and output files
cb9ee50
raw
history blame
2.55 kB
# Gradio seems to truncate files without keeping the extension, so we need to truncate the file prefix ourself
import os
import pathlib
from typing import List
import zipfile
import ffmpeg
from more_itertools import unzip
from src.download import ExceededMaximumDuration, download_url
MAX_FILE_PREFIX_LENGTH = 17
class AudioSource:
def __init__(self, source_path, source_name = None):
self.source_path = source_path
self.source_name = source_name
# Load source name if not provided
if (self.source_name is None):
file_path = pathlib.Path(self.source_path)
self.source_name = file_path.name
def get_full_name(self):
return self.source_name
def get_short_name(self, max_length: int = MAX_FILE_PREFIX_LENGTH):
file_path = pathlib.Path(self.source_name)
short_name = file_path.stem[:max_length] + file_path.suffix
return short_name
def __str__(self) -> str:
return self.source_path
class AudioSourceCollection:
def __init__(self, sources: List[AudioSource]):
self.sources = sources
def __iter__(self):
return iter(self.sources)
def get_audio_source_collection(urlData: str, multipleFiles: List, microphoneData: str, input_audio_max_duration: float = -1) -> List[AudioSource]:
output: List[AudioSource] = []
if urlData:
# Download from YouTube. This could also be a playlist or a channel.
output.extend([ AudioSource(x) for x in download_url(urlData, input_audio_max_duration, playlistItems=None) ])
else:
# Add input files
if (multipleFiles is not None):
output.extend([ AudioSource(x.name) for x in multipleFiles ])
if (microphoneData is not None):
output.append(AudioSource(microphoneData))
total_duration = 0
# Calculate total audio length. We do this even if input_audio_max_duration
# is disabled to ensure that all the audio files are valid.
for source in output:
audioDuration = ffmpeg.probe(source.source_path)["format"]["duration"]
total_duration += float(audioDuration)
# Ensure the total duration of the audio is not too long
if input_audio_max_duration > 0:
if float(total_duration) > input_audio_max_duration:
raise ExceededMaximumDuration(videoDuration=total_duration, maxDuration=input_audio_max_duration, message="Video(s) is too long")
# Return a list of audio sources
return output