import gradio as gr
import random

from llm.openai import Llm

# # mock for testing
# from llm.mock import Llm

llm = Llm()

def assistant_response(prompt):
    answer = llm.chatcompletion(prompt)
    return answer

def respond(message, chat_history):
    answer = llm.chatcompletion(message)
    print(answer)
    chat_history.append((message, answer))
    return "", chat_history

title = "OpenAI Assistant API: " + llm.assistant.name

if llm.assistant.description is None:
    model = llm.assistant.model
    description = f"このデモはOpenAI Assistant APIのデモです。テキストボックスにテキストを入力すると、{model}モデルが応答します。"
else:
    description = llm.assistant.description

with gr.Blocks() as demo:
    gr.Markdown(
    f"""
    # {title}
    {description}
    """)
    
    chatbot = gr.Chatbot()
    msg = gr.Textbox()
    clear = gr.ClearButton([msg, chatbot])

    msg.submit(respond, [msg, chatbot], [msg, chatbot])

if __name__ == "__main__":
    demo.launch()