Alexvatti commited on
Commit
b875a81
·
verified ·
1 Parent(s): 894229c

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -0
app.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from flask import Flask, request, jsonify
2
+ from openai import OpenAI
3
+ import os
4
+
5
+ app = Flask(__name__)
6
+ openai_client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
7
+
8
+ assistant_id = "asst_R7FO44rNY25nJeSszxHktjAR" # Replace with your actual Assistant ID
9
+
10
+ @app.route('/chat', methods=['POST'])
11
+ def chat():
12
+ user_message = request.json.get("message")
13
+
14
+ thread = openai_client.beta.threads.create()
15
+ openai_client.beta.threads.messages.create(thread.id, role="user", content=user_message)
16
+
17
+ run = openai_client.beta.threads.runs.create(thread_id=thread.id, assistant_id=assistant_id)
18
+
19
+ while run.status != "completed":
20
+ run = openai_client.beta.threads.runs.retrieve(thread_id=thread.id, run_id=run.id)
21
+
22
+ messages = openai_client.beta.threads.messages.list(thread_id=thread.id)
23
+ bot_reply = messages.data[0].content[0].text.value
24
+
25
+ return jsonify({"reply": bot_reply})