Update app.py

app.py

@@ -109,6 +109,8 @@ def conversation(qa_chain, message, history):
     return qa_chain, gr.update(value=""), new_history, response_source1, response_source1_page, response_source2, response_source2_page, response_source3, response_source3_page
 
 
+# ... (código anterior permanece igual)
+
 def demo():
     """Main demo application with enhanced layout"""
     theme = gr.themes.Default(
@@ -131,6 +133,7 @@ def demo():
     with gr.Blocks(theme=theme, css=custom_css) as demo:
         vector_db = gr.State()
         qa_chain = gr.State()
+        language = gr.State(value="en")  # Novo estado para controle de idioma
         
         # Header
         gr.HTML(
@@ -171,6 +174,14 @@ def demo():
                         type="index"
                     )
                     
+                    # Novo botão de seleção de idioma
+                    language_btn = gr.Radio(
+                        choices=["English", "Português"],
+                        label="Response Language",
+                        value="English",
+                        type="value"
+                    )
+                    
                     with gr.Accordion("Advanced Settings", open=False):
                         slider_temperature = gr.Slider(
                             minimum=0.01,
@@ -269,6 +280,12 @@ def demo():
         )
 
         # Event Handlers
+        language_btn.change(
+            lambda x: "en" if x == "English" else "pt",
+            inputs=language_btn,
+            outputs=language
+        )
+        
         db_btn.click(
             initialize_database,
             inputs=[document],
@@ -288,14 +305,14 @@ def demo():
 
         msg.submit(
             conversation,
-            inputs=[qa_chain, msg, chatbot],
+            inputs=[qa_chain, msg, chatbot, language],
             outputs=[qa_chain, msg, chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page],
             queue=False
         )
         
         submit_btn.click(
             conversation,
-            inputs=[qa_chain, msg, chatbot],
+            inputs=[qa_chain, msg, chatbot, language],
             outputs=[qa_chain, msg, chatbot, doc_source1, source1_page, doc_source2, source2_page, doc_source3, source3_page],
             queue=False
         )
@@ -309,5 +326,36 @@ def demo():
 
     demo.queue().launch(debug=True)
 
+# Modifique a função de conversão para incluir o idioma
+def conversation(qa_chain, message, history, lang):
+    """Handle conversation and document analysis"""
+    # Adicione instrução de idioma à mensagem
+    if lang == "pt":
+        message += " (Responda em Português)"
+    else:
+        message += " (Respond in English)"
+    
+    formatted_chat_history = format_chat_history(message, history)
+    response = qa_chain.invoke({"question": message, "chat_history": formatted_chat_history})
+    response_answer = response["answer"]
+    
+    # Remova a instrução de idioma do histórico do chat
+    if "(Respond" in message:
+        message = message.split(" (Respond")[0]
+    
+    if response_answer.find("Helpful Answer:") != -1:
+        response_answer = response_answer.split("Helpful Answer:")[-1]
+    
+    response_sources = response["source_documents"]
+    response_source1 = response_sources[0].page_content.strip()
+    response_source2 = response_sources[1].page_content.strip()
+    response_source3 = response_sources[2].page_content.strip()
+    response_source1_page = response_sources[0].metadata["page"] + 1
+    response_source2_page = response_sources[1].metadata["page"] + 1
+    response_source3_page = response_sources[2].metadata["page"] + 1
+    new_history = history + [(message, response_answer)]
+    
+    return qa_chain, gr.update(value=""), new_history, response_source1, response_source1_page, response_source2, response_source2_page, response_source3, response_source3_page
+
 if __name__ == "__main__":
     demo()