from dotenv import load_dotenv
from openai import OpenAI
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)

stream = client.chat.completions.create(
model="gpt-4",
messages=[{
    "role": "user",
    "content": "Schreibe eine 10-Zeilen-Geschichte für meinen 5-Jährigen Sohn."}],
stream=True,
)

for chunk in stream:
    if chunk.choices[0].delta.content is not None:
        print(chunk.choices[0].delta.content, end="")