Update main.py

main.py

@@ -155,7 +155,32 @@ async def set_starters():
 async def on_message(message: cl.Message):
     await cl.Message(f"> SURVEYIA").send()
     agent = create_agent("./public/ExpeCFA_LP_CAA_7-5-2024.csv")
-    result = agent.run(message.content)
+    cb = cl.AsyncLangchainCallbackHandler()
+    try:
+        res = await agent.acall("Réponds en langue française à la question suivante :\n" + message.content + "\nDétaille la réponse en faisant une analyse complète en 2000 mots minimum.", callbacks=[cb])
+        await cl.Message(author="COPILOT",content=GoogleTranslator(source='auto', target='fr').translate(res['output'])).send()
+        listPrompts_name = f"Liste des requêtes"
+        prompt_elements = []
+        prompt_elements.append(
+            cl.Text(content=library(), name=listPrompts_name)
+        )
+        await cl.Message(content="📚 Bibliothèque de questions : " + listPrompts_name, elements=prompt_elements).send()
+    except ValueError as e:
+        res = str(e)
+        resArray = res.split(":")
+        ans = ''
+        if str(res).find('parsing') != -1:
+            for i in range(2,len(resArray)):
+                ans += resArray[i]
+            await cl.Message(author="COPILOT",content=ans.replace("`","")).send()
+            listPrompts_name = f"Liste des requêtes"
+            prompt_elements = []
+            prompt_elements.append(
+                cl.Text(content=library(), name=listPrompts_name)
+            )
+            await cl.Message(content="📚 Bibliothèque de questions : " + listPrompts_name, elements=prompt_elements).send()
+        else:
+            await cl.Message(author="COPILOT",content="Reformulez votre requête, s'il vous plait 😃").send()
     # Query the agent.
     #response = query_agent(agent=agent, query=message.content)
     # Decode the response.