|
import os, sys
|
|
import signal
|
|
from flask import Flask, request, redirect
|
|
|
|
now_dir = os.getcwd()
|
|
sys.path.append(now_dir)
|
|
|
|
from core import run_download_script
|
|
|
|
app = Flask(__name__)
|
|
|
|
|
|
@app.route("/download/<path:url>", methods=["GET"])
|
|
def download(url):
|
|
file_path = run_download_script(url)
|
|
if file_path == "Model downloaded successfully.":
|
|
if "text/html" in request.headers.get("Accept", ""):
|
|
return redirect("https://applio.org/models/downloaded", code=302)
|
|
else:
|
|
return ""
|
|
else:
|
|
return "Error: Unable to download file", 500
|
|
|
|
|
|
@app.route("/shutdown", methods=["POST"])
|
|
def shutdown():
|
|
print("This Flask server is shutting down... Please close the window!")
|
|
os.kill(os.getpid(), signal.SIGTERM)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(host="localhost", port=8000)
|
|
|
