Update app.py

app.py

@@ -98,19 +98,16 @@ def handle_userinput(user_question):
     # 대화 체인을 사용하여 사용자 질문에 대한 응답을 생성합니다.
     response = st.session_state.conversation({'question': user_question})
     # 대화 기록을 저장합니다.
-    if 'chat_history' not in st.session_state:
-        st.session_state.chat_history = []
+    st.session_state.chat_history = response['chat_history']
 
-    st.session_state.chat_history.append({'user': user_question, 'bot': response['answer']})
+    for i, message in enumerate(st.session_state.chat_history):
+        if i % 2 == 0:
+            st.write(user_template.replace(
+                "{{MSG}}", message.content), unsafe_allow_html=True)
+        else:
+            st.write(bot_template.replace(
+                "{{MSG}}", message.content), unsafe_allow_html=True)
 
-    # 업데이트된 대화를 표시
-    chat_output = ""
-    for message in st.session_state.chat_history:
-        chat_output += f"You: {message['user']}\n"
-        chat_output += f"Bot: {message['bot']}\n"
-
-    st.session_state.chat_output = chat_output
-    st.text_area("Chat with the System:", value=chat_output, height=200, max_chars=None, key="chat_output", disabled=True)
 
 def main():
     load_dotenv()