import os
os.environ["OPENAI_API_KEY"] = os.environ.get("GPT_API_KEY")

from langchain.chat_models import ChatOpenAI
from langchain import ConversationChain
import gradio as gr

# 랭체인 모델 초기화
llm = ChatOpenAI(temperature=0, model_name='gpt-4o-mini')
conversation = ConversationChain(llm=llm, verbose=True)

# Gradio 챗봇 응답 함수
def response(message, history):
    # 랭체인을 사용하여 대화 예측
    response_text = conversation.predict(input=message)
    return response_text

# Gradio 챗봇 인터페이스 설정 및 실행
gr.ChatInterface(
    fn=response,
    type="messages"
).launch()