bstraehle's picture
Update app.py
06d9591 verified
raw
history blame
1.08 kB
import gradio as gr
import json, os, time
from openai import OpenAI
def show_json(obj):
print("###")
print(json.loads(obj.model_dump_json()))
print("###")
def chat(message, history):
client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
assistant = client.beta.assistants.create(
name="Math Tutor",
instructions="You are a personal math tutor. Answer questions briefly, in a sentence or less.",
model="gpt-4-1106-preview",
)
show_json(assistant)
thread = client.beta.threads.create()
show_json(thread)
message = client.beta.threads.messages.create(
thread_id=thread.id,
role="user",
content="I need to solve the equation `3x + 11 = 14`. Can you help me?",
)
show_json(message)
run = client.beta.threads.runs.create(
thread_id=thread.id,
assistant_id=assistant.id,
)
show_json(run)
###
for i in range(len(message)):
time.sleep(0.5)
yield "You typed: " + message[: i+1]
gr.ChatInterface(chat).launch()