|
|
import os |
|
|
import threading |
|
|
import urllib.request |
|
|
import time |
|
|
import sys |
|
|
import random |
|
|
import argparse |
|
|
import librosa |
|
|
from tqdm.auto import tqdm |
|
|
import torch |
|
|
import soundfile as sf |
|
|
import torch.nn as nn |
|
|
from datetime import datetime |
|
|
import numpy as np |
|
|
import shutil |
|
|
from gui import create_interface |
|
|
from pyngrok import ngrok |
|
|
from assets.i18n.i18n import I18nAuto |
|
|
|
|
|
import warnings |
|
|
warnings.filterwarnings("ignore") |
|
|
|
|
|
def generate_random_port(): |
|
|
"""Generates a random port between 1000 and 9000.""" |
|
|
return random.randint(1000, 9000) |
|
|
|
|
|
def start_gradio(port, share=False): |
|
|
"""Starts the Gradio interface with optional sharing.""" |
|
|
demo = create_interface() |
|
|
demo.launch( |
|
|
server_port=port, |
|
|
server_name='0.0.0.0', |
|
|
share=share, |
|
|
allowed_paths=[os.path.join(os.path.expanduser("~"), "Music-Source-Separation", "input"), "/tmp", "/content"], |
|
|
inline=False |
|
|
) |
|
|
|
|
|
def start_localtunnel(port, i18n): |
|
|
"""Starts the Gradio interface with localtunnel sharing.""" |
|
|
print(i18n("starting_localtunnel").format(port=port)) |
|
|
os.system('npm install -g localtunnel &>/dev/null') |
|
|
|
|
|
with open('url.txt', 'w') as file: |
|
|
file.write('') |
|
|
os.system(f'lt --port {port} >> url.txt 2>&1 &') |
|
|
time.sleep(2) |
|
|
|
|
|
endpoint_ip = urllib.request.urlopen('https://ipv4.icanhazip.com').read().decode('utf8').strip("\n") |
|
|
with open('url.txt', 'r') as file: |
|
|
tunnel_url = file.read().replace("your url is: ", "").strip() |
|
|
|
|
|
print(i18n("share_link").format(url=tunnel_url)) |
|
|
print(i18n("password_ip").format(ip=endpoint_ip)) |
|
|
|
|
|
start_gradio(port, share=False) |
|
|
|
|
|
def start_ngrok(port, ngrok_token, i18n): |
|
|
"""Starts the Gradio interface with ngrok sharing.""" |
|
|
print(i18n("starting_ngrok").format(port=port)) |
|
|
try: |
|
|
ngrok.set_auth_token(ngrok_token) |
|
|
ngrok.kill() |
|
|
tunnel = ngrok.connect(port) |
|
|
print(i18n("ngrok_url").format(url=tunnel.public_url)) |
|
|
|
|
|
start_gradio(port, share=False) |
|
|
except Exception as e: |
|
|
print(i18n("ngrok_error").format(error=str(e))) |
|
|
sys.exit(1) |
|
|
|
|
|
def main(method="gradio", port=None, ngrok_token=""): |
|
|
"""Main entry point for the application.""" |
|
|
|
|
|
i18n = I18nAuto() |
|
|
|
|
|
|
|
|
port = port or generate_random_port() |
|
|
print(i18n("selected_port").format(port=port)) |
|
|
|
|
|
|
|
|
if method == "gradio": |
|
|
print(i18n("starting_gradio_with_sharing")) |
|
|
start_gradio(port, share=True) |
|
|
elif method == "localtunnel": |
|
|
start_localtunnel(port, i18n) |
|
|
elif method == "ngrok": |
|
|
if not ngrok_token: |
|
|
print(i18n("ngrok_token_required")) |
|
|
sys.exit(1) |
|
|
start_ngrok(port, ngrok_token, i18n) |
|
|
else: |
|
|
print(i18n("invalid_method")) |
|
|
sys.exit(1) |
|
|
|
|
|
|
|
|
try: |
|
|
while True: |
|
|
time.sleep(5) |
|
|
except KeyboardInterrupt: |
|
|
print(i18n("process_stopped")) |
|
|
sys.exit(0) |
|
|
|
|
|
if __name__ == "__main__": |
|
|
parser = argparse.ArgumentParser(description="Music Source Separation Web UI") |
|
|
parser.add_argument("--method", type=str, default="gradio", choices=["gradio", "localtunnel", "ngrok"], help="Sharing method (default: gradio)") |
|
|
parser.add_argument("--port", type=int, default=None, help="Server port (default: random between 1000-9000)") |
|
|
parser.add_argument("--ngrok-token", type=str, default="", help="Ngrok authentication token (required for ngrok)") |
|
|
args = parser.parse_args() |
|
|
|
|
|
main(method=args.method, port=args.port, ngrok_token=args.ngrok_token) |
|
|
|
