import os

import openai
from flask import Flask, redirect, render_template, request, url_for

app = Flask(__name__)
openai.api_key = os.getenv("OPENAI_API_KEY")


@app.route("/", methods=("GET", "POST"))
def index():
    if request.method == "POST":
        animal = request.form["animal"]
        response = openai.Completion.create(
            max_tokens=1024,
            n=1,
            stop=None,
            model="text-davinci-001",
            prompt=generate_prompt(animal),
            temperature=0.8,
        )
        return redirect(url_for("index", result=response.choices[0].text.replace('/r/n', '<br>')))

    result = request.args.get("result")
    return render_template("index.html", result=result)


def generate_prompt(animal):
    return animal
        # animal.capitalize()
    # )


if '__main__' == __name__:
    app.run(port=10800, host='0.0.0.0')