Created app.py

added supporting templates too.
This commit is contained in:
Atin Pothiraj 2023-05-30 22:34:06 -04:00
parent 59ab174acb
commit 75fcf820b5
2 changed files with 43 additions and 0 deletions

28
flask_testing/app.py Normal file
View File

@ -0,0 +1,28 @@
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":
task_form = request.form["task_form"]
response = openai.Completion.create(model="text-davinci-003",
prompt = generate_prompt(task_form),
temperature=0,
max_tokens=100)
return redirect(url_for("index", result=response.choices[0].text))
result = request.args.get("result")
return render_template("index.html", result=result)
def generate_prompt(task_form):
base_text = "You are a software manager at the company Alphabet. Q: Format a message that specifically delegates the parts of the following task to your team of engineers."
return base_text + "TASK: " + task_form

View File

@ -0,0 +1,15 @@
<!DOCTYPE html>
<head>
<title>Quickstart</title>
</head>
<body>
<h3>Name your task</h3>
<form action="/" method="post">
<input type="text" name="task_form" placeholder="Enter a task" required />
<input type="submit" value="Generate subtasks" />
</form>
{% if result %}
<div class="result">{{ result }}</div>
{% endif %}
</body>