mirror of
https://github.com/SkalaraAI/management-llm.git
synced 2025-04-17 19:00:16 -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