from flask import Flask, request, send_file, Response
import requests
import os

app = Flask(__name__)

PLANTUML_URL = "http://127.0.0.1:8080"

@app.route("/")
def home():
    return open("index.html", "r", encoding="utf-8").read()

@app.route("/render")
def render():
    src = request.args.get("src", "")
    r = requests.get(PLANTUML_URL + "/render", params={"src": src}, stream=True)
    return Response(r.content, mimetype=r.headers.get("Content-Type"))

@app.route("/render/svg")
def render_svg():
    src = request.args.get("src", "")
    r = requests.get(PLANTUML_URL + "/render/svg", params={"src": src}, stream=True)
    return Response(r.content, mimetype=r.headers.get("Content-Type"))

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860)