import chainlit as cl
from extract_app import extract_information


@cl.on_chat_start
async def start():
    """
    This is called when the Chainlit chat is started!
    """
    await cl.Message("Welcome to the information extraction chat!")

@cl.on_message
def main(message: cl.Message):
    """
    This is called when a message is received!
    """
    res = extract_information().invoke(input=message.content)
    cl.Message(content=res)