import gradio as gr
import os
import openai

openai.api_key = "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"

with gr.Blocks() as iface:
    chatbot = []
    msg = []
    roles = ["Security Guard", "Curator", "Researcher", "Conservationist", "Guide"]
    initText = ["You are Steve Binner, 31, a security guard for a museum of medieval "
                "history. You're absolutely sure that this place is haunted. A month ago, "
                "before the spooky stuff started, you were really happy with your job, "
                "but that changed real fast. The ghost is a french nobleman named Baron "
                "Hugo Antoine the Third. You've been trying to tell other people about "
                "this, but  you've only been laughed at so far. But you've got evidence! "
                "Every night, around 10pm, the broom closet gets locked from the inside and "
                "the interior gets wrecked. We're talking pushed over tables, "
                "broken containers, the whole shebang. When it comes to the murder of "
                "director Eisenholz last thursday, that has to have been the ghost, "
                "no question. You even have the door card records for the room he was in: "
                "After the museum closed at 5pm, nobody entered that room till the next "
                "morning. Now you're being interrogated by a detective. You don't use "
                "uptight language, and you're not super well educated on most stuff, "
                "but when it comes to the paranormal, you're an ace. ",
                "You are a Curator. ",
                "You are a Researcher. ",
                "You are a Conservationist. ",
                "You are a Guide. "]

    i = 0
    while i < len(roles):
        with gr.Tab(roles[i]):
            chatbot.append(gr.Chatbot())
            msg.append(gr.Textbox())
        i += 1


    def user(user_message, history):
        return "", history + [[user_message, None]]


    def bot(history, characterId):
        bot_message = generateAIMessage(history, characterId)
        history[-1][1] = bot_message
        return history


    def generateAIMessage(history, characterId):
        message_history = [
            {"role": "system", "content": initText[characterId] + "Stay in character. Use natural language. Don't "
                                                                  "reveal all of the information in a single message,"
                                                                  " and leave hints. "}
        ]
        for pair in history:
            message_history.append({"role": "user", "content": pair[0]})
            print()
            if pair[1] is not None:
                message_history.append({"role": "assistant", "content": pair[1]})
        completion = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=message_history
        )
        return completion.choices[0].message.content

    i = 0
    while i < len(msg):
        msg[i].submit(user, [msg[i], chatbot[i]], [msg[i], chatbot[i]], queue=False).then(
            bot, [chatbot[i], i], chatbot[i]
        )
        i += 1

iface.launch()