File size: 834 Bytes
d2c314d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
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)
|