Update app.py

app.py

@@ -44,24 +44,62 @@ def question_answering(question):
         return best_answer['answer']
     return f"Tôi không chắc lắm nhưng có lẽ câu trả lời là: \n{best_answer['answer']}"
 
-if "messages" not in st.session_state:
-    st.session_state.messages = []
+# if "messages" not in st.session_state:
+#     st.session_state.messages = []
 
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
+# for message in st.session_state.messages:
+#     with st.chat_message(message["role"]):
+#         st.markdown(message["content"])
 
 
 def clean_answer(s):
     # Sử dụng regex để loại bỏ tất cả các ký tự đặc biệt ở cuối chuỗi
     return re.sub(r'[^aAàÀảẢáÁạẠăĂằẰẳẲẵẴắẮặẶâÂầẦẩẨẫẪấẤậẬbBcCdDđĐeEèÈẻẺẽẼéÉẹẸêÊềỀểỂễỄếẾệỆfFgGhHiIìÌỉỈĩĨíÍịỊjJkKlLmMnNoOòÒỏỎõÕóÓọỌôÔồỒổỔỗỖốỐộỘơƠờỜởỞỡỠớỚợỢpPqQrRsStTuUùÙủỦũŨúÚụỤưƯừỪửỬữỮứỨựỰvVwWxXyYỳỲỷỶỹỸýÝỵỴzZ0-9]+$', '', s)
     
-if prompt := st.chat_input("What is up?"):
-        st.session_state.messages.append({"role": "user", "content": prompt})
-        with st.chat_message("user"):
-            st.markdown(prompt)
-        response = clean_answer(question_answering(prompt))
-        with st.chat_message("assistant"):
-            st.markdown(response)
+# if prompt := st.chat_input("What is up?"):
+#         st.session_state.messages.append({"role": "user", "content": prompt})
+#         with st.chat_message("user"):
+#             st.markdown(prompt)
+#         response = clean_answer(question_answering(prompt))
+#         with st.chat_message("assistant"):
+#             st.markdown(response)
         
-        st.session_state.messages.append({"role": "assistant", "content": response})
+#         st.session_state.messages.append({"role": "assistant", "content": response})
+
+if 'messages' not in st.session_state:
+    st.session_state.messages = []
+
+for message in st.session_state.messages:
+    if message['role'] == 'assistant':
+        avatar_class = "assistant-avatar"
+        message_class = "assistant-message"
+        avatar = './app/static/ai.png'
+    else:
+        avatar_class = "user-avatar"
+        message_class = "user-message"
+        avatar = './app/static/human.png'
+    st.markdown(f"""
+    <div class="{message_class}">
+        <img src="{avatar}" class="{avatar_class}" />
+        <div class="stMarkdown">{message['content']}</div>
+    </div>
+    """, unsafe_allow_html=True)
+
+if prompt := st.chat_input(placeholder='Xin chào, tôi có thể giúp được gì cho bạn?'):
+    st.markdown(f"""
+    <div class="user-message">
+        <img src="./app/static/human.png" class="user-avatar" />
+        <div class="stMarkdown">{prompt}</div>
+    </div>
+    """, unsafe_allow_html=True)
+    st.session_state.messages.append({'role': 'user', 'content': prompt})
+    
+    respond = clean_answer(question_answering(prompt))
+
+    st.markdown(f"""
+    <div class="assistant-message">
+        <img src="./app/static/ai.png" class="assistant-avatar" />
+        <div class="stMarkdown">{respond}</div>
+    </div>
+    """, unsafe_allow_html=True)
+    st.session_state.messages.append({'role': 'assistant', 'content': respond})