update

README.md

@@ -0,0 +1,13 @@
+---
+title: Chatbot Bk1
+emoji: 📈
+colorFrom: gray
+colorTo: green
+sdk: streamlit
+sdk_version: 1.29.0
+app_file: app.py
+pinned: false
+---
+
+ChatbotBK11
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,70 @@
+import streamlit as st
+from rag import rag
+import logging
+
+# 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
+guiding_questions = {
+    "Có những loại chương trình đào tạo thạc sĩ nào?": False,
+    "Cho tôi thông tin về chương trình đào tạo thạc sĩ ứng dụng?": False,
+    "Tiêu chuẩn huy chương vàng được quy định như thế nào?": False,
+    "Miễn thi được quy định như thế nào?": False,
+    "Yêu cầu ngoại ngữ đối với học viên thạc sĩ?": False
+}
+
+def main():
+    st.set_page_config(page_title="Chatbot BK1")
+
+    st.image(["logo.jpg"], width=100)
+    # with st.columns(3)[1]:
+    #     st.image(["logo.jpg"])
+
+    st.title("Chatbot Phòng Đào Tạo")
+    # st.markdown("<h1 style='text-align: center'>Chatbot Phòng Đào Tạo</h1>", unsafe_allow_html=True)
+    
+    st.subheader("Tôi có thể giải đáp các thắc mắc về quy định học vụ của Trường Đại Học Bách Khoa - ĐHQG TP.HCM", divider='rainbow')
+
+    # Hiển thị sidebar với các câu hỏi hướng dẫn
+    st.sidebar.subheader("Có thể bạn quan tâm những câu hỏi dưới đây:")
+
+    # set initial message
+    if "messages" not in st.session_state.keys():
+        st.session_state.messages = [
+            {"role": "assistant", "content": "Xin chào, tôi có thể giúp gì cho bạn"}
+        ]
+
+    if "messages" in st.session_state.keys():
+        # display messages
+        for message in st.session_state.messages:
+            with st.chat_message(message["role"]):
+                st.write(message["content"])
+
+    # get user input
+    user_prompt = st.chat_input()
+    for question in guiding_questions.keys():
+        if st.sidebar.button(question, key=question):
+            user_prompt = question
+            guiding_questions[question] = True  # Đánh dấu câu hỏi được chọn
+    handle_user_input(user_prompt)
+
+def handle_user_input(user_prompt):
+    if user_prompt is not None:
+        st.session_state.messages.append({"role": "user", "content": user_prompt})
+        with st.chat_message("user"):
+            st.write(user_prompt)
+
+    # process user input
+    if st.session_state.messages[-1]["role"] != "assistant":
+        with st.chat_message("assistant"):
+            with st.spinner("Loading..."):
+                ai_response = rag(user_prompt)
+                if ai_response == "Encountered some errors. Please recheck your request!":
+                    st.write("Xin lỗi, tôi không có thông tin về câu hỏi này!")
+                else:
+                    st.write(ai_response)
+
+        new_ai_message = {"role": "assistant", "content": ai_response}
+        st.session_state.messages.append(new_ai_message)
+
+
+if __name__ == '__main__':
+    main()