Update app.py

app.py

@@ -26,7 +26,6 @@ url_api_extract_reference_model = f"{routing_response_module}/extract_references
 with open("./static/styles.css") as f:
     st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
 
-
 if 'messages' not in st.session_state:
     st.session_state.messages = [{'role': 'assistant', 'content': "Xin chào. Tôi là trợ lý AI giáo dục Việt Nam được phát triển bởi Đào Thị Ngọc Ánh. Rất vui khi được hỗ trợ bạn trong học tập!"}]
 
@@ -162,130 +161,130 @@ if prompt := st.chat_input(placeholder='Tôi có thể giúp được gì cho b
     
     message_placeholder = st.empty()
     
-    full_response = ""
-    classify_result = classify_question(question=prompt).json()
-
-    print(f"The type of user query: {classify_result}")
-
-    if classify_result == "EDUCATION_RELATED":
-        retrieved_context = retrieve_context(question=prompt, top_k=10)
-        retrieved_context = [item['text'] for item in retrieved_context]
-        reranked_context = rerank_context(url_rerank_module=url_api_reranker_model,
-                                      question=prompt,
-                                      relevant_docs=retrieved_context,
-                                      top_k=5)[0]
-        
-        abs_answer = get_abstractive_answer(context=reranked_context, question=prompt)
-
-        if isinstance(abs_answer, str):
-            full_response = abs_answer
-            message_placeholder.markdown(f"""
-            <div class="assistant-message">
-                <img src="./app/static/ai.jpg" class="assistant-avatar" />
-                <div class="stMarkdown">{full_response}</div>
-            </div>
-            """, unsafe_allow_html=True)
-        else:
-            full_response = ""
-            for line in abs_answer.iter_lines():
-                if line:
-                    line = line.decode('utf-8')
-                    if line.startswith('data: '):
-                        data_str = line[6:]
-                        if data_str == '[DONE]':
-                            break
+    # full_response = ""
+    # classify_result = classify_question(question=prompt).json()
+
+    # print(f"The type of user query: {classify_result}")
+
+    # if classify_result == "EDUCATION_RELATED":
+    retrieved_context = retrieve_context(question=prompt, top_k=10)
+    retrieved_context = [item['text'] for item in retrieved_context]
+    reranked_context = rerank_context(url_rerank_module=url_api_reranker_model,
+                                  question=prompt,
+                                  relevant_docs=retrieved_context,
+                                  top_k=5)[0]
+    
+    abs_answer = get_abstractive_answer(context=reranked_context, question=prompt)
+
+    if isinstance(abs_answer, str):
+        full_response = abs_answer
+        message_placeholder.markdown(f"""
+        <div class="assistant-message">
+            <img src="./app/static/ai.jpg" class="assistant-avatar" />
+            <div class="stMarkdown">{full_response}</div>
+        </div>
+        """, unsafe_allow_html=True)
+    else:
+        full_response = ""
+        for line in abs_answer.iter_lines():
+            if line:
+                line = line.decode('utf-8')
+                if line.startswith('data: '):
+                    data_str = line[6:]
+                    if data_str == '[DONE]':
+                        break
+                    
+                    try:
+                        data = json.loads(data_str)
+                        token = data.get('token', '')
+                        full_response += token
                         
-                        try:
-                            data = json.loads(data_str)
-                            token = data.get('token', '')
-                            full_response += token
-                            
-                            message_placeholder.markdown(f"""
-                            <div class="assistant-message">
-                                <img src="./app/static/ai.jpg" class="assistant-avatar" />
-                                <div class="stMarkdown">{full_response}●</div>
-                            </div>
-                            """, unsafe_allow_html=True)
-                            
-                        except json.JSONDecodeError:
-                            pass
+                        message_placeholder.markdown(f"""
+                        <div class="assistant-message">
+                            <img src="./app/static/ai.jpg" class="assistant-avatar" />
+                            <div class="stMarkdown">{full_response}●</div>
+                        </div>
+                        """, unsafe_allow_html=True)
+                        
+                    except json.JSONDecodeError:
+                        pass
         
             # refs = st.expander("Tài liệu tham khảo", expanded=False)
             # refs_list = get_references(context=reranked_context, question=prompt, answer=full_response)
             # print(refs_list)
             # refs.write(f"{refs_list}")
 
-    elif classify_result == "ABOUT_CHATBOT":
-        answer = introduce_system(question=prompt)
-
-        if isinstance(answer, str):
-            full_response = answer
-            message_placeholder.markdown(f"""
-            <div class="assistant-message">
-                <img src="./app/static/ai.jpg" class="assistant-avatar" />
-                <div class="stMarkdown">{full_response}</div>
-            </div>
-            """, unsafe_allow_html=True)
-        else:
-            full_response = ""
-            for line in answer.iter_lines():
-                if line:
-                    line = line.decode('utf-8')
-                    if line.startswith('data: '):
-                        data_str = line[6:]
-                        if data_str == '[DONE]':
-                            break
+    # elif classify_result == "ABOUT_CHATBOT":
+    #     answer = introduce_system(question=prompt)
+
+    #     if isinstance(answer, str):
+    #         full_response = answer
+    #         message_placeholder.markdown(f"""
+    #         <div class="assistant-message">
+    #             <img src="./app/static/ai.jpg" class="assistant-avatar" />
+    #             <div class="stMarkdown">{full_response}</div>
+    #         </div>
+    #         """, unsafe_allow_html=True)
+    #     else:
+    #         full_response = ""
+    #         for line in answer.iter_lines():
+    #             if line:
+    #                 line = line.decode('utf-8')
+    #                 if line.startswith('data: '):
+    #                     data_str = line[6:]
+    #                     if data_str == '[DONE]':
+    #                         break
                         
-                        try:
-                            data = json.loads(data_str)
-                            token = data.get('token', '')
-                            full_response += token
+    #                     try:
+    #                         data = json.loads(data_str)
+    #                         token = data.get('token', '')
+    #                         full_response += token
                             
-                            message_placeholder.markdown(f"""
-                            <div class="assistant-message">
-                                <img src="./app/static/ai.jpg" class="assistant-avatar" />
-                                <div class="stMarkdown">{full_response}●</div>
-                            </div>
-                            """, unsafe_allow_html=True)
+    #                         message_placeholder.markdown(f"""
+    #                         <div class="assistant-message">
+    #                             <img src="./app/static/ai.jpg" class="assistant-avatar" />
+    #                             <div class="stMarkdown">{full_response}●</div>
+    #                         </div>
+    #                         """, unsafe_allow_html=True)
                             
-                        except json.JSONDecodeError:
-                            pass
+    #                     except json.JSONDecodeError:
+    #                         pass
     
-    else:
-        answer = response_unrelated_question(question=prompt)
-
-        if isinstance(answer, str):
-            full_response = answer
-            message_placeholder.markdown(f"""
-            <div class="assistant-message">
-                <img src="./app/static/ai.jpg" class="assistant-avatar" />
-                <div class="stMarkdown">{full_response}</div>
-            </div>
-            """, unsafe_allow_html=True)
-        else:
-            full_response = ""
-            for line in answer.iter_lines():
-                if line:
-                    line = line.decode('utf-8')
-                    if line.startswith('data: '):
-                        data_str = line[6:]
-                        if data_str == '[DONE]':
-                            break
+    # else:
+    #     answer = response_unrelated_question(question=prompt)
+
+    #     if isinstance(answer, str):
+    #         full_response = answer
+    #         message_placeholder.markdown(f"""
+    #         <div class="assistant-message">
+    #             <img src="./app/static/ai.jpg" class="assistant-avatar" />
+    #             <div class="stMarkdown">{full_response}</div>
+    #         </div>
+    #         """, unsafe_allow_html=True)
+    #     else:
+    #         full_response = ""
+    #         for line in answer.iter_lines():
+    #             if line:
+    #                 line = line.decode('utf-8')
+    #                 if line.startswith('data: '):
+    #                     data_str = line[6:]
+    #                     if data_str == '[DONE]':
+    #                         break
                         
-                        try:
-                            data = json.loads(data_str)
-                            token = data.get('token', '')
-                            full_response += token
+    #                     try:
+    #                         data = json.loads(data_str)
+    #                         token = data.get('token', '')
+    #                         full_response += token
                             
-                            message_placeholder.markdown(f"""
-                            <div class="assistant-message">
-                                <img src="./app/static/ai.jpg" class="assistant-avatar" />
-                                <div class="stMarkdown">{full_response}●</div>
-                            </div>
-                            """, unsafe_allow_html=True)
+    #                         message_placeholder.markdown(f"""
+    #                         <div class="assistant-message">
+    #                             <img src="./app/static/ai.jpg" class="assistant-avatar" />
+    #                             <div class="stMarkdown">{full_response}●</div>
+    #                         </div>
+    #                         """, unsafe_allow_html=True)
                             
-                        except json.JSONDecodeError:
-                            pass   
+    #                     except json.JSONDecodeError:
+    #                         pass   
 
     message_placeholder.markdown(f"""
     <div class="assistant-message">