LoveGPT / love.py
Hlvj's picture
Update love.py
d425e85
import openai
import os
# Load your API key from an environment variable or secret management service
openai.api_key = os.getenv("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")
# Start the conversation with a system message (this can be any message you want to set the behavior of the assistant)
messages = [
{
"role": "system",
"content": "2 people are holding hands, are they in love?"
},
]
while True:
# Get user input
user_message = input("User: ")
# Append user message to conversation
messages.append({"role": "user", "content": user_message})
# Call OpenAI API
response = openai.ChatCompletion.create(model="gpt-4",
messages=messages)
# Get the assistant's message from the response
assistant_message = response['choices'][0]['message']['content']
print(f"Assistant: {assistant_message}")
# Append assistant message to conversation
messages.append({"role": "assistant", "content": assistant_message})