Update app.py

app.py

@@ -2,19 +2,25 @@ import streamlit as st
 import torch
 from transformers import pipeline
 
-st.set_page_config(page_title="ChatViLD", page_icon="./app/static/ai.png", layout="centered", initial_sidebar_state="expanded")
+st.set_page_config(page_title="ViBidLawQA - Hệ thống hỏi đáp trực tuyến luật Việt Nam", page_icon="./app/static/ai.png", layout="centered", initial_sidebar_state="expanded")
 
 with open("./static/styles.css") as f:
     st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True)
 
-st.markdown("<h1 style='text-align: center;'>ChatViLD: Hệ thống hỏi đáp trực tuyến cho văn bản pháp luật Việt Nam</h1>", unsafe_allow_html=True)
+st.markdown(f"""
+    <div class=logo_area>
+        <img src="./app/static/ai.png"/>
+    </div>
+    """, unsafe_allow_html=True)
+st.markdown("<h2 style='text-align: center;'>ViBidLawQA</h2>", unsafe_allow_html=True)
+
 
 context = st.sidebar.text_area(label='Nội dung văn bản pháp luật Việt Nam:', placeholder='Vui lòng nhập nội dung văn bản pháp luật Việt Nam tại đây...', height=925)
 
 device = 0 if torch.cuda.is_available() else -1
 if 'model' not in st.session_state:
     print('Some errors occurred!')
-    st.session_state.model = pipeline("question-answering", model='./model', device=device)
+    st.session_state.model = pipeline("question-answering", model='./models/vi-mrc-large/model', device=device)
 
 def get_answer(context, question):
     return st.session_state.model(context=context, question=question, max_answer_len=512)
@@ -38,7 +44,7 @@ for message in st.session_state.messages:
     </div>
     """, unsafe_allow_html=True)
 
-if prompt := st.chat_input(placeholder='Xin chào, tôi có thể giúp được gì cho bạn?'):
+if prompt := st.chat_input(placeholder='Tôi có thể giúp được gì cho bạn?'):
     st.markdown(f"""
     <div class="user-message">
         <img src="./app/static/human.png" class="user-avatar" />
@@ -49,10 +55,16 @@ if prompt := st.chat_input(placeholder='Xin chào, tôi có thể giúp được
     
     respond = get_answer(context=context, question=prompt)['answer']
 
+    respond = respond.strip()
+    respond = respond.rstrip('.,!?;:')
+    respond = respond.capitalize()
+    if not respond.endswith('.'):
+        respond += '.'
+
     st.markdown(f"""
     <div class="assistant-message">
         <img src="./app/static/ai.png" class="assistant-avatar" />
         <div class="stMarkdown">{respond}</div>
     </div>
     """, unsafe_allow_html=True)
-    st.session_state.messages.append({'role': 'assistant', 'content': respond})
+    st.session_state.messages.append({'role': 'assistant', 'content': respond})