import os
import openai
import gradio as gr

course_username = os.environ["COURSE_USERNAME"]

george_password = os.environ["george_PWD"]
patill_password = os.environ["patill_PWD"]
erini_password = os.environ["erini_PWD"]
claire_password = os.environ["claire_PWD"]
shruteek_password = os.environ["shruteek_PWD"]
samuel_password = os.environ["samuel_PWD"]

george_api_key = os.environ["george_KEY"]
patill_api_key = os.environ["patill_KEY"]
erini_api_key = os.environ["erini_KEY"]
claire_api_key = os.environ["claire_KEY"]
shruteek_api_key = os.environ["shruteek_KEY"]
samuel_api_key = os.environ["samuel_KEY"]

password_to_api_key_dict = {
    george_password: george_api_key,
    patill_password: patill_api_key,
    erini_password: erini_api_key,
    claire_password: claire_api_key,
    shruteek_password: shruteek_api_key,
    samuel_password: samuel_api_key
    }


class Conversation:
    def __init__(self):
        self.assistant_id = ""
        self.thread_id = ""

    def create_thread(self):
        new_thread = client.beta.threads.create()
        self.thread_id = new_thread.id

    def respond_to_query(self, query, history):
        query_message = client.beta.threads.messages.create(
            thread_id=self.thread_id, role="user", content=query
        )
        streamed_run = client.beta.threads.runs.create(
            thread_id=self.thread_id,
            assistant_id=self.assistant_id,
            stream=True
        )
        current_response = "The Design Thinkerer: "
        for event in streamed_run:
            if event.event == "thread.message.delta":
                for new_text in event.data.delta.content:
                    current_response += new_text.text.value
                    yield current_response

current_conversation = Conversation()

with gr.Blocks(fill_height=True) as demo:
    username_textbox = gr.Textbox(label="Username")
    password_textbox = gr.Textbox(label="Password")
    login_btn = gr.Button("Login")

    assistant_id_textbox = gr.Textbox(label="Assistant ID", visible=False)
    submit_btn = gr.Button("Submit", visible=False)

    with gr.Column(visible=False) as chat_container:
        AI_chatbot = gr.ChatInterface(fn=current_conversation.respond_to_query, type="messages")


    def login_to_webapp(username, password):
        if (username == course_username) and (password in password_to_api_key_dict.keys()):
            global client
            student_api_key = password_to_api_key_dict[password]
            client = openai.OpenAI(api_key=student_api_key)
            return [gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=True), gr.update(visible=True)]
        else:
            gr.Info('The Username or Password was incorrect.')
            return [gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)]
    elements_changed_by_login = [username_textbox, password_textbox, login_btn, assistant_id_textbox, submit_btn]
    
    def set_assistant_and_show_chatbot(assistant_id):
        current_conversation.assistant_id = assistant_id
        current_conversation.create_thread()
        return [gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)]
    elements_changed_for_assistant = [assistant_id_textbox, submit_btn, chat_container]

    login_btn.click(
        inputs=[username_textbox, password_textbox],
        fn=login_to_webapp,
        outputs=elements_changed_by_login
        )

    submit_btn.click(  
            inputs=[assistant_id_textbox],
            fn=set_assistant_and_show_chatbot,
            outputs=elements_changed_for_assistant
        )

demo.launch()