mirror of
https://github.com/voice-cloning-app/Voice-Cloning-App.git
synced 2026-02-24 20:20:22 +01:00
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
import os
|
|
import sys
|
|
import shutil
|
|
import webbrowser
|
|
from engineio.async_drivers import threading # noqa
|
|
from flask_socketio import SocketIO
|
|
from flask import Flask
|
|
|
|
from application.check_ffmpeg import check_ffmpeg
|
|
|
|
|
|
def load_paths():
|
|
paths = {
|
|
"datasets": os.path.join("data", "datasets"),
|
|
"models": os.path.join("data", "models"),
|
|
"hifigan": os.path.join("data", "hifigan"),
|
|
"results": os.path.join("data", "results"),
|
|
"languages": os.path.join("data", "languages"),
|
|
"training": os.path.join("data", "training"),
|
|
}
|
|
for path in paths.values():
|
|
os.makedirs(path, exist_ok=True)
|
|
return paths
|
|
|
|
|
|
def cleanup_mei():
|
|
"""
|
|
Rudimentary workaround for https://github.com/pyinstaller/pyinstaller/issues/2379
|
|
"""
|
|
mei_bundle = getattr(sys, "_MEIPASS", None)
|
|
|
|
if mei_bundle:
|
|
dir_mei, current_mei = mei_bundle.split("_MEI")
|
|
for file in os.listdir(dir_mei):
|
|
if file.startswith("_MEI") and not file.endswith(current_mei):
|
|
try:
|
|
shutil.rmtree(os.path.join(dir_mei, file))
|
|
except PermissionError: # mainly to allow simultaneous pyinstaller instances
|
|
pass
|
|
|
|
|
|
paths = load_paths()
|
|
static = os.path.join("application", "static")
|
|
app = Flask(__name__, template_folder=static, static_folder=static)
|
|
app.config["SEND_FILE_MAX_AGE_DEFAULT"] = 0
|
|
socketio = SocketIO(app, async_mode="threading", logger=True, engineio_logger=True, debug=True)
|
|
from application.views import * # noqa
|
|
|
|
if __name__ == "__main__":
|
|
cleanup_mei()
|
|
check_ffmpeg()
|
|
webbrowser.open_new_tab("http://localhost:5000")
|
|
socketio.run(app, host='0.0.0.0')
|