Update main.py

main.py

@@ -1,104 +1,258 @@
-import argparse
-import gc
-import hashlib
-import json
-import os
-import shlex
-import subprocess
-from contextlib import suppress
-from urllib.parse import urlparse, parse_qs
-
+import torch, os, traceback, sys, warnings, shutil, numpy as np
 import gradio as gr
 import librosa
-import numpy as np
-import soundfile as sf
-import sox
+import asyncio
+import rarfile
+import edge_tts
 import yt_dlp
-from pedalboard import Pedalboard, Reverb, Compressor, HighpassFilter
-from pedalboard.io import AudioFile
-from pydub import AudioSegment
-
-
-
-
-def get_youtube_video_id(url, ignore_playlist=True):
- """
- Examples:
- http://youtu.be/SA2iWivDJiE
- http://www.youtube.com/watch?v=_oPAwA_Udwc&feature=feedu
- http://www.youtube.com/embed/SA2iWivDJiE
- http://www.youtube.com/v/SA2iWivDJiE?version=3&hl=en_US
- """
- query = urlparse(url)
- if query.hostname == 'youtu.be':
- if query.path[1:] == 'watch':
- return query.query[2:]
- return query.path[1:]
-
- if query.hostname in {'www.youtube.com', 'youtube.com', 'music.youtube.com'}:
- if not ignore_playlist:
- # use case: get playlist id not current video in playlist
- with suppress(KeyError):
- return parse_qs(query.query)['list'][0]
- if query.path == '/watch':
- return parse_qs(query.query)['v'][0]
- if query.path[:7] == '/watch/':
- return query.path.split('/')[1]
- if query.path[:7] == '/embed/':
- return query.path.split('/')[2]
- if query.path[:3] == '/v/':
- return query.path.split('/')[2]
-
- # returns None for invalid YouTube url
- return None
-
-
-def yt_download(link):
- ydl_opts = {
- 'format': 'bestaudio',
- 'outtmpl': '%(title)s',
- 'nocheckcertificate': True,
- 'ignoreerrors': True,
- 'no_warnings': True,
- 'quiet': True,
- 'extractaudio': True,
- 'postprocessors': [{'key': 'FFmpegExtractAudio', 'preferredcodec': 'mp3'}],
- }
- with yt_dlp.YoutubeDL(ydl_opts) as ydl:
- result = ydl.extract_info(link, download=True)
- download_path = ydl.prepare_filename(result, outtmpl='%(title)s.mp3')
-
- return download_path
-
-
-def raise_exception(error_msg, is_webui):
- if is_webui:
- raise gr.Error(error_msg)
- else:
- raise Exception(error_msg)
-
-
-
-with gr.Blocks(theme=gr.themes.Soft) as app:
- gr.HTML("<h1> youtube downloader </h1>")
+import ffmpeg
+import gdown
+import subprocess
+import wave
+import soundfile as sf
+from scipy.io import wavfile
+from datetime import datetime
+from urllib.parse import urlparse
+from mega import Mega
 
+now_dir = os.getcwd()
+tmp = os.path.join(now_dir, "TEMP")
+shutil.rmtree(tmp, ignore_errors=True)
+os.makedirs(tmp, exist_ok=True)
+os.environ["TEMP"] = tmp
+from lib.infer_pack.models import (
+ SynthesizerTrnMs256NSFsid,
+ SynthesizerTrnMs256NSFsid_nono,
+ SynthesizerTrnMs768NSFsid,
+ SynthesizerTrnMs768NSFsid_nono,
+)
+from fairseq import checkpoint_utils
+from vc_infer_pipeline import VC
+from config import Config
+config = Config()
 
-def download_from_url(url, audio_name):
- ydl.download([url])
 
