Update app.py

app.py

@@ -81,17 +81,19 @@ def format_chat_history(message, chat_history):
 
 def conversation(qa_chain, message, history, language):
     formatted_chat_history = format_chat_history(message, history)
-    response = qa_chain.invoke({"question": message, "chat_history": formatted_chat_history})
+    
+    # Ajustar o prompt com instrução de idioma
+    if language == "Português":
+        prompt = f"Responda em português: {message}"
+    else:
+        prompt = f"Answer in English: {message}"
+    
+    # Enviar o prompt ajustado ao qa_chain
+    response = qa_chain.invoke({"question": prompt, "chat_history": formatted_chat_history})
     response_answer = response["answer"]
     if response_answer.find("Helpful Answer:") != -1:
         response_answer = response_answer.split("Helpful Answer:")[-1]
     
-    # Ajustar resposta com base no idioma
-    if language == "Português":
-        response_answer = f"Resposta em português: {response_answer}"
-    else:
-        response_answer = f"Response in English: {response_answer}"
-
     response_sources = response["source_documents"]
     response_source1 = response_sources[0].page_content.strip()
     response_source2 = response_sources[1].page_content.strip()