import openai

# 設置OpenAI API的密鑰
openai.api_key = '11'

# 定義聊天模型的輸入提示
def get_chat_response(prompt):
    response = openai.Completion.create(
        engine='text-davinci-003',  # 指定使用的模型
        prompt=prompt,
        max_tokens=50  # 生成聊天回復的最大長度
    )
    chat_reply = response.choices[0].text.strip()
    return chat_reply

# 進入聊天循環
print("歡迎來到聊天AI！輸入'退出'可以結束對話。")
while True:
    user_input = input("用戶：")
    
    # 檢查用戶輸入是否為退出指令
    if user_input == '退出':
        print("聊天已結束。")
        break
    
    # 構建聊天模型的輸入提示
    chat_prompt = f"用戶：{user_input}\nAI："
    
    # 調用聊天模型獲取回復
    ai_response = get_chat_response(chat_prompt)
    
    # 打印AI的回復
    print("AI：" + ai_response)