import openai

openai.api_key  = " ... "

def get_completion(prompt, model="gpt-3.5-turbo"):
    messages = [{"role": "user", "content": prompt}]
    response = openai.ChatCompletion.create(
        model=model,
        messages=messages,
        temperature=0, # this is the degree of randomness of the model's output
    )
    return response.choices[0].message["content"]

import openai
from secret import apikey

print("使用情形: https://platform.openai.com/account/usage")

message_history = []
openai.api_key = apikey
def talk_with(persona):
    while True:
        user_input = input('> ')
        if user_input == "":
            return message_history

        message_history.append({"role": "user", "content": user_input})
        query = [{"role": "system", "content": persona}]
        query.extend(message_history)
        result = openai.ChatCompletion.create(
          model="gpt-3.5-turbo",
          messages=query
        )
        gpt_message = result["choices"][0]["message"]
        message_history.append({"role": gpt_message["role"], "content": gpt_message["content"]})
        price = (result['usage']['total_tokens'] / 1000) * 0.002 * 31
        print(f"諮詢費: {price}元", )
        print("GPT: " + gpt_message["content"])

try:
    talk_with(persona="""You are a helpful engineer expert. You answer question by providing a short explanation and a list of easy to follow steps.""")
except:
    print(message_history)