Update app.py

app.py

@@ -439,42 +439,6 @@ def generate_question(question):
     print('T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5T5')
     return response
 
-# st.title("Chatbot Roberta")
-# st.write("Hi! Tôi là trợ lý của bạn trong việc trả lời các câu hỏi.")
-# text = st.text_input("User: ", key="input")
-
-# if 'chat_history' not in st.session_state:
-#     st.session_state['chat_history'] = []
-
-
-# def get_response(text):
-#     st.subheader("The Answer is:")
-#     st.write(text)
-#     answer, context = chatRoberta(text)
-#     result = answer[0]['answer']
-#     if result == "":
-#         return "Xin lỗi, tôi không thể tìm được đáp án phù hợp cho câu hỏi này ... Hãy thử trả lời bằng câu hỏi khác!"
-#     return result
-
-# if st.button("Chat!"):
-#     st.session_state['chat_history'].append(("User", text))
-
-#     response = get_response(text)
-
-#     st.subheader("The Response is:")
-#     message = st.empty()
-#     result = ""
-#     for chunk in response:
-#         result += chunk
-#         message.markdown(result + "❚ ")
-#     message.markdown(result)
-#     st.session_state['chat_history'].append(("Bot", result))
-
-# for i, (sender, message) in enumerate(st.session_state['chat_history']):
-#     if sender == "User":
-#         st.text_area(f"User:", value=message, height=100, max_chars=None, key=f"user_{i}")
-#     else:
-#         st.text_area(f"Bot:", value=message, height=100, max_chars=None, key=f"bot_{i}")
 
 def get_response(text):
     # Thay thế hàm này bằng model của bạn để lấy câu trả lời từ bot
@@ -487,6 +451,40 @@ def get_response(text):
         return generate_question(text)
     return result
 
+# st.title("General Law Chatbot")
+
+# # Khởi tạo lịch sử tin nhắn
+# if "messages" not in st.session_state:
+#     st.session_state.messages = []
+
+# # Hiển thị các tin nhắn từ lịch sử
+# for message in st.session_state.messages:
+#     with st.chat_message(message["role"]):
+#         st.markdown(message["content"])
+
+# # Nhận input từ người dùng
+# if prompt := st.chat_input("What is up?"):
+#     # Thêm tin nhắn của người dùng vào lịch sử
+#     st.session_state.messages.append({"role": "user", "content": prompt})
+    
+#     # Hiển thị tin nhắn của người dùng trong giao diện
+#     with st.chat_message("user"):
+#         st.markdown(prompt)
+
+#     # Lấy câu trả lời từ bot
+#     response = get_response(prompt)
+
+#     # Hiển thị câu trả lời của bot trong giao diện
+#     with st.chat_message("assistant"):
+#         st.markdown(response)
+
+#     # Thêm câu trả lời của bot vào lịch sử
+#     st.session_state.messages.append({"role": "assistant", "content": response})
+
+# Đọc file CSV và tạo dictionary từ file
+df = pd.read_csv("./data-law/Data_law_2807.csv")  # Đường dẫn đến file CSV của bạn
+qa_dict = dict(zip(df['question'], df['answer']))
+
 st.title("General Law Chatbot")
 
 # Khởi tạo lịch sử tin nhắn
@@ -507,8 +505,11 @@ if prompt := st.chat_input("What is up?"):
     with st.chat_message("user"):
         st.markdown(prompt)
 
-    # Lấy câu trả lời từ bot
-    response = get_response(prompt)
+    # Kiểm tra xem prompt có trong dictionary không
+    if prompt in qa_dict:
+        response = qa_dict[prompt]
+    else:
+        response = "Sorry, I don't understand that question."
 
     # Hiển thị câu trả lời của bot trong giao diện
     with st.chat_message("assistant"):