-def youtube_to_wav(url, audio_name):
- with yt_dlp.YoutubeDL(ydl_opts) as ydl:
- download_from_url(url, audio_name)
+def vc_single(
+ sid,
+ vc_audio_mode,
+ input_audio_path,
+ input_upload_audio,
+ vocal_audio,
+ tts_text,
+ tts_voice,
+ f0_up_key,
+ f0_file,
+ f0_method,
+ file_index,
+ index_rate,
+ filter_radius,
+ resample_sr,
+ rms_mix_rate,
+ protect
+): # spk_item, input_audio0, vc_transform0,f0_file,f0method0
+ global tgt_sr, net_g, vc, hubert_model, version, cpt
+ try:
+ logs = []
+ print(f"Converting...")
+ logs.append(f"Converting...")
+ yield "\n".join(logs), None
+ if vc_audio_mode == "Input path" or "Youtube" and input_audio_path != "":
+ audio, sr = librosa.load(input_audio_path, sr=16000, mono=True)
+ elif vc_audio_mode == "Upload audio":
+ selected_audio = input_upload_audio
+ if vocal_audio:
+ selected_audio = vocal_audio
+ elif input_upload_audio:
+ selected_audio = input_upload_audio
+ sampling_rate, audio = selected_audio
+ duration = audio.shape[0] / sampling_rate
+ audio = (audio / np.iinfo(audio.dtype).max).astype(np.float32)
+ if len(audio.shape) > 1:
+ audio = librosa.to_mono(audio.transpose(1, 0))
+ if sampling_rate != 16000:
+ audio = librosa.resample(audio, orig_sr=sampling_rate, target_sr=16000)
+ elif vc_audio_mode == "TTS Audio":
+ if tts_text is None or tts_voice is None:
+ return "You need to enter text and select a voice", None
+ asyncio.run(edge_tts.Communicate(tts_text, "-".join(tts_voice.split('-')[:-1])).save("tts.mp3"))
+ audio, sr = librosa.load("tts.mp3", sr=16000, mono=True)
+ input_audio_path = "tts.mp3"
+ f0_up_key = int(f0_up_key)
+ times = [0, 0, 0]
+ if hubert_model == None:
+ load_hubert()
+ if_f0 = cpt.get("f0", 1)
+ audio_opt = vc.pipeline(
+ hubert_model,
+ net_g,
+ sid,
+ audio,
+ input_audio_path,
+ times,
+ f0_up_key,
+ f0_method,
+ file_index,
+ # file_big_npy,
+ index_rate,
+ if_f0,
+ filter_radius,
+ tgt_sr,
+ resample_sr,
+ rms_mix_rate,
+ version,
+ protect,
+ f0_file=f0_file
+ )
 
-# Gradio interface
-iface = gr.Interface(
- fn=youtube_to_wav,
- inputs=["text", "text"],
- outputs="text",
- live=True,
- capture_session=True
-)
+def download_audio(url, audio_provider):
+ logs = []
+ os.makedirs("dl_audio", exist_ok=True)
+ if url == "":
+ logs.append("URL required!")
+ yield None, "\n".join(logs)
+ return None, "\n".join(logs)
+ if audio_provider == "Youtube":
+ logs.append("Downloading the audio...")
+ yield None, "\n".join(logs)
+ ydl_opts = {
+ 'noplaylist': True,
+ 'format': 'bestaudio/best',
+ 'postprocessors': [{
+ 'key': 'FFmpegExtractAudio',
+ 'preferredcodec': 'wav',
+ }],
+ "outtmpl": 'result/dl_audio/audio',
+ }
+ audio_path = "result/dl_audio/audio.wav"
+ with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+ ydl.download([url])
+ logs.append("Download Complete.")
+ yield audio_path, "\n".join(logs)
 
