| # 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"] |