File size: 858 Bytes
4a78a80 cce6aec 4a78a80 df0e7bb 4a78a80 df0e7bb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
from openai import OpenAI
import gradio as gr
import json
import os
client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
assistant = os.environ.get("ASSISTANT")
thread = client.beta.threads.create()
def predict(input,history):
message = client.beta.threads.messages.create(
thread_id=thread.id,
role="user",
content=input
)
run = client.beta.threads.runs.create(
thread_id=thread.id,
assistant_id=assistant
)
while True:
run = client.beta.threads.runs.retrieve( thread_id=thread.id,run_id=run.id)
if run.status=='completed':
messages = client.beta.threads.messages.list(thread_id=thread.id)
current_message= messages.data[0]
return(current_message.content[0].text.value)
chatbot = gr.Chatbot(show_copy_button=True)
gr.ChatInterface(fn=predict, chatbot=chatbot).launch(share=True) |