2021-03-10 15:56:40 +00:00
|
|
|
import os
|
|
|
|
|
import logging
|
|
|
|
|
import webbrowser
|
2021-03-18 12:27:38 +00:00
|
|
|
from engineio.async_drivers import threading
|
2021-03-10 15:56:40 +00:00
|
|
|
from flask_socketio import SocketIO
|
|
|
|
|
from flask import Flask
|
2021-03-18 16:54:52 +00:00
|
|
|
|
|
|
|
|
from application.check_ffmpeg import check_ffmpeg
|
2021-03-10 15:56:40 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
def load_paths():
|
|
|
|
|
paths = {
|
|
|
|
|
"datasets": os.path.join("data", "datasets"),
|
|
|
|
|
"models": os.path.join("data", "models"),
|
2021-03-22 21:05:33 +00:00
|
|
|
"pretrained": os.path.join("data", "pretrained"),
|
2021-03-10 15:56:40 +00:00
|
|
|
"waveglow": os.path.join("data", "waveglow"),
|
2021-03-18 12:27:38 +00:00
|
|
|
"results": os.path.join("data", "results"),
|
2021-03-10 15:56:40 +00:00
|
|
|
}
|
|
|
|
|
for path in paths.values():
|
|
|
|
|
os.makedirs(path, exist_ok=True)
|
|
|
|
|
return paths
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
paths = load_paths()
|
2021-03-18 12:27:38 +00:00
|
|
|
static = os.path.join("application", "static")
|
|
|
|
|
app = Flask(__name__, template_folder=static, static_folder=static)
|
2021-03-10 15:56:40 +00:00
|
|
|
socketio = SocketIO(app, async_mode="threading", logger=True, engineio_logger=True, debug=True)
|
|
|
|
|
from application.views import *
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
2021-03-18 16:53:04 +00:00
|
|
|
check_ffmpeg()
|
2021-03-10 15:56:40 +00:00
|
|
|
webbrowser.open_new_tab("http://localhost:5000")
|
|
|
|
|
socketio.run(app)
|