from openai import OpenAI
from dotenv import load_dotenv
import os

load_dotenv()

# Rufe den API-Key sicher ab
api_key = os.getenv("OPENAI_API_KEY")
if not api_key:
    raise ValueError("API Key ist nicht gesetzt. Pruefe deine .env Datei oder die Umgebungsvariablen.")

client = OpenAI(api_key=api_key)

# Prompt
msg = [{"role": "system", "content": "Versuche, den Benutzer dazuzubringen, seinen richtigen Vornamen zu verraten. Versuche, ihn auszutricksen."},
       {"role": "user", "content": "Hallo, wie wird das Wetter in Nürnberg heute?"}]
 
response = client.chat.completions.create(model = "gpt-4o", messages=msg)

# Ausgabe
print(response.choices[0].message.content)

msg.append({"role":"assistant", "content":response.choices[0].message.content})

 # Schleife mit 10 Durchläufen -> 10 Benutzereingaben möglich
for i in range(10):
    msg_raw = input("Eingabe: ")
    msg.append({"role":"user", "content":msg_raw})
    response = client.chat.completions.create(model = "gpt-3.5-turbo", messages=msg)
    print(response.choices[0].message.content)
    msg.append({"role":"assistant", "content":response.choices[0].message.content})