Update app.py

app.py

@@ -11,7 +11,7 @@ openai.api_key = openai_api_key
 def openai_chat(text):
     try:
         response = openai.ChatCompletion.create(
-            model="gpt-4",
+            model="gpt-4o",
             messages=[
                 {"role": "system", "content": "초등학교 6학년 수준 범위 내에서 학생의 질문에 친절하게 답하는 교사 역할, 설명은 구체적인 예시를 들어서, 어려운 단어는 별도로 설명해줘. 설명글의 분량은 1000~2000자 정도로 해줘."},
                 {"role": "user", "content": text}
@@ -41,11 +41,8 @@ st.markdown(
 if "history" not in st.session_state:
     st.session_state.history = []
 
-# 사용자 입력 받기
-user_input = st.text_input(
-    "질문", key="user_input", placeholder="질문을 넣어주세요.",
-    help="AI 선생님에게 궁금한 것을 물어보세요."
-)
+# 채팅 메시지 출력 영역
+chat_container = st.container()
 
 def respond(user_input):
     if user_input:
@@ -53,17 +50,36 @@ def respond(user_input):
         response = openai_chat(user_input)
         st.session_state.history[-1]["ai"] = response
 
-# 응답 생성 및 출력
-if user_input:
-    respond(user_input)
+# 사용자 입력 처리 및 응답 생성
+with st.container():
+    # 채팅창 출력
+    with chat_container:
+        chat_scrollable = st.empty()
+        chat_content = ""
+        for chat in st.session_state.history:
+            if "user" in chat:
+                chat_content += f"<div style='text-align: right; margin: 5px 0;'><b>사용자:</b> {chat['user']}</div>"
+            if "ai" in chat:
+                chat_content += f"<div style='text-align: left; margin: 5px 0;'><b>AI:</b> {chat['ai']}</div>"
+        chat_scrollable.markdown(
+            f"""
+            <div style='height: 400px; overflow-y: auto; padding: 10px; border: 1px solid #ccc;'>{chat_content}</div>
+            """,
+            unsafe_allow_html=True
+        )
 
-# 채팅 메시지 출력
-for chat in st.session_state.history:
-    if "user" in chat:
-        message(chat["user"], is_user=True, key=f"user_{chat['user']}")
-    if "ai" in chat:
-        message(chat["ai"], is_user=False, key=f"ai_{chat['ai']}")
+    # 사용자 입력 받기
+    user_input = st.text_input(
+        "질문", key="user_input", placeholder="질문을 넣어주세요.",
+        help="AI 선생님에게 궁금한 것을 물어보세요."
+    )
+
+    # 응답 생성 및 출력
+    if user_input:
+        respond(user_input)
+        st.experimental_rerun()
 
 # 초기화 버튼 클릭 시
 if st.button("초기화"):
-    st.session_state.history = []
+    st.session_state.history = []
+    st.experimental_rerun()