diff --git a/.gitignore b/.gitignore index 894a44cc0..183da3b50 100644 --- a/.gitignore +++ b/.gitignore @@ -102,3 +102,6 @@ venv.bak/ # mypy .mypy_cache/ + +#openaikey +openaiapikey.txt \ No newline at end of file diff --git a/app.py b/app.py index d82c51f0d..c0ec5a465 100644 --- a/app.py +++ b/app.py @@ -1,6 +1,23 @@ -from flask import Flask +from flask import Flask, request +from chatai import AI_Response + app = Flask(__name__) +@app.route("/api", methods=["GET", "POST"]) +def hello(): + if request.method == "GET": + data = request.args.get("prompt") + elif request.method == "POST": + jsondata = request.get_json() + data = jsondata["prompt"] + + response = AI_Response(data) + return response + @app.route('/') def hello_world(): return 'Hello, World!' + + +if __name__ == "__main__": + app.run() \ No newline at end of file diff --git a/chatai.py b/chatai.py new file mode 100644 index 000000000..b2593cc52 --- /dev/null +++ b/chatai.py @@ -0,0 +1,12 @@ +import os +import openai + +def AI_Response(text:str): + openai.api_key = os.environ.get("OPENAI_API_KEY") + response = openai.Completion.create( + model="text-davinci-003", + prompt=text, + max_tokens=100, + temperature=0 + ) + return response diff --git a/requirements.txt b/requirements.txt index 147ddd086..a9302e31c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,4 @@ Flask Gunicorn +numpy +openai