Spaces:
Running
Running
| """ | |
| This module contains functions to manage audio files. | |
| """ | |
| import os | |
| import shutil | |
| from pathlib import PurePath | |
| import gradio as gr | |
| from backend.common import INTERMEDIATE_AUDIO_DIR, OUTPUT_AUDIO_DIR, display_progress | |
| from backend.exceptions import InputMissingError, InvalidPathError, PathNotFoundError | |
| from common import GRADIO_TEMP_DIR | |
| def get_output_audio() -> list[tuple[str, str]]: | |
| """ | |
| Get the name and path of all output audio files. | |
| Returns | |
| ------- | |
| list[tuple[str, str]] | |
| A list of tuples containing the name and path of each output audio file. | |
| """ | |
| if os.path.isdir(OUTPUT_AUDIO_DIR): | |
| named_output_files = [ | |
| (file_name, os.path.join(OUTPUT_AUDIO_DIR, file_name)) | |
| for file_name in os.listdir(OUTPUT_AUDIO_DIR) | |
| ] | |
| return sorted(named_output_files, key=lambda x: x[0]) | |
| return [] | |
| def delete_intermediate_audio( | |
| song_dirs: list[str], | |
| progress_bar: gr.Progress | None = None, | |
| percentage: float = 0.0, | |
| ) -> str: | |
| """ | |
| Delete intermediate audio files in provided song directories. | |
| Parameters | |
| ---------- | |
| song_dirs : list[str] | |
| Paths of song directories to delete intermediate audio files for. | |
| progress_bar : gr.Progress, optional | |
| Gradio progress bar to update. | |
| percentage : float, default=0.0 | |
| Percentage to display in the progress bar. | |
| Returns | |
| ------- | |
| str | |
| Success message. | |
| Raises | |
| ------ | |
| InputMissingError | |
| If no song directories are provided. | |
| PathNotFoundError | |
| If a song directory does not exist. | |
| InvalidPathError | |
| If a song directory is not located in the root of the intermediate audio directory. | |
| """ | |
| if not song_dirs: | |
| raise InputMissingError( | |
| "Song directories missing! Please provide a non-empty list of song" | |
| " directories." | |
| ) | |
| display_progress( | |
| "[~] Deleting intermediate audio files for selected songs...", | |
| percentage, | |
| progress_bar, | |
| ) | |
| for song_dir in song_dirs: | |
| if not os.path.isdir(song_dir): | |
| raise PathNotFoundError(f"Song directory '{song_dir}' does not exist.") | |
| if not PurePath(song_dir).parent == PurePath(INTERMEDIATE_AUDIO_DIR): | |
| raise InvalidPathError( | |
| f"Song directory '{song_dir}' is not located in the root of the" | |
| " intermediate audio directory." | |
| ) | |
| shutil.rmtree(song_dir) | |
| return "[+] Successfully deleted intermediate audio files for selected songs!" | |
| def delete_all_intermediate_audio( | |
| progress_bar: gr.Progress | None = None, percentage: float = 0.0 | |
| ) -> str: | |
| """ | |
| Delete all intermediate audio files. | |
| Parameters | |
| ---------- | |
| progress_bar : gr.Progress, optional | |
| Gradio progress bar to update. | |
| percentage : float, default=0.0 | |
| Returns | |
| ------- | |
| str | |
| Success message. | |
| """ | |
| display_progress( | |
| "[~] Deleting all intermediate audio files...", percentage, progress_bar | |
| ) | |
| if os.path.isdir(INTERMEDIATE_AUDIO_DIR): | |
| shutil.rmtree(INTERMEDIATE_AUDIO_DIR) | |
| return "[+] All intermediate audio files successfully deleted!" | |
| def delete_output_audio( | |
| output_audio_files: list[str], | |
| progress_bar: gr.Progress | None = None, | |
| percentage: float = 0.0, | |
| ) -> str: | |
| """ | |
| Delete selected output audio files. | |
| Parameters | |
| ---------- | |
| output_audio_files : list[str] | |
| Paths of output audio files to delete. | |
| progress_bar : gr.Progress, optional | |
| Gradio progress bar to update. | |
| percentage : float, default=0.0 | |
| Percentage to display in the progress bar. | |
| Returns | |
| ------- | |
| str | |
| Success message. | |
| Raises | |
| ------ | |
| InputMissingError | |
| If no output audio files are provided. | |
| PathNotFoundError | |
| If an output audio file does not exist. | |
| InvalidPathError | |
| If an output audio file is not located in the root of the output audio directory. | |
| """ | |
| if not output_audio_files: | |
| raise InputMissingError( | |
| "Output audio files missing! Please provide a non-empty list of output" | |
| " audio files." | |
| ) | |
| display_progress( | |
| "[~] Deleting selected output audio files...", percentage, progress_bar | |
| ) | |
| for output_audio_file in output_audio_files: | |
| if not os.path.isfile(output_audio_file): | |
| raise PathNotFoundError( | |
| f"Output audio file '{output_audio_file}' does not exist." | |
| ) | |
| if not PurePath(output_audio_file).parent == PurePath(OUTPUT_AUDIO_DIR): | |
| raise InvalidPathError( | |
| f"Output audio file '{output_audio_file}' is not located in the root of" | |
| " the output audio directory." | |
| ) | |
| os.remove(output_audio_file) | |
| return "[+] Successfully deleted selected output audio files!" | |
| def delete_all_output_audio( | |
| progress_bar: gr.Progress | None = None, percentage: float = 0.0 | |
| ) -> str: | |
| """ | |
| Delete all output audio files. | |
| Parameters | |
| ---------- | |
| progress_bar : gr.Progress, optional | |
| Gradio progress bar to update. | |
| percentage : float, default=0.0 | |
| Percentage to display in the progress bar. | |
| Returns | |
| ------- | |
| str | |
| Success message. | |
| """ | |
| display_progress("[~] Deleting all output audio files...", percentage, progress_bar) | |
| if os.path.isdir(OUTPUT_AUDIO_DIR): | |
| shutil.rmtree(OUTPUT_AUDIO_DIR) | |
| return "[+] All output audio files successfully deleted!" | |
| def delete_all_audio( | |
| progress_bar: gr.Progress | None = None, percentage: float = 0.0 | |
| ) -> str: | |
| """ | |
| Delete all audio files. | |
| Parameters | |
| ---------- | |
| progress_bar : gr.Progress, optional | |
| Gradio progress bar to update. | |
| percentage : float, default=0.0 | |
| Percentage to display in the progress bar. | |
| Returns | |
| ------- | |
| str | |
| Success message. | |
| """ | |
| display_progress("[~] Deleting all audio files...", percentage, progress_bar) | |
| if os.path.isdir(INTERMEDIATE_AUDIO_DIR): | |
| shutil.rmtree(INTERMEDIATE_AUDIO_DIR) | |
| if os.path.isdir(OUTPUT_AUDIO_DIR): | |
| shutil.rmtree(OUTPUT_AUDIO_DIR) | |
| return "[+] All audio files successfully deleted!" | |
| def delete_gradio_temp_dir() -> None: | |
| """ | |
| Delete the directory where Gradio stores temporary files. | |
| """ | |
| if os.path.isdir(GRADIO_TEMP_DIR): | |
| shutil.rmtree(GRADIO_TEMP_DIR) | |