ChatGPTのAPIを簡単に使ってみた。

import openai
import deepl
import pickle
import datetime
import os

openai.organization = "org-EAsxLPeDr1QvJyQVOYggQmMc" # OPENAIで取得したorganization idを入力してください。
openai.api_key = os.environ["OPENAI_API_KEY"]　# OPENAIで取得したAPIKeyを入力してください。


def select_translate():
    dic = {'y': True, 'yes': True, 'n': False, 'no': False}
    while True:
        can_translated = dic.get(input('ENに翻訳して実行するか [Y]es/[N]o? >> ').lower(), -1)
        if type(can_translated) is bool:
            break
        print('Error! Input again.')
    return can_translated


def save_chat(messages,response):
    file_name = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
    chat_file = open(f'./outputs/{file_name}.pickle','wb')
    messages.append(response)
    pickle.dump(messages,chat_file)
    chat_file.close()


def main(input_prompt,can_translate=True):
    if can_translate:
        input_to_chatgpt = deepl.translate(source_language='JA',target_language='EN',text=input_prompt)
    else:
        input_to_chatgpt = input_prompt

    messages = [
            {"role": "system", "content": "You are a helpful assistant."},
            {"role": "user", "content": input_to_chatgpt}
        ]

    res = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=messages,
        max_tokens=100
    )

    # print(input_prompt)
    print(input_to_chatgpt)
    print(res['choices'][0]['message']['content'])
    print(f"{deepl.translate(source_language='EN',target_language='JA',text=res['choices'][0]['message']['content'])}")

    save_chat(messages=messages,response={'response':res})
    print(res)

if __name__ == "__main__":
    user_input = input('ChatGPTに聞きたいことを入力:')
    can_translated = select_translate()
    main(input_prompt=user_input,can_translate=bool(can_translated))