# Store API keys in HF Space Settings -> Variables & Secrets
OPENAI_KEY = os.environ.get("OPENAI_KEY")

def call_gpt(prompt: str):
    import requests
    headers = {"Authorization": f"Bearer {OPENAI_KEY}"}
    response = requests.post(
        "https://api.openai.com/v1/chat/completions",
        json={"model": "gpt-3.5-turbo", "messages": [{"role": "user", "content": prompt}]},
        headers=headers
    )
    return response.json()["choices"][0]["message"]["content"]