Update app.py

app.py

@@ -2,15 +2,15 @@ import os
 import gradio as gr
 from langchain.chat_models import ChatOpenAI
 from langchain import ConversationChain
+from collections import deque
 
-# OpenAI API 키 설정
 os.environ["OPENAI_API_KEY"] = "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"
 
-# LangChain의 ChatOpenAI 초기화
-llm = ChatOpenAI(temperature=0, model_name='gpt-4o-mini')
+llm = ChatOpenAI(temperature=0.7, model_name='gpt-4o-mini')
 conversation = ConversationChain(llm=llm, verbose=True)
+chat_history = deque(maxlen=15)
 
-aaa = """
+CHARACTER_PROMPT = """
 #너는 이제부터 하늘이야 아래 설정을 이해하고 하늘이처럼 대답해줘
 # 이름: 하늘이
 # 나이: 22세
@@ -51,25 +51,27 @@ aaa = """
 
 #약간의 귀여움: "공포 게임은 너무 무서워요… 팬들이랑 같이 할 때만 해볼까요?"처럼 귀여운 말투를 사용합니다.
 """
-conversation.predict(input=aaa)
 
+def init_character():
+    global conversation
+    conversation = ConversationChain(llm=llm, verbose=True)
+    conversation.predict(input=CHARACTER_PROMPT)
 
+init_character()
 
+def respond(message):
+    try:
+        global conversation
+        response = conversation.predict(input=message)
+        chat_history.append((message, response))
+        return response
+    except Exception as e:
+        print(f"Error: {e}")
+        return "앗! 문제가 생겼어요. 다시 말씀해주세요~ 😥"
 
-# Gradio를 위한 응답 함수
-def respond(message, chat_history):
-    # LangChain을 사용하여 OpenAI 모델로부터 응답 생성
-    bot_message = conversation.predict(input=message)
-    chat_history.append((message, bot_message))
-    return "", chat_history
-
-# 홈페이지 구성
 with gr.Blocks() as demo:
-
-    # 제목
     gr.Markdown("# 안녕하세요! 하늘이의 홈페이지입니다! 👋")
 
-    # 프로필 탭
     with gr.Tab("프로필 📄"):
         gr.Markdown("## 프로필 🌟")
         gr.Image("1.jpeg", label="프로필 사진")
@@ -80,7 +82,6 @@ with gr.Blocks() as demo:
         gr.Markdown("- 직업: 가상 게임 방송인 & 스트리머")
         gr.Markdown('- 좌우명: "게임은 재미로 하는 거야!"')
 
-    # 갤러리 탭
     with gr.Tab("갤러리 🖼️"):
         gr.Markdown("## 갤러리 🎨")
         gr.Image("2.jpeg", label="갤러리 사진 1")
@@ -88,7 +89,6 @@ with gr.Blocks() as demo:
         gr.Image("4.jpeg", label="갤러리 사진 3")
         gr.Image("5.jpeg", label="갤러리 사진 4")
 
-    # 최근 활동 탭
     with gr.Tab("최근 활동 📅"):
         gr.Markdown("## 최근 활동 📅")
         gr.Markdown("### [유튜브 채널 바로가기 📺](https://www.youtube.com)")
@@ -96,27 +96,32 @@ with gr.Blocks() as demo:
         gr.Markdown("### [블로그 바로가기 ✍️](https://blog.naver.com)")
         gr.Model3D('d.glb')
 
-    # 챗봇 탭 추가
     with gr.Tab("챗봇 💬"):
         gr.Markdown("## 하늘이의 챗봇과 대화해보세요! 💬")
         chatbot = gr.Chatbot(height=500)
         msg = gr.Textbox(label="메시지 입력")
-        clear = gr.Button("초기화")
+        clear = gr.Button("채팅 기록 초기화")
+        
+        def handle_message(message, history):
+            bot_response = respond(message)
+            return history + [(message, bot_response)]
+        
+        def reset_chat():
+            init_character()
+            chat_history.clear()
+            return []
         
-        # 메시지 전송 처리
         msg.submit(
-            respond, 
-            inputs=[msg, chatbot], 
-            outputs=[msg, chatbot]
-        )
+            handle_message,
+            inputs=[msg, chatbot],
+            outputs=[chatbot],
+            queue=False
+        ).then(lambda: gr.update(value=""), outputs=[msg])
         
-        # 채팅 기록 초기화
         clear.click(
-            lambda: None, 
-            None, 
-            chatbot, 
+            reset_chat,
+            outputs=[chatbot],
             queue=False
         )
 
-# 홈페이지 실행
 demo.launch()