Wiesli_Jonas / app.py
Jonas Wiesli
fixed wrong json call
9e9bf1a
raw
history blame
2.94 kB
import gradio as gr
import os
import openai
openai.api_key = "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"
with gr.Blocks() as iface:
chatbot = gr.Chatbot()
msg = gr.Textbox()
def user(user_message, history):
return "", history + [[user_message, None]]
def bot(history):
bot_message = generateAIMessage(history)
history[-1][1] = bot_message
return history
def generateAIMessage(history):
message_history = [
{"role": "system", "content": "You are Steve Binner, 31, a security guard for a museum of medieval "
"history. You're absolutely sure that this place is haunted. Before the "
"spooky stuff started, you were really happy with your job, "
"but that changed real fast. The ghost is a french nobleman named 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. 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
msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
bot, chatbot, chatbot
)
iface.launch()