Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+from language_detection import detect_browser_language
 from langchain_community.document_loaders import PyPDFLoader
 from langchain_core.messages import HumanMessage, AIMessageChunk, AIMessage
 from langchain_huggingface import HuggingFaceEmbeddings
@@ -13,6 +14,9 @@ from langgraph.errors import GraphRecursionError
 from graph import get_graph
 from langchain_core.runnables import RunnableConfig
 
+if 'language' not in st.session_state:
+    st.session_state.language = detect_browser_language()
+
 if 'read_file' not in st.session_state:
     st.session_state.read_file = False
     st.session_state.retriever = None
@@ -28,24 +32,44 @@ def get_session_by_id(session_id: str) -> BaseChatMessageHistory:
     return st.session_state.chat_history[session_id]
 
 if not st.session_state.read_file:
-    st.title('🤓 Upload your PDF to talk with it', anchor=False)
-    file = st.file_uploader('Upload a PDF file', type='pdf')
+
+    st.title('🤓 Faça o Upload de seu arquivo PDF para que possamos conversar sobre ele' 
+             if st.session_state.language.startswith("pt") 
+             else '🤓 Upload your PDF to talk with it',
+             anchor=False)
+    file = st.file_uploader('Carregue o seu arquivo PDF' 
+                            if st.session_state.language.startswith("pt") 
+                            else 'Upload a PDF file', type='pdf')
     if file:
-        with st.status('🤗 Booting up the things!', expanded=True):
-            with st.spinner('📁 Uploading the PDF...', show_time=True):
+        with st.status('🤗 Deixando tudo pronto!' 
+                       if st.session_state.language.startswith("pt") 
+                       else '🤗 Booting up the things!', expanded=True):
+            with st.spinner('📁 Carregando o PDF...' 
+                            if st.session_state.language.startswith("pt") 
+                            else '📁 Uploading the PDF...', show_time=True):
                 with open('file.pdf', 'wb') as f:
                     f.write(file.read())
                     loader = PyPDFLoader('file.pdf')
                     documents = loader.load_and_split(RecursiveCharacterTextSplitter(chunk_size=2000, chunk_overlap=200))
-                st.success('📁 File uploaded successfully!!!')
-            with st.spinner('🧐 Reading the file...', show_time=True):
+                st.success('📁 Arquivo carregado com sucesso!!!' 
+                           if st.session_state.language.startswith("pt") 
+                           else '📁 File uploaded successfully!!!')
+            with st.spinner('🧐 Lendo o arquivo...' 
+                           if st.session_state.language.startswith("pt") 
+                           else '🧐 Reading the file...', show_time=True):
                 vstore = InMemoryVectorStore.from_documents(documents, HuggingFaceEmbeddings(model_name='all-MiniLM-L6-v2'))
                 st.session_state.retriever = vstore.as_retriever()
-                st.success('🧐 File read successfully!!!')
+                st.success('🧐 Arquivo lido com sucesso!!!' 
+                           if st.session_state.language.startswith("pt") 
+                           else '🧐 File read successfully!!!')
             os.remove('file.pdf')
-            with st.spinner('😴 Waking up the LLM...', show_time=True):
+            with st.spinner('😴 Acordando o LLM...'
+                            if st.session_state.language.startswith("pt") 
+                            else '😴 Waking up the LLM...', show_time=True):
                 st.session_state.graph = get_graph(st.session_state.retriever)
-                st.success('😁 LLM awakened!!!')
+                st.success('😁 LLM pronto e operante!!!'
+                            if st.session_state.language.startswith("pt") 
+                            else '😁 LLM awakened!!!')
             st.balloons()
         placeholder = st.empty()
         for _ in range(5, -1, -1):
@@ -57,18 +81,25 @@ if not st.session_state.read_file:
 if st.session_state.read_file:
 
     st.title('🤗 DocAI', anchor=False)
-    st.subheader('Chat with your document!', anchor=False)
+    st.subheader('Converse com seu PDF!'
+                 if st.session_state.language.startswith("pt") 
+                 else 'Chat with your document!', anchor=False)
 
     if st.session_state.first_msg:
         st.session_state.first_msg = False
-        get_session_by_id('chat42').add_message(AIMessage(content='Hello, how are you? How about we talk about the '
-                                                                  'document you sent me to read?'))
+        get_session_by_id('chat42').add_message(AIMessage(content='E aí, como você tá? Que tal a gente conversar sobre o '
+                                                                  'arquivo que você me passou pra ler?'
+                                                                    if st.session_state.language.startswith("pt") 
+                                                                    else 'Hello, how are you? How about we talk about the '
+                                                                         'document you sent me to read?'))
 
     for msg in get_session_by_id('chat42').messages:
         with st.chat_message(name='user' if isinstance(msg, HumanMessage) else 'ai'):
             st.write(msg.content)
 
-    prompt = st.chat_input('Try to ask something about your file!')
+    prompt = st.chat_input('Tente perguntar algo sobre o seu arquivo!'
+                            if st.session_state.language.startswith("pt") 
+                            else 'Try to ask something about your file!')
     if prompt:
         with st.chat_message(name='user'):
             st.write(prompt)
@@ -102,7 +133,9 @@ if st.session_state.read_file:
                 for msg in get_message():
                     full_response += msg
                     if '<tool>' in full_response:
-                        with tool_placeholder.status('Reading document...', expanded=True):
+                        with tool_placeholder.status('Lendo documento...'
+                                                     if st.session_state.language.startswith("pt") 
+                                                     else 'Reading document...', expanded=True):
                             if 'tool_message_placeholder' not in placeholders:
                                 placeholders['tool_message_placeholder'] = st.empty()
                             placeholders['tool_message_placeholder'].write(full_response
@@ -116,7 +149,9 @@ if st.session_state.read_file:
                     else:
                         prompt_message_placeholder.write(full_response.replace('$', '\$'))
             except GraphRecursionError:
-                message = 'Não consegui responder a sua pergunta. 😥 Poderia me perguntar outra coisa?'
+                message = ('Não consegui responder a sua pergunta. 😥 Poderia me perguntar outra coisa?'
+                            if st.session_state.language.startswith("pt") 
+                            else 'I can't answer your question. 😥 Can u ask me other question?'
                 full_response = ''
                 for letter in message:
                     full_response += letter