Update main.py

main.py

@@ -106,13 +106,15 @@ async def on_chat_start():
     res = await cl.AskActionMessage(
         content="<div style='width:100%;text-align:center'> </div>",
         actions=[
-            cl.Action(name="continue", value="continue", label="🎓 Pédagogie durable : «quels sont les modèles d'apprentissage dans les universités?»"),
-            cl.Action(name="cancel", value="cancel", label="🎓 Lieux d'apprentissage : «donne des exemples de lieu d'apprentissage dans les universités?»"),
+            cl.Action(name="article", value="Pédagogie durable", label="🎓 Pédagogie durable : exemple : «quels sont les modèles d'apprentissage dans les universités?»"),
+            cl.Action(name="article", value="Lieux d'apprentissage", label="🎓 Lieux d'apprentissage : exemple : «donne des exemples de lieu d'apprentissage dans les universités?»"),
+            cl.Action(name="jdlp", value="Journée de La Pédagogie", label="🎓 Journée de La Pédagogie : exemple : «Quelles sont les bonnes pratiques des plateformes de e-learning?»"),
         ],
         timeout="3600"
     ).send()
 
-    if res:
+    if res and res.get("value") == "continue":
+        await cl.Message(f"Vous pouvez requêter sur la thématique : {res.get('value')} ({res.get('name')})").send()
         cl.user_session.set("selectRequest", res.get("value"))
         
     template = """Answer the question based only on the following context:
@@ -168,23 +170,13 @@ async def on_message(message: cl.Message):
                 )
 
     async with cl.Step(type="run", name="QA Assistant"):
-        if cl.user_session.get("selectRequest"):
-            async for chunk in runnable.astream(
-                cl.user_session.get("selectRequest"),
-                config=RunnableConfig(callbacks=[
-                    cl.LangchainCallbackHandler(),
-                    PostMessageHandler(msg)
-                ]),
-            ):
-                await msg.stream_token(chunk)
-        else:
-            async for chunk in runnable.astream(
-                message.content,
-                config=RunnableConfig(callbacks=[
-                    cl.LangchainCallbackHandler(),
-                    PostMessageHandler(msg)
-                ]),
-            ):
-                await msg.stream_token(chunk)
+        async for chunk in runnable.astream(
+            cl.user_session.get("selectRequest"),
+            config=RunnableConfig(callbacks=[
+                cl.LangchainCallbackHandler(),
+                PostMessageHandler(msg)
+            ]),
+        ):
+            await msg.stream_token(chunk)
 
     await msg.send()