app.py

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":
        store = request.form["store"]
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=generate_messages(store),
            temperature=0.6
        )
        return redirect(url_for("index", result=response['choices'][0]['message']['content']))

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


def generate_messages(store):
    return [{"role": "user", "content": "Suggest a cool name for the " + store}]