Update main.py

main.py

@@ -10,7 +10,7 @@ from pathlib import Path
 import chainlit as cl
 #from mistralai.client import MistralClient
 #from mistralai import Mistral, UserMessage, SystemMessage
-
+from deep_translator import GoogleTranslator
 from operator import itemgetter
 from langchain_huggingface import HuggingFaceEndpoint
 from langchain.schema.runnable import Runnable, RunnablePassthrough, RunnableLambda
@@ -184,15 +184,23 @@ def Generation_completion(client, data, question):
 @traceable(run_type="chain", name="Mistral Assistant des datas Gustaviz",)
 async def Affichage_reponse(response, question, data):
     memory = cl.user_session.get("memory")
-    msg = cl.Message(author="COPILOT",content="")
+    result = cl.Message(author="COPILOT",content="")
     
-    async for chunk in response.astream({"question": question, "data": data},config=RunnableConfig(callbacks=[cl.AsyncLangchainCallbackHandler(stream_final_answer=True)])):
+    #async for chunk in response.astream({"question": question, "data": data},config=RunnableConfig(callbacks=[cl.AsyncLangchainCallbackHandler(stream_final_answer=True)])):
+    #    time.sleep(0.125)
+    #    await msg.stream_token(chunk.replace('Ã','é').replace('©','').replace('Ã','è').replace('¨','').replace('â','\'').replace('€','').replace('™','').replace('Å','oe').replace('“','').replace('®','î').replace('´','ô').replace('<','').replace('>','').replace('/',''))
+
+    msg = response.ainvoke({"question": question, "data": data})
+    msg = msg.replace('Ã','é').replace('©','').replace('Ã','è').replace('¨','').replace('â','\'').replace('€','').replace('™','').replace('Å','oe').replace('“','').replace('®','î').replace('´','ô').replace('<','').replace('>','').replace('/','')
+    msg = GoogleTranslator(source='auto', target='fr').translate(msg[0:4999])
+    msgList = msg.split(' ')
+    for chunk in msgList:
         time.sleep(0.125)
-        await msg.stream_token(chunk.replace('Ã','é').replace('©','').replace('Ã','è').replace('¨','').replace('â','\'').replace('€','').replace('™','').replace('Å','oe').replace('“','').replace('®','î').replace('´','ô').replace('<','').replace('>','').replace('/',''))
-    
-    await msg.send()
+        await result.stream_token(chunk)
+        
+    await result.send()
     memory.chat_memory.add_user_message(question)
-    memory.chat_memory.add_ai_message(msg.content)
+    memory.chat_memory.add_ai_message(msg)
     
     return msg