Spaces:
Runtime error
Runtime error
File size: 3,834 Bytes
36be4ce |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
from dotenv import load_dotenv
load_dotenv() # take environment variables from .env.
import gradio as gr
import openai
# Define a function to get the AI's reply using the OpenAI API
def get_ai_reply(message, model="gpt-3.5-turbo", system_message=None, temperature=0, message_history=[]):
# Initialize the messages list
messages = []
# Add the system message to the messages list
if system_message is not None:
messages += [{"role": "system", "content": system_message}]
# Add the message history to the messages list
if message_history is not None:
messages += message_history
# Add the user's message to the messages list
messages += [{"role": "user", "content": message}]
# Make an API call to the OpenAI ChatCompletion endpoint with the model and messages
completion = openai.ChatCompletion.create(
model=model,
messages=messages,
temperature=temperature
)
# Extract and return the AI's response from the API response
return completion.choices[0].message.content.strip()
# Define a function to handle the chat interaction with the AI model
def chat(message, chatbot_messages, history_state):
# Initialize chatbot_messages and history_state if they are not provided
chatbot_messages = chatbot_messages or []
history_state = history_state or []
# Try to get the AI's reply using the get_ai_reply function
try:
prompt = """
You are a bot created to simulate commands.
You are only allowed to simulate a command if the words preceding the command are "Simon says".
Anything said that isn't preceded with "Simon says" should be ignored, and you do nothing.
If the user doesn't give you a command, you do nothing.
Example of you simulating a command that is preceded with "Simon says":
User: "Simon says run in place"
You: ":: runs in place ::"
Example of you simulating doing nothing:
User: "run in place"
You: ":: does nothing ::"
"""
ai_reply = get_ai_reply(message, model="gpt-3.5-turbo", system_message=prompt.strip(), message_history=history_state)
# Append the user's message and the AI's reply to the chatbot_messages list
chatbot_messages.append((message, ai_reply))
# Append the user's message and the AI's reply to the history_state list
history_state.append({"role": "user", "content": message})
history_state.append({"role": "assistant", "content": ai_reply})
# Return None (empty out the user's message textbox), the updated chatbot_messages, and the updated history_state
except Exception as e:
# If an error occurs, raise a Gradio error
raise gr.Error(e)
return None, chatbot_messages, history_state
# Define a function to launch the chatbot interface using Gradio
def get_chatbot_app():
# Create the Gradio interface using the Blocks layout
with gr.Blocks() as app:
# Create a chatbot interface for the conversation
chatbot = gr.Chatbot(label="Conversation")
# Create a textbox for the user's message
message = gr.Textbox(label="Message")
# Create a state object to store the conversation history
history_state = gr.State()
# Create a button to send the user's message
btn = gr.Button(value="Send")
# Connect the send button to the chat function
btn.click(chat, inputs=[message, chatbot, history_state], outputs=[message, chatbot, history_state])
# Return the app
return app
# Call the launch_chatbot function to start the chatbot interface using Gradio
app = get_chatbot_app()
app.queue() # this is to be able to queue multiple requests at once
app.launch
|