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)

# Audiodatei transkribieren
audio_file = open("C:\Pythonprojekte\stylistic_translation_of_an_audio\Data\die_temp_audio.wav", "rb")
transcript = client.audio.transcriptions.create(model="whisper-1", file=audio_file)

# Prompt
temp = 0.8
maxt = 1000
n = 1
user = "test-1"
modell = "gpt-3.5-turbo"
role_content = "Übersetze im Stil von Shakespeare: {}".format(transcript.text)
msg = [{"role": "user", "content": role_content}]

response = client.chat.completions.create(model=modell, messages=msg, temperature=temp, max_tokens=maxt, n=n, user=user)

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