Update app.py

app.py

@@ -21,7 +21,9 @@ from openai import OpenAI
 assistant_id = os.environ.get('assistant_id')
 client = OpenAI(api_key = os.environ.get('api_key'))
 my_assistant = client.beta.assistants.retrieve(assistant_id)
-
+assistant_report_id = os.environ.get('assistant_report_id')
+client_report = OpenAI(api_key = os.environ.get('api_key'))
+my_assistant_report = client_report.beta.assistants.retrieve(assistant_report_id)
 
 app = Flask(__name__)
 cors = CORS(app)
@@ -205,7 +207,7 @@ def bot():
         role = "user",
         content = domanda
     )
-    run = client.beta.threads.runs.create(thread_id = thread_id, assistant_id = my_assistant.id)
+    run = client_report.beta.threads.runs.create(thread_id = thread_id, assistant_id = my_assistant.id)
     print(f"[] Run Created : {run.id}")
     
     while run.status != 'completed':
@@ -230,7 +232,46 @@ def bot():
     response = jsonify(output)
    
     return response
- 
+
+@app.route('/bot_report',methods=['POST','GET'])
+@cross_origin()
+def bot_report():
+    dati=[]
+    input_json = request.get_json()
+    domanda=input_json['domanda']
+    thread = client_report.beta.threads.create()
+    thread_id=thread.id
+    
+    messaggio = client_report.beta.threads.messages.create(
+        thread_id = thread_id,
+        role = "user",
+        content = domanda
+    )
+    run = client_report.beta.threads.runs.create(thread_id = thread_id, assistant_id = my_assistant_report.id)
+    print(f"[] Run Created : {run.id}")
+    
+    while run.status != 'completed':
+        run = client_report.beta.threads.runs.retrieve(
+          thread_id=thread_id,
+          run_id=run.id
+        )
+        print(run.status)
+        time.sleep(5)
+
+    # ottengo tutti i messaggi nel thread
+    message_responses = client_report.beta.threads.messages.list(thread_id = thread_id)
+    latest_messages = message_responses.data[0]
+    
+    text = latest_messages.content[0].text.value
+    text = re.sub(r'【.*?】', '', text)
+    d={}
+    d['thread']=thread_id
+    d['risposta']=text
+    dati.append(d)
+    output=dati
+    response = jsonify(output)
+   
+    return response
 
 if __name__ == "__main__":
     app.run(host='0.0.0.0',  port=7860)