Update app.py

app.py

@@ -94,6 +94,8 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, pr
             temperature=temperature,
             max_new_tokens=max_tokens,
             top_k=top_k,
+            # Adiciona um prompt de sistema para responder em português
+            system_prompt="Por favor, responda em português."
         )
 
     progress(0.75, desc="Definindo memória de buffer...")
@@ -114,7 +116,7 @@ def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, pr
     )
     progress(0.9, desc="Concluído!")
     return qa_chain
-
+    
 # Função para gerar um nome de coleção válido
 def create_collection_name(filepath):
     collection_name = Path(filepath).stem
@@ -161,8 +163,10 @@ def format_chat_history(message, chat_history):
 
 # Função para realizar a conversa com o chatbot
 def conversation(qa_chain, message, history):
-    formatted_chat_history = format_chat_history(message, history)
-    response = qa_chain({"question": message, "chat_history": formatted_chat_history})
+    # Adiciona uma instrução para responder em português
+    message_with_language = "Por favor, responda em português: " + message
+    formatted_chat_history = format_chat_history(message_with_language, history)
+    response = qa_chain({"question": message_with_language, "chat_history": formatted_chat_history})
     response_answer = response["answer"]
     if response_answer.find("Resposta útil:") != -1:
         response_answer = response_answer.split("Resposta útil:")[-1]