Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import os | |
| import openai | |
| openai.api_key = os.environ['KEY'] | |
| theme = gr.themes.Monochrome( | |
| text_size="lg" | |
| ) | |
| css = "@import url(https://fonts.googleapis.com/css2?family=Silkscreen:wght@400;700&display=swap);#component-3," \ | |
| "#component-4{margin-bottom:10px}.gradio-container{" \ | |
| "background-color:#cda678;font-family:Silkscreen;position:relative}#component-1{min-height:75vh}#component-4{" \ | |
| "margin-top:auto}.tabs.svelte-btpldm{height:100%}.tabitem{background-color:#002366;border:1px solid " \ | |
| "#fff}#component-17,#component-25,#component-33,#component-41,#component-9{" \ | |
| "min-height:70vh;max-height:70vh}.secondary.svelte-58yet2{" \ | |
| "background-color:#002366;font-weight:400}#component-10,#component-11,#component-18,#component-19," \ | |
| "#component-26,#component-27,#component-34,#component-35,#component-42,#component-43{" \ | |
| "background-color:#aaa}.message.svelte-a99nd8.svelte-a99nd8{" \ | |
| "padding:15px;position:relative}.user.svelte-a99nd8.svelte-a99nd8{" \ | |
| "margin-right:10px}.bot.svelte-a99nd8.svelte-a99nd8{" \ | |
| "padding-left:15px;margin-left:10px}.bot.svelte-a99nd8.svelte-a99nd8:before," \ | |
| ".user.svelte-a99nd8.svelte-a99nd8:after{" \ | |
| "content:\"\";width:10px;height:10px;position:absolute;bottom:-1px;border:1px solid var(" \ | |
| "--color-border-accent);background-color:var(--color-accent-soft)}.bot.svelte-a99nd8.svelte-a99nd8:before{" \ | |
| "left:-10px;border-right:0}.user.svelte-a99nd8.svelte-a99nd8:after{right:-10px;border-left:0}textarea{" \ | |
| "background-color:#fff;min-height:50px!important}#component-48,#component-51,#component-61{" \ | |
| "position:absolute;left:50%;top:50%;transform:translateX(-50%) translateY(" \ | |
| "-50%);width:50%;height:50vh;padding:80px 20px 20px;background-color:#fff;border:1px solid " \ | |
| "#002366}#component-51{display:flex;flex-direction:column}#component-51.hide.svelte-1btyfsc{" \ | |
| "display:none}#component-51:before{content:\"The Killer is: " \ | |
| "\";position:absolute;top:20px;left:20px;color:#000;font-size:var(--text-lg)}#component-51>:not(:last-child){" \ | |
| "flex-grow:1}#component-49{position:initial;background-image:url(" \ | |
| "\"file/img/map.jpg\");background-position:center;background-repeat:no-repeat;background-size:contain" \ | |
| "}#component-62{position:initial}#component-50,#component-57,#component-64{" \ | |
| "position:absolute;top:20px;right:20px;width:44px;height:44px;min-width:44px;min" \ | |
| "-height:44px;line-height:0}.output-class.svelte-1s28oeb.svelte-1s28oeb.svelte-1s28oeb,div.svelte-1eq475l{" \ | |
| "font-weight:400}br+br+br{display:none}" | |
| with gr.Blocks(theme=theme, css=css) as iface: | |
| chatbot = [] | |
| msg = [] | |
| accusationButtons = [] | |
| culpritId = 3 | |
| names = ["Steve Binner", "Albert Wright", "Dominic Schwartz", "Brianna Small", "Lyanne Brimes"] | |
| 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 had 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. You’re not going to rule out " | |
| "that possession could’ve been used. You found the body at around 2am in the warehouse during your " | |
| "second patrol for the night. Because of technical maintenance, all security cams were turned off " | |
| "during this week, because they had to be readied for the big exhibition that’ll happened soon. The " | |
| "murder weapon was a sword., stabbed through the director’s heart. It was very fancy, with engravings " | |
| "of that odd symbol that the boy scouts used when you were a kid. You’re not sure where that’s from, " | |
| "but you suspect it could’ve belonged to the baron. If asked about Brianna Small’s keycard, " | |
| "talk about how the last few days were stressful, because she accidentally destroyed her card. You " | |
| "had to open every door for her, and she made a big fuss about it, so everybody knows that she needs " | |
| "help. She’s a pain to work with. Now you're being interrogated by a detective. Director Eisenholz " | |
| "has been murdered. 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 Albert Wright, 61, the curator of this museum. You have been the curator here since it " | |
| "opened 32 years ago, because your family has close ties with the Eisenholz family. You knew Richard " | |
| "Eisenholz since he was a boy and would never hurt him. Even though you weren’t always on the same " | |
| "page, your values and traditions were the same, which was very important to you and your families. " | |
| "And imagine the scandal it would be, if a member of an upper-class family would commit such an " | |
| "atrocity. On the day of the murder, you left at 4pm to go check out if everything was alright in the " | |
| "other museum that you own. If asked, explain how you couldn’t have gotten in the storage room (where " | |
| "the body was found) because your card was missing for a few days. You and Mr. Eisenholz were in a " | |
| "tight spot lately, because you needed something new to display. There weren’t a lot of new artifacts " | |
| "lately and the Box Miss Small, the conservationist, was examining, wasn’t that promising according " | |
| "to Eisenholz. You were making requests to Mr. Eisenholz if he could lend you something novel from " | |
| "his collection of family heirlooms, but he wasn’t as cooperative as you’d wished for. Luckily the " | |
| "discovery Miss Small made saved the museum. You lost your keycard for the museum doors a couple of " | |
| "days ago, which is somewhat annoying but manageable since your visits are somewhat sparce anyway. " | |
| "Miss Small is a very good employee, mostly since that discovery she made. It was her idea to open " | |
| "that box, even though Eisenholz said there would be nothing in it. But with her skills, she opened " | |
| "it and - as she’d expected - a priceless object was inside. The museum immediately put it on " | |
| "display, since there weren’t any new things lately and we had less visitors, but now it’s time for a " | |
| "huge promotional campaign. Mister Schwartz, the resident researcher, is an enthusiastic and " | |
| "passionate employee. He was at a conference representing the museum recently. But his passion can " | |
| "push it too far sometimes. You got involved in a discussion with him and Mr. Eisenholz. I think it " | |
| "was something about an artifact the museum supposedly culturally appropriated or some other " | |
| "nonsense, but you didn’t really care. You’re too old for this kind of topic where young people think " | |
| "they have to posture how good and right they are. Back in your day, things were more normal. Miss " | |
| "Brimes, the visitor’s chaperone, is quite a pleasant lady. She’s a treat to be around and you have " | |
| "had a plethora of great interactions with her, since she frequently visited Richard’s offices for " | |
| "one reason or another. You don’t really know much about Mister Binner, the nightwatchman. Word is " | |
| "going around that he’s a bit of a nutcase, but as long as he does his job well, you do not mind. Now " | |
| "you're being interrogated by a detective. Director Eisenholz has been murdered. Use sophisticated " | |
| "language. You are an old-fashioned gentleman, but you are not afraid to voice your complaints. ", | |
| "You are Dominic Schwartz, 35, and the researcher in this museum. Start the conversation by " | |
| "explaining that you love the history of medieval times, mostly regarding French monarchs. You are so " | |
| "lucky to be able to work in this museum and do the research of the artifacts! Inform the detective " | |
| "about various neat facts about French medieval history. Importantly, mention that the museum uses a " | |
| "lot of symbols of a dragon to make everything look more authentic, despite the fact that back in the " | |
| "day they weren’t depicted and for sure never in France, since in France, the depiction of a beast " | |
| "was the Loup-Garou; a half bear half wolf creature. The dragon was feared during Gaul period, " | |
| "largely forgotten during medieval times, and only made a comeback after the revolution. the " | |
| "Napoleonic Era. You may have exposed a bit too much here, but it’s probably fine, since only an " | |
| "expert in this field (such as yourself, of course) would know this. This is very important to you. " | |
| "You think it’s an injustice that the museum continues to falsely educate the public on certain " | |
| "topics. For instance, the gigantic banner depicting a siege in exhibition hall #3? Obviously " | |
| "Hispanic, just look at the uniforms. You’ve had multiple disagreements with a lot of the higher-ups, " | |
| "including curator Wright and director Eisenholz. At the night of the murder, you weren’t at the " | |
| "museum. You left earlier because you had to go to a conference. The curator, Mister Wright, " | |
| "can vouch for that because he was responsible for covering the expenses (it was a business trip, " | |
| "after all). You mention that there is even a YouTube video of you holding a talk at that conference " | |
| "about the retrieved diary of Baron Hugo Antoin the Third, a lesser known French noble. If asked " | |
| "about Brianna, the conservationist, say that she only cares about things she was involved with, " | |
| "anything other she doesn’t care if it breaks. If asked about Lyanne Brimes, the guide, say that you " | |
| "think she is a nice person, but it is very obvious that Mr Eisenholz and she got something going on. " | |
| "It could be enough of a motive for the murder, but you’re hesitant to make accusations. If asked " | |
| "about Albert Wright, the curator, say that you think he is a self-absorbed man who believes in " | |
| "classism and capitalism like Mr Eisenholz, but at least Mr Eisenholz was more sensible. As an " | |
| "anti-capitalist yourself, you can’t stand his greedy behavior. If asked about Steve, the security " | |
| "guard, talk about how he’s a bit of a weirdo, but harmless. His ghost stories would be very funny if " | |
| "he wasn’t serious. If asked about a “boy scout symbol”, you know that it’d be a fleur-de-lis, " | |
| "with all its historical context. Now you're being interrogated by a detective. Director Eisenholz " | |
| "has been murdered. Your personality is enthusiastic, but slightly arrogant and annoying, " | |
| "although you’d never admit that. Your thoughts tend to bounce all over the place.", | |
| "You are Brianna Small, 43, a conservationist at a museum of medieval history, but you would call " | |
| "yourself almost a scientist. You worked here for decades at the same position, but you are hoping " | |
| "for a promotion soon. You knew the victim mainly because he was owner of almost 50% of the artifacts " | |
| "in this museum and, as a conservationist, you held a few meetings with him discussing the " | |
| "preservation. You had a few inputs and ideas on the display and handling of the artifacts and " | |
| "Eisenholz usually agreed, which made working with him pleasant. At the night of the murder, " | |
| "you left the building at 18:00 because you made a new discovery a few days ago and wanted to " | |
| "document it. Your office is in the west wing and has a small window, where you saw Steve, " | |
| "the security guard, pass by a few times. You think Steve could have been the murderer because he has " | |
| "access to every room and has been acting crazy lately. He thinks ghosts are real, but from a " | |
| "scientific viewpoint that could never be true. You couldn’t have entered the warehouse and killed " | |
| "the director because your security keycard fell in acid two days ago. Steve. the guard, can confirm " | |
| "that, because he had to open all the rooms for you. A few weeks ago, Eisenholz donated a new " | |
| "artifact to the museum. It was a box made of wood with magnificent engravings, but it was sealed. " | |
| "You asked Eisenholz if you could try to open it and, with your skills of a conservationist, " | |
| "you’d be able to do so without damaging the box. Eisenholz was hesitant at first, mostly due to that " | |
| "pesky curator, Albert Wright, butting in. But after a few discussions he let me do it. And of " | |
| "course, it worked. What you found was beyond imagination. There was a golden ring belonging to a " | |
| "famous French emperor inside in perfect condition. Importantly, it had the form of a dragon with " | |
| "ruby crystals as eyes. Since this is a valuable discovery and the museum hasn’t got anything new " | |
| "lately, you wanted to talk to the director as soon as possible to immediately display it in a secure " | |
| "glass container. You wonder if Dominic, a researcher at the museum, can find out anything about the " | |
| "ring. He has been occupied lately on an artifact he thinks doesn’t belong here. If he keeps up his " | |
| "extremist attitude, he will cost you all the museum and your jobs. You aren’t well acquainted with " | |
| "Lyanne Brimes, the tour guide, apart from occasionally meeting her in the halls and making small " | |
| "talk. She seems nice, but somewhat ambitionless. If asked about it, you have no idea what the symbol " | |
| "on the weapon means. Now you're being interrogated by a detective. Director Eisenholz has been " | |
| "murdered. You’re professional, driven & to the point. ", | |
| "You are Lyanne Brimes, 27, a tour guide for a museum of medieval history. You graduated college a " | |
| "few years ago and have a degree in history. You work here part time. You know director Richard " | |
| "Eisenholz very well because you had a secret relationship with him. You don’t like talking about " | |
| "this. You met every Thursday in the broom closet. You thought nobody knew, " | |
| "but you are not sure anymore. Lately Eisenholz was behaving very strangely. Something was bothering " | |
| "him. You wanted to talk with him about it on the night of the murder at 5 pm, but he never came. You " | |
| "had to wait 3 hours to get out of the closet, because you heard Steve Binner walking up and down the " | |
| "east wing and talking to a portrait at the end of the corridor. On your way out, you noticed the " | |
| "door to the storage room was open but, thought it might have been Steve (the security guard) lurking " | |
| "around. 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 were fighting lately because a medallion, that has been in his " | |
| "family possession for centuries, but he didn’t want to give it to the museum because it was too " | |
| "valuable.You don’t know Brianna Small (the conservationist) very well, but if you had to say " | |
| "something about her, it would be that she is very smart and ambitious but too focused on her merits. " | |
| "You’ve met the curator, Albert Wright, before, due to his proximity to the director. He’s a stuffy " | |
| "old type, but you never had many problems with him. He doesn’t have a clue you and Richard were " | |
| "together. The same couldn’t be said about Dominic Schwartz, the researcher. He’s probably aware that " | |
| "something saucy is going on, but he never said anything about it. He’s alright, but sometimes you " | |
| "wish he’d stop talking so much. He’s had some disagreements with his superiors before. Now you're " | |
| "being interrogated by a detective. Director Eisenholz has been murdered. You use formal language, " | |
| "and you are very well educated in history stuff, mostly medieval. You have a bubbly personality."] | |
| with gr.Row() as chatRow: | |
| with gr.Column(scale=1): | |
| questionCounter = gr.Label(label="Remaining Questions", value="20") | |
| mapButton = gr.Button("Show Floor Plan") | |
| finishButton = gr.Button("Pick a Culprit") | |
| with gr.Column(scale=7): | |
| i = 0 | |
| while i < len(roles): | |
| with gr.Tab("The " + roles[i]): | |
| with gr.Row(): | |
| with gr.Column(scale=9): | |
| chatbot.append(gr.Chatbot(label=names[i] + ", " + roles[i])) | |
| msg.append(gr.Textbox(label="", placeholder="Ask a question...")) | |
| with gr.Column(scale=3): | |
| characterImage = gr.Label(label="") | |
| i += 1 | |
| with gr.Row(visible=False) as mapRow: | |
| with gr.Column(): | |
| closeMapButton = gr.Button("X") | |
| with gr.Row(visible=False) as solvingRow: | |
| i = 0 | |
| while i < len(roles): | |
| accusationButtons.append(gr.Button("The " + roles[i])) | |
| i += 1 | |
| closeSolvingButton = gr.Button("X") | |
| with gr.Row(visible=False) as resultRow: | |
| with gr.Column(): | |
| resultLabel = gr.Label(label="Result") | |
| with gr.Row() as introRow: | |
| with gr.Column(): | |
| introLabel = gr.Label(label="You are a detective investigating a murder. The victim was the director of a " | |
| "medieval museum, at which his body was found. There are five possible " | |
| "suspects for you to interrogate, all of which are members of the faculty. " | |
| "You can only ask 20 Questions. Good luck. ") | |
| closeIntroButton = gr.Button("X") | |
| 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, question_counter_text): | |
| return bot(0, history, question_counter_text) | |
| def bot1(history, question_counter_text): | |
| return bot(1, history, question_counter_text) | |
| def bot2(history, question_counter_text): | |
| return bot(2, history, question_counter_text) | |
| def bot3(history, question_counter_text): | |
| return bot(3, history, question_counter_text) | |
| def bot4(history, question_counter_text): | |
| return bot(4, history, question_counter_text) | |
| def bot(number, history, question_counter_text): | |
| bot_message = generate_ai_message(history, question_counter_text, number) | |
| history[-1][1] = bot_message | |
| return history | |
| def generate_ai_message(history, question_counter_text, character_id): | |
| if check_if_questions_left(question_counter_text): | |
| init_sys_message = initText[character_id] + "Stay in character. Use natural language. Don't reveal all of" \ | |
| "the information in a single message, and leave hints. Never " \ | |
| "reveal everything in a single message." | |
| message_history = [ | |
| {"role": "system", "content": init_sys_message} | |
| ] | |
| 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 update_questions(question_counter_text): | |
| remaining_questions = int(question_counter_text["label"]) | |
| if remaining_questions > 0: | |
| remaining_questions -= 1 | |
| return str(remaining_questions) | |
| def check_if_questions_left(question_counter_text): | |
| remaining_questions = int(question_counter_text["label"]) | |
| return remaining_questions > 0 | |
| def disable_input(): | |
| return gr.update(interactive=False, placeholder="Waiting for an answer..."), \ | |
| gr.update(interactive=False, placeholder="Waiting for an answer..."), \ | |
| gr.update(interactive=False, placeholder="Waiting for an answer..."), \ | |
| gr.update(interactive=False, placeholder="Waiting for an answer..."), \ | |
| gr.update(interactive=False, placeholder="Waiting for an answer...") | |
| def enable_input(): | |
| return gr.update(interactive=True, placeholder="Ask a question..."), \ | |
| gr.update(interactive=True, placeholder="Ask a question..."), \ | |
| gr.update(interactive=True, placeholder="Ask a question..."), \ | |
| gr.update(interactive=True, placeholder="Ask a question..."), \ | |
| gr.update(interactive=True, placeholder="Ask a question...") | |
| def try_to_solve(): | |
| return {solvingRow: gr.update(visible=True)} | |
| def show_map(): | |
| return {mapRow: gr.update(visible=True)} | |
| def hide_solve_box(): | |
| return {solvingRow: gr.update(visible=False)} | |
| def hide_map(): | |
| return {mapRow: gr.update(visible=False)} | |
| def hide_intro(): | |
| return {introRow: gr.update(visible=False)} | |
| def show_result(): | |
| return { | |
| chatRow: gr.update(visible=False), | |
| solvingRow: gr.update(visible=False), | |
| resultRow: gr.update(visible=True) | |
| } | |
| def declare_loss(): | |
| return "You failed! The real culprit got away... " | |
| def declare_victory(): | |
| return "Congratulations, you found the killer! " | |
| bot_functions = [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( | |
| disable_input, None, [msg[0], msg[1], msg[2], msg[3], msg[4]] | |
| ).then( | |
| bot_functions[i], [chatbot[i], questionCounter], chatbot[i] | |
| ).then( | |
| update_questions, questionCounter, questionCounter | |
| ).then( | |
| enable_input, None, [msg[0], msg[1], msg[2], msg[3], msg[4]] | |
| ) | |
| i += 1 | |
| mapButton.click(show_map, None, mapRow) | |
| closeMapButton.click(hide_map, None, mapRow) | |
| finishButton.click(try_to_solve, None, solvingRow) | |
| closeSolvingButton.click(hide_solve_box, None, solvingRow) | |
| closeIntroButton.click(hide_intro, None, introRow) | |
| i = 0 | |
| while i < len(roles): | |
| if i == culpritId: | |
| accusationButtons[i].click( | |
| show_result, None, [chatRow, solvingRow, resultRow] | |
| ).then( | |
| declare_victory, None, resultLabel | |
| ) | |
| else: | |
| accusationButtons[i].click( | |
| show_result, None, [chatRow, solvingRow, resultRow] | |
| ).then( | |
| declare_loss, None, resultLabel | |
| ) | |
| i += 1 | |
| iface.launch() | |