-# Launch Gradio interface
-iface.launch()
+def change_audio_mode(vc_audio_mode):
+ if vc_audio_mode == "Input path":
+ return (
+ # Input & Upload
+ gr.Textbox.update(visible=True),
+ gr.Checkbox.update(visible=False),
+ gr.Audio.update(visible=False),
+ # Youtube
+ gr.Dropdown.update(visible=False),
+ gr.Textbox.update(visible=False),
+ gr.Textbox.update(visible=False),
+ gr.Button.update(visible=False),
+ # Splitter
+ gr.Dropdown.update(visible=True),
+ gr.Textbox.update(visible=True),
+ gr.Button.update(visible=True),
+ gr.Button.update(visible=False),
+ gr.Audio.update(visible=False),
+ gr.Audio.update(visible=True),
+ gr.Audio.update(visible=True),
+ gr.Slider.update(visible=True),
+ gr.Slider.update(visible=True),
+ gr.Audio.update(visible=True),
+ gr.Button.update(visible=True),
+ # TTS
+ gr.Textbox.update(visible=False),
+ gr.Dropdown.update(visible=False)
+ )
+ # elif vc_audio_mode == "Upload audio":
+ # return (
+ # # Input & Upload
+ # gr.Textbox.update(visible=False),
+ ## gr.Checkbox.update(visible=True),
+ # gr.Audio.update(visible=True),
+ # Youtube
+ # gr.Dropdown.update(visible=False),
+ # gr.Textbox.update(visible=False),
+ # gr.Textbox.update(visible=False),
+ # gr.Button.update(visible=False),
+ # Splitter
+ # gr.Dropdown.update(visible=True),
+ # gr.Textbox.update(visible=True),
+ # gr.Button.update(visible=False),
+ # gr.Button.update(visible=True),
+ # gr.Audio.update(visible=False),
+ # gr.Audio.update(visible=True),
+ # gr.Audio.update(visible=True),
+ # gr.Slider.update(visible=True),
+ # gr.Slider.update(visible=True),
+ # gr.Audio.update(visible=True),
+ # gr.Button.update(visible=True),
+ # TTS
+ # gr.Textbox.update(visible=False),
+ $ gr.Dropdown.update(visible=False)
+ )
+ elif vc_audio_mode == "Youtube":
+ return (
+ # Input & Upload
+ # gr.Textbox.update(visible=False),
+ # gr.Checkbox.update(visible=False),
+ # gr.Audio.update(visible=False),
+ # Youtube
+ gr.Dropdown.update(visible=True),
+ gr.Textbox.update(visible=True),
+ gr.Textbox.update(visible=True),
+ gr.Button.update(visible=True),
+ # Splitter
+ # gr.Dropdown.update(visible=True),
+ # gr.Textbox.update(visible=True),
+ # gr.Button.update(visible=True),
+ # gr.Button.update(visible=False),
+ # gr.Audio.update(visible=True),
+ # gr.Audio.update(visible=True),
+ # gr.Audio.update(visible=True),
+ # gr.Slider.update(visible=True),
+ # gr.Slider.update(visible=True),
+ # gr.Audio.update(visible=True),
+ # gr.Button.update(visible=True),
+ # TTS
+ # gr.Textbox.update(visible=False),
+ # gr.Dropdown.update(visible=False)
+ )
+ # elif vc_audio_mode == "TTS Audio":
+ # return (
+ # # Input & Upload
+ # gr.Textbox.update(visible=False),
+ # gr.Checkbox.update(visible=False),
+ # gr.Audio.update(visible=False),
+ # Youtube
+ # gr.Dropdown.update(visible=False),
+ # gr.Textbox.update(visible=False),
+ # gr.Textbox.update(visible=False),
+ # gr.Button.update(visible=False),
+ # Splitter
+ # gr.Dropdown.update(visible=False),
+ # gr.Textbox.update(visible=False),
+ # gr.Button.update(visible=False),
+ # gr.Button.update(visible=False),
+ # gr.Audio.update(visible=False),
+ # gr.Audio.update(visible=False),
+ # gr.Audio.update(visible=False),
+ # gr.Slider.update(visible=False),
+ # gr.Slider.update(visible=False),
+ # gr.Audio.update(visible=False),
+ # gr.Button.update(visible=False),
+ # TTS
+ # gr.Textbox.update(visible=True),
+ # gr.Dropdown.update(visible=True)
+ )
+ 
+with gr.Blocks() as app:
+ gr.Markdown(
+ "# <center> YOUTUBE DOWNLOADER\n"
+ )
+ 
+ with gr.TabItem("Inference"):
 
+ with gr.Row():
+ with gr.Column():
+ vc_audio_mode = gr.Dropdown(label="Input voice", choices=["Youtube"], allow_custom_value=False, value="Youtube")
+ vc_download_audio = gr.Dropdown(label="Provider", choices=["Youtube"], allow_custom_value=False, visible=False, value="Youtube", info="Select provider (Default: Youtube)")
+ vc_link = gr.Textbox(label="Youtube URL", visible=True, info="Example: https://www.youtube.com/watch?v=Nc0sB1Bmf-A", placeholder="https://www.youtube.com/watch?v=...")
+ vc_log_yt = gr.Textbox(label="Output Information", visible=False, interactive=True)
+ vc_download_button = gr.Button("Download Audio", variant="primary", visible=True)
+ vc_audio_preview = gr.Audio(label="Downloaded Audio Preview", visible=True)
+ )
+ app.queue().launch()