Update app.py

app.py

@@ -16,7 +16,6 @@ import os
 OPENAI_API_KEY = os.getenv("OPENAI_API")
 TOKEN=os.getenv('HF_TOKEN')
 subprocess.run(["huggingface-cli", "login", "--token", TOKEN, "--add-to-git-credential"])
-st.sidebar.title("Welcome to MBAL Chatbot")
 class PDFChatbot:
     def __init__(self):
        self.azure_client = openai.OpenAI()
@@ -98,13 +97,13 @@ Please provide a helpful response based on the insurance document content above.
             }
         ]
         # Add conversation history
-        for msg in self.conversation_history[-2:]:  # Keep last 6 messages for context
+        for msg in self.conversation_history[-6:]:  # Keep last 6 messages for context
            messages.append(msg)
         # Get response from Azure OpenAI
         response = self.azure_client.chat.completions.create(
            model="gpt-4o-mini",
            messages=messages,
-           max_tokens=300, #TODO
+           max_tokens=800, #TODO
            temperature=0.7
         )
         bot_response = response.choices[0].message.content
@@ -114,9 +113,8 @@ Please provide a helpful response based on the insurance document content above.
         return bot_response
 
 def main():
-    # st.set_page_config(page_title="Insurance PDF Chatbot", page_icon="🛡️", layout="wide")
-    st.title("🛡️ Insurance Policy Assistant")
-    st.markdown("Upload your insurance policy PDF and ask questions about your coverage, claims, deductibles, and more!")
+    # st.set_page_config(page_title="Chatbot t", page_icon="🛡️", layout="wide")
+    st.title("🛡️ Chatbot tư vấn bảo hiểm của công ty bảo hiểm nhân thọ MB Ageas Life")
     # Initialize chatbot
     if 'chatbot' not in st.session_state:
         st.session_state.chatbot = PDFChatbot()
@@ -130,12 +128,12 @@ def main():
             st.session_state.chat_history = []
             st.rerun()
     # Main chat interface
-    st.header("💬 Ask About Your Insurance Policy")
+    st.header("💬 Hãy đặt câu hỏi cho tôi về gói bảo hiểm bạn quan tâm")
     # Display chat history
     for i, (question, answer) in enumerate(st.session_state.chat_history):
         with st.container():
-            st.markdown(f"**You:** {question}")
-            st.markdown(f"**Insurance Assistant:** {answer}")
+            st.markdown(f"**Câu hỏi:** {question}")
+            st.markdown(f"**Trợ lý bảo hiểm:** {answer}")
             st.divider()
         # Chat input
     user_question = st.chat_input("Hãy đặt những câu hỏi về hợp đồng bảo hiểm cơ bản...")