nnngoc commited on
Commit
85cc062
1 Parent(s): ac684ba
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -1,6 +1,5 @@
1
  import streamlit as st
2
  from rag import rag
3
- import logging
4
 
5
  # Tạo ra một dictionary lưu trữ mapping giữa câu hỏi và giá trị tương ứng của nút
6
  guiding_questions = {
@@ -12,7 +11,7 @@ guiding_questions = {
12
  }
13
 
14
  def main():
15
- st.set_page_config(layout="wide")
16
 
17
  st.image(["logo.jpg"], width=100)
18
  # with st.columns(3)[1]:
@@ -57,7 +56,10 @@ def handle_user_input(user_prompt):
57
  with st.chat_message("assistant"):
58
  with st.spinner("Loading..."):
59
  ai_response = rag(user_prompt)
60
- st.write(ai_response)
 
 
 
61
 
62
  new_ai_message = {"role": "assistant", "content": ai_response}
63
  st.session_state.messages.append(new_ai_message)
 
1
  import streamlit as st
2
  from rag import rag
 
3
 
4
  # Tạo ra một dictionary lưu trữ mapping giữa câu hỏi và giá trị tương ứng của nút
5
  guiding_questions = {
 
11
  }
12
 
13
  def main():
14
+ st.set_page_config(page_title="Chatbot BK1")
15
 
16
  st.image(["logo.jpg"], width=100)
17
  # with st.columns(3)[1]:
 
56
  with st.chat_message("assistant"):
57
  with st.spinner("Loading..."):
58
  ai_response = rag(user_prompt)
59
+ if ai_response == "Encountered some errors. Please recheck your request!":
60
+ st.write("Xin lỗi, tôi không có thông tin về câu hỏi này!")
61
+ else:
62
+ st.write(ai_response)
63
 
64
  new_ai_message = {"role": "assistant", "content": ai_response}
65
  st.session_state.messages.append(new_ai_message)