Update app.py

app.py

@@ -171,117 +171,120 @@ 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()
+    try:
+        classify_result = classify_question(question=prompt).json()
 
-    print(f"The type of user query: {classify_result}")
-
-    if classify_result == "BIDDING_RELATED":
-        abs_answer = get_abstractive_answer(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
+        print(f"The type of user query: {classify_result}")
+    
+        if classify_result == "BIDDING_RELATED":
+            abs_answer = get_abstractive_answer(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
                             
-                            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)
+                            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
+    
+        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
                             
-                        except json.JSONDecodeError:
-                            pass
-
-    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)
+                            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
+        
         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
-                            
-                            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
+            answer = response_unrelated_question(question=prompt)
     
-    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
-                            
-                            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)
+            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
                             
-                        except json.JSONDecodeError:
-                            pass
+                            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
+    except error:
+        full_response = "Hiện tại bot đang nghỉ xíu để sạc pin 🔌. Bạn hãy quay lại sau nhé!"
 
     message_placeholder.markdown(f"""
     <div class="assistant-message">