Update main.py

main.py

@@ -82,7 +82,7 @@ async def Affichage_reponse(response):
     return msg
 
 @cl.step(type="tool", show_input=True)
-async def Affichage_question_contexte(answer):
+async def Affichage_question_contexte(answer, question):
     try:
         indexDeb = answer.index("Question en relation avec le sujet")
         indexDebLength = len("Question en relation avec le sujet")
@@ -98,14 +98,12 @@ async def Affichage_question_contexte(answer):
             cl.Action(name="Activer", value=extraction + "?", description="Activer la question en relation avec le contexte.")
         ]
     await cl.Message(author="COPILOT", content="🌐 Question en relation avec le contexte : " + extraction + "?", actions=actions).send()
-
-@cl.step(type="tool", show_input=True)
-async def Affichage_recherche_web(answer):
-    actions = [
-            cl.Action(name="Site web", value=answer, description="Rechercher sur le site web de l'université.")
+    
+    actionsWeb = [
+            cl.Action(name="Site web", value=question, description="Rechercher sur le site web de l'université.")
         ]
-    await cl.Message(author="COPILOT", content="🌐 Rechercher sur le site web de l'université : " + answer, actions=actions).send()
-        
+    await cl.Message(author="COPILOT", content="🌐 Rechercher sur le site web de l'université : " + question, actions=actionsWeb).send()
+     
 @cl.action_callback("Activer")
 async def on_action(action):
     client = cl.user_session.get("client")
@@ -118,7 +116,7 @@ async def on_action(action):
     
     answer = msg.content
     
-    await Affichage_question_contexte(answer)
+    await Affichage_question_contexte(answer, question)
     
 @cl.action_callback("Site web")
 async def on_action(action):
@@ -133,7 +131,7 @@ async def on_action(action):
     
     answer = msg.content
     
-    await Affichage_question_contexte(answer)
+    await Affichage_question_contexte(answer, question)
     
     
 @cl.set_chat_profiles
@@ -198,8 +196,7 @@ async def main(message: cl.Message):
     msg = await Affichage_reponse(response)
     
     answer = msg.content
-    await Affichage_question_contexte(answer)
-    await Affichage_recherche_web(message.content)
+    await Affichage_question_contexte(answer, message.content)
 
 @cl.on_stop
 def on_stop():