import os
from openai import OpenAI
client = OpenAI()
from search import answer_question


# 最初にメッセージを表示する
print("質問を入力してください")

conversation_history = []

while True:
    # ユーザーの入力した文字を変数「user_input」に格納
    user_input = input()

    # ユーザーの入力した文字が「exit」の場合はループを抜ける
    if user_input == "exit":
        break
    
    conversation_history.append({"role": "user", "content": user_input})
    response = client.chat.completions.create(
        model="gpt-4o-mini",
        messages = conversation_history,
    )

    # ChatGPTの応答内容を会話履歴に追加
    chatgpt_response = response.choices[0].message.content
    conversation_history.append({"role": "assistant", "content":chatgpt_response})

    #ターミナルにChatGPTの返答を表示
    print("ChatGPT:", chatgpt_response)