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)

def chat_completion(prompt, model="gpt-4", temperature=0):
    res = client.chat.completions.create(
        model=model,
        messages=[{"role": "user", "content": prompt}],
        temperature=temperature,
    )
    print(res.choices[0].message.content)

# Context, task and role
prompt = """
Kontext: Ich habe meine 3 Kumpels zum Abendessen eingeladen. Einer davon ist Veganer.
Aufgabe: Was könnte ich für uns zum Abendessen vorbereiten. Mach mir drei Vorschläge.
"""
chat_completion(prompt)

# Thinking Step by Step
prompt = "Wieviel ist 743 * 877? Hole tief Luft und gehe Schritt für Schritt vor."
chat_completion(prompt)

# Few Shot Learning
prompt = """
Ich gehe nach Hause --> 😊 gehe 🏠
Mein Hund ist traurig --> Mein 🐶 ist 😞
Ich renne schnell --> 😊 renne ⚡
Ich liebe meine Frau --> 😊 ❤️ meine Frau
Das Mädchen spielt mit dem Ball --> Das 👧 🎮 mit dem 🏀
Der Junge schreibt einem Mädchen einen Brief --> 
"""
chat_completion(prompt)