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()