Update main.py

main.py

@@ -297,7 +297,6 @@ async def setup_agent(settings):
             author="🌐🌐🌐",content=f"⚠️ Pas de contexte : {settings['rome']}\n👍 Type de fiche : {settings['type']}\n⛔ Vous ne pouvez pas élaborer de fiche!"
         ).send()
     else:
-        print(settings)
         await cl.Message(
                 author="🌐🌐🌐",content=f"👍 Changement de contexte : {settings['rome']}\n👍 Type de fiche : {settings['type']}"
             ).send()
@@ -367,9 +366,11 @@ async def setup_agent(settings):
                 author="🌐🌐🌐",content="Note de composante sectorielle de(s) code(s) ROME : " + romeListArray[0] + "\n\n" + completion_NCS.completion
             ).send()
             cl.user_session.set("NCS" + romeListArray[0], completion_NCS.completion)
-            #memory = ConversationBufferMemory()
-            #memory.chat_memory.add_user_message(romeListArray[0])
-            #memory.chat_memory.add_ai_message(completion_NCS.completion)
+            listEmplois_name = f"Liste des emplois"
+            text_elements.append(
+                cl.Text(content="Question : " + romeListArray[0] + "\n\nRéponse :\n" + context, name=listEmplois_name)
+            )
+            await cl.Message(author="🌐🌐🌐",content="Source Pôle Emploi : " + listEmplois_name, elements=text_elements).send()
         elif settings['type'] == 'Fiche synoptique - FCS':
             if cl.user_session.get("NCS" + romeListArray[0]):
                 FCS = cl.TaskList()
@@ -442,8 +443,6 @@ async def setup_agent(settings):
                 ).send()
                 await cl.sleep(2)
                 await FCS.remove()
-                                
-                #print(cl.user_session.get("FCS" + romeListArray[0]))
             else:
                 await cl.Message(
                     author="🌐🌐🌐",content="⛔ Vous n'avez pas encore créé de Note sectorielle!"