chatGPT 입력 처리하기

import requests

def get_answer(question):
    API_KEY = "YOUR_API_KEY"
    model = "text-davinci-002"
    prompt = (f"{question}"
             )

    response = requests.post(
        "https://api.openai.com/v1/engines/{}/jobs".format(model),
        headers={
            "Content-Type": "application/json",
            "Authorization": f"Bearer {API_KEY}"
        },
        json={
            "prompt": prompt,
            "max_tokens": 100,
            "n": 1,
            "stop": None,
            "temperature": 0.5,
        }
    )
    if response.status_code != 200:
        raise ValueError("Failed to generate completions")

    message = response.json()["choices"][0]["text"]
    return message.strip()

question = input("What is your question? ")
answer = get_answer(question)
print("ChatGPT: ", answer)