import requests
import json
def generate_text(prompt):
headers = {
'Content-Type': 'application/json',
'Authorization': 'Bearer <API_KEY>'
}
data = {
"prompt": prompt,
"max_tokens": 1024,
"n":1,
"stop":None,
"temperature": 0.5
}
response = requests.post('https://api.openai.com/v1/engines/text-davinci-003/completions', headers=headers, data=json.dumps(data))
if response.status_code == 200:
return response.json()['choices'][0]['text']
else:
return 'Error: ' + str(response.status_code)