Update app.py

app.py

@@ -3,22 +3,22 @@ from extract_app import extract_information
 
 
 @cl.on_chat_start
-def start():
+async def start():
     """
     This is called when the Chainlit chat is started!
     """
-    cl.Message("Welcome to the information extraction chat!").send()
+    cl.user_session.set("chain", extract_information())
+    await cl.Message("Welcome to the information extraction chat!").send()
 
 @cl.on_message
 async def main(message: cl.Message):
     """
     This is called when a message is received!
     """
-    # res = await extract_information().ainvoke({"input": message})["text"]
-    res = await extract_information().ainvoke({"input": message})
+    chain = cl.user_session.get("chain")
+    res = await chain.ainvoke({"input": message})
     res = res["text"]
-    
-    # res = await cl.make_async(extract_information().invoke)(
-    #     input=message.content, callbacks=[cl.LangchainCallbackHandler()]
-    # )
-    await cl.Message(content=res).send()
+    out = "Name \tAge\n" + ''.join(
+        [f"{person.name}\t{person.age}\n" for person in res]
+    )
+    await cl.Message(content=out).send()