| import gradio as gr | |
| import os, shutil | |
| '''import subprocess, os | |
| assets_folder = "assets" | |
| if not os.path.exists(assets_folder): | |
| os.makedirs(assets_folder) | |
| files = { | |
| "rmvpe/rmvpe.pt":"https://huggingface.co/Rejekts/project/resolve/main/rmvpe.pt", | |
| "hubert/hubert_base.pt":"https://huggingface.co/Rejekts/project/resolve/main/hubert_base.pt", | |
| "pretrained_v2/D40k.pth":"https://huggingface.co/Rejekts/project/resolve/main/D40k.pth", | |
| "pretrained_v2/G40k.pth":"https://huggingface.co/Rejekts/project/resolve/main/G40k.pth", | |
| "pretrained_v2/f0D40k.pth":"https://huggingface.co/Rejekts/project/resolve/main/f0D40k.pth", | |
| "pretrained_v2/f0G40k.pth":"https://huggingface.co/Rejekts/project/resolve/main/f0G40k.pth" | |
| } | |
| for file, link in files.items(): | |
| file_path = os.path.join(assets_folder, file) | |
| if not os.path.exists(file_path): | |
| try: | |
| subprocess.run(['wget', link, '-O', file_path], check=True) | |
| except subprocess.CalledProcessError as e: | |
| print(f"Error downloading {file}: {e}")''' | |
| def show_available(filepath): | |
| return os.listdir(filepath) | |
| def upload_file(file): | |
| audio_formats = ['.wav', '.mp3', '.ogg', '.flac', '.aac'] | |
| file_name, file_extension = os.path.splitext(file.name) | |
| if file_extension.lower() in audio_formats: | |
| shutil.move(file.name,'audios') | |
| elif file_extension.lower().endswith('.pth'): | |
| shutil.move(file.name,'assets/weights') | |
| elif file_extension.lower().endswith('.index'): | |
| shutil.move(file.name,'logs') | |
| else: | |
| print("Filetype not compatible") | |
| return {"choices":show_available('audios'),"__type__": "update"} | |
| with gr.Blocks() as app: | |
| with gr.Row(): | |
| dropbox = gr.File(label="Upload files") | |
| audio_picker = gr.Dropdown(label="",choices=show_available('audios')) | |
| dropbox.upload(fn=upload_file, inputs=[dropbox],outputs=[audio_picker]) | |
| app.launch() |