Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import os | |
| import openai | |
| theme = gr.themes.Monochrome( | |
| primary_hue="purple", | |
| secondary_hue="indigo", | |
| text_size="text_lg", | |
| radius_size="radius_sm", | |
| font=[gr.themes.GoogleFont('Silkscreen'), 'ui-sans-serif', 'system-ui', 'sans-serif'], | |
| ).set( | |
| body_background_fill='*primary_100', | |
| body_background_fill_dark='*primary_950', | |
| body_text_color='*neutral_800', | |
| body_text_color_subdued='*primary_100', | |
| background_fill_primary='*secondary_900' | |
| ) | |
| openai.api_key = "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" | |
| with gr.Blocks() as iface: | |
| chatbot = [] | |
| msg = [] | |
| accusationButtons = [] | |
| culpritId = 1 | |
| 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() as chatRow: | |
| with gr.Column(scale=2): | |
| questionCounter = gr.Label(label="Remaining Questions", value="24") | |
| mapButton = gr.Button("Show Floor Plan") | |
| 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="", elem_classes="character-image-" + str(i)) | |
| i += 1 | |
| with gr.Row(visible=False, elem_classes="abs-window") as mapRow: | |
| with gr.Column(): | |
| closeMapButton = gr.Button("X", elem_classes="close-button") | |
| with gr.Row(visible=False, elem_classes="abs-window") as solvingRow: | |
| i = 0 | |
| while i < len(roles): | |
| with gr.Column(): | |
| accusationButtons.append(gr.Button(roles[i])) | |
| i += 1 | |
| closeSolvingButton = gr.Button("X", elem_classes="close-button") | |
| with gr.Row(visible=False) as resultRow: | |
| with gr.Column(): | |
| resultLabel = gr.Label(label="Result") | |
| 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): | |
| initSysMessage = initText[characterId] + "Stay in character. Use natural language. Don't reveal all of " \ | |
| "the information in a single message, and leave hints." | |
| message_history = [ | |
| {"role": "system", "content": initSysMessage} | |
| ] | |
| 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["label"]) | |
| if remainingQuestions > 0: | |
| remainingQuestions -= 1 | |
| return str(remainingQuestions) | |
| def checkIfQuestionsLeft(questionCounterText): | |
| remainingQuestions = int(questionCounterText["label"]) | |
| return remainingQuestions > 0 | |
| def tryToSolve(): | |
| return {solvingRow: gr.update(visible=True)} | |
| def showMap(): | |
| return {mapRow: gr.update(visible=True)} | |
| def hideSolveBox(): | |
| return {solvingRow: gr.update(visible=False)} | |
| def hideMap(): | |
| return {mapRow: gr.update(visible=False)} | |
| def showResult(): | |
| return {chatRow: gr.update(visible=False), solvingRow: gr.update(visible=False), resultRow: gr.update(visible=True)} | |
| def declareLoss(): | |
| return "You failed! The real culprit got away... " | |
| def declareVictory(): | |
| return "Congratulations, you found the killer! " | |
| botfunctions = [bot0, bot1, bot2, bot3, bot4] | |
| i = 0 | |
| while i < len(roles): | |
| msg[i].submit(user, [msg[i], chatbot[i]], [msg[i], chatbot[i]], queue=False).then( | |
| botfunctions[i], [chatbot[i], questionCounter], chatbot[i] | |
| ).then( | |
| updateQuestions, questionCounter, questionCounter | |
| ) | |
| i += 1 | |
| mapButton.click(showMap, None, mapRow) | |
| closeMapButton.click(hideMap, None, mapRow) | |
| finishButton.click(tryToSolve, None, solvingRow) | |
| closeSolvingButton.click(hideSolveBox, None, solvingRow) | |
| i = 0 | |
| while i < len(roles): | |
| if i == culpritId: | |
| accusationButtons[i].click(showResult, None, [chatRow, solvingRow, resultRow]).then(declareVictory, None, resultLabel) | |
| else: | |
| accusationButtons[i].click(showResult, None, [chatRow, solvingRow, resultRow]).then(declareLoss, None, resultLabel) | |
| i += 1 | |
| iface.launch() | |