Spaces:
Sleeping
Sleeping
| 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", "Tour 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 thursday, 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 Lyanne Brimes, 27, a tour guide for a medieval history museum. You graduated from university " | |
| "a few years ago and have a degree in history. You work here part-time. You know director Eisenholz " | |
| "very well because you had a secret relationship. You'd prefer not to disclose any info about it " | |
| "unless pressed. You met every Thursday in the broom cupboard for your private meeting. You thought " | |
| "no one knew, but you are not sure anymore. Lately Eisenholz has been behaving very strangely. " | |
| "Something was troubling him. You tried to talk to him about it the" | |
| "night of the murder, but he never showed up. After waiting in the closet for 2 hours, you decided to " | |
| "go home. On your way out, you noticed the door to the room with all the weapons was open, " | |
| "but you thought Steve, the security guard, might be lurking around. Some might think Steve could be " | |
| "the murderer because he's been going crazy lately with the whole ghost thing and because he has " | |
| "access to every room, but you don't think he could have done it because he's kind-hearted. You are " | |
| "not the murderer, but if you had to guess who it was, you would say it was the curator. The curator " | |
| "and Eisenholz have been arguing recently about a medallion that has been in his family for " | |
| "centuries, but he doesn't want to give it to the museum because it's too valuable. You think you " | |
| "have a good nose for murder because you read detective novels. Now you're being questioned by a " | |
| "detective. You use formal language and are very well educated in history, especially medieval " | |
| "history. You have a bubbly personality. "] | |
| with gr.Row(): | |
| with gr.Column(scale=2): | |
| questionCounter = gr.Label(label="Remaining Questions", value="24") | |
| finishButton = gr.Button("Try to Solve") | |
| with gr.Column(scale=10): | |
| i = 0 | |
| while i < len(roles): | |
| with gr.Tab(roles[i]): | |
| with gr.Row(): | |
| with gr.Column(scale=9): | |
| chatbot.append(gr.Chatbot(label=roles[i])) | |
| msg.append(gr.Textbox(label="")) | |
| with gr.Column(scale=3): | |
| characterImage = gr.Label(label="") | |
| i += 1 | |
| with gr.Row(visible=False) as solvingBox: | |
| i = 0 | |
| while i < len(roles): | |
| with gr.Column(scale=1): | |
| finishButton = gr.Button(roles[i]) | |
| i += 1 | |
| def user(user_message, history): | |
| return "", history + [[user_message, None]] | |
| # this is an ugly workaround, you cannot pass regular integers to gradio functions, so i had to do it this way | |
| def bot0(history, questionCounterText): | |
| bot_message = generateAIMessage(history, questionCounterText, 0) | |
| history[-1][1] = bot_message | |
| return history | |
| def bot1(history, questionCounterText): | |
| bot_message = generateAIMessage(history, questionCounterText, 1) | |
| history[-1][1] = bot_message | |
| return history | |
| def bot2(history, questionCounterText): | |
| bot_message = generateAIMessage(history, questionCounterText, 2) | |
| history[-1][1] = bot_message | |
| return history | |
| def bot3(history, questionCounterText): | |
| bot_message = generateAIMessage(history, questionCounterText, 3) | |
| history[-1][1] = bot_message | |
| return history | |
| def bot4(history, questionCounterText): | |
| bot_message = generateAIMessage(history, questionCounterText, 4) | |
| history[-1][1] = bot_message | |
| return history | |
| def generateAIMessage(history, questionCounterText, characterId): | |
| if checkIfQuestionsLeft(questionCounterText): | |
| 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 | |
| else: | |
| return "--- YOU'VE RUN OUT OF QUESTIONS, MAKE YOUR CHOICE ---" | |
| def updateQuestions(questionCounterText): | |
| remainingQuestions = int(questionCounterText["value"]) | |
| if remainingQuestions > 0: | |
| remainingQuestions -= 1 | |
| return str(remainingQuestions) | |
| def checkIfQuestionsLeft(questionCounterText): | |
| remainingQuestions = int(questionCounterText["value"]) | |
| return remainingQuestions > 0 | |
| def tryToSolve(): | |
| print("ayy you clicked") | |
| return {solvingBox: gr.update(visible=True)} | |
| # again, ugly workaround | |
| msg[0].submit(user, [msg[0], chatbot[0]], [msg[0], chatbot[0]], queue=False).then( | |
| bot0, [chatbot[0], questionCounter], chatbot[0] | |
| ).then( | |
| updateQuestions, questionCounter, questionCounter | |
| ) | |
| msg[1].submit(user, [msg[1], chatbot[1]], [msg[1], chatbot[1]], queue=False).then( | |
| bot1, [chatbot[1], questionCounter], chatbot[1] | |
| ).then( | |
| updateQuestions, questionCounter, questionCounter | |
| ) | |
| msg[2].submit(user, [msg[2], chatbot[2]], [msg[2], chatbot[2]], queue=False).then( | |
| bot2, [chatbot[2], questionCounter], chatbot[2] | |
| ).then( | |
| updateQuestions, questionCounter, questionCounter | |
| ) | |
| msg[3].submit(user, [msg[3], chatbot[3]], [msg[3], chatbot[3]], queue=False).then( | |
| bot3, [chatbot[3], questionCounter], chatbot[3] | |
| ).then( | |
| updateQuestions, questionCounter, questionCounter | |
| ) | |
| msg[4].submit(user, [msg[4], chatbot[4]], [msg[4], chatbot[4]], queue=False).then( | |
| bot4, [chatbot[4], questionCounter], chatbot[4] | |
| ).then( | |
| updateQuestions, questionCounter, questionCounter | |
| ) | |
| finishButton.click(tryToSolve, None, solvingBox) | |
| iface.launch() | |