Jesus Carrasco commited on
Commit
eda05b2
·
1 Parent(s): 005c96d

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +46 -0
app.py ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import openai
2
+ import os
3
+ from flask import Flask, request, jsonify
4
+ from flask_cors import CORS
5
+
6
+ # Set your OpenAI API key
7
+ openai.api_key = os.environ.get("OPENAI_API_KEY")
8
+
9
+ app = Flask(__name__)
10
+ CORS(app)
11
+
12
+ @app.route("/api/chat", methods=["POST"])
13
+ def chat():
14
+ user_messages = request.json["messages"]
15
+
16
+ # Add a system message to set the context for the assistant
17
+ system_message = {
18
+ 'role': 'system',
19
+ 'content': 'You are ChatGPT, a large language model trained by OpenAI. Answer as concisely as possible. Knowledge cutoff: 2021-09 Current date: 2023-04-15'
20
+ }
21
+ user_messages.insert(0, system_message)
22
+
23
+ # Create a prompt for GPT-3.5 Turbo
24
+ prompt = ""
25
+ for message in user_messages:
26
+ prompt += f"{message['role']}: {message['content']}\n"
27
+ prompt += "Assistant:"
28
+
29
+ # Make a request to the OpenAI API
30
+ response = openai.Completion.create(
31
+ engine="gpt-3.5-turbo",
32
+ prompt=prompt,
33
+ max_tokens=150,
34
+ n=1,
35
+ stop=None,
36
+ temperature=0.5,
37
+ )
38
+
39
+ # Extract the generated response from the API response
40
+ assistant_message = response.choices[0].text.strip()
41
+ total_tokens = response['usage']['total_tokens']
42
+
43
+ return jsonify({"message": assistant_message, "total_tokens": total_tokens})
44
+
45
+ if __name__ == "__main__":
46
+ app.run(debug=True)