mirror of
https://github.com/SkalaraAI/management-llm.git
synced 2025-04-09 15:00:19 -04:00
Created app.py
added supporting templates too.
This commit is contained in:
parent
59ab174acb
commit
75fcf820b5
28
flask_testing/app.py
Normal file
28
flask_testing/app.py
Normal 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
|
15
flask_testing/templates/index.html
Normal file
15
flask_testing/templates/index.html
Normal 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>
|
Loading…
Reference in New Issue
Block a user