import gradio as gr
import json, openai, os, time
from openai import OpenAI

client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))

def show_json(obj):
    print("###")
    print(json.loads(obj.model_dump_json()))
    print("###")

def wait_on_run(run, thread):
    while run.status == "queued" or run.status == "in_progress":
        run = client.beta.threads.runs.retrieve(
            thread_id=thread.id,
            run_id=run.id,
        )
        time.sleep(0.5)
    return run

def chat(message, history):    
    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)

    run = wait_on_run(run, thread)
    
    show_json(run)

    messages = client.beta.threads.messages.list(thread_id=thread.id)
    
    show_json(messages)
    ###
    
    #for i in range(len(message)):
    #    time.sleep(0.5)
    #    yield "You typed: " + message[: i+1]
    return "Done"

gr.ChatInterface(chat).launch()