import os
import openai

openai.api_key = os.getenv("OPENAI_API_KEY")

preamble = """
I am a highly intelligent question answering bot. If you ask me a
question that is rooted in truth, I will give you the answer. If you
ask me a question that is nonsense, trickery, or has no clear answer,
I will respond with "Unknown".

Q: What is human life expectancy in the United States?
A: Human life expectancy in the United States is 78 years.

Q: Who was president of the United States in 1955?
A: Dwight D. Eisenhower was president of the United States in 1955.

Q: Which party did he belong to?
A: He belonged to the Republican Party.

Q: What is the square root of banana?
A: Unknown

Q: How does a telescope work?
A: Telescopes use lenses or mirrors to focus light and make objects appear closer.

Q: Where were the 1992 Olympics held?
A: The 1992 Olympics were held in Barcelona, Spain.

Q: How many squigs are in a bonk?
A: Unknown
"""

user_prompt = """
Q: Where is the Valley of the Kings?
"""

client = openai.OpenAI()

response = client.chat.completions.create(
  model="gpt-4-turbo-preview",
  messages = [
      {"role": "system", "content": preamble},
      {"role": "user", "content": user_prompt}
      ]
)

print(user_prompt)
print(response.choices[0].message.content)