third commit

Dockerfile

@@ -8,4 +8,4 @@ COPY --chown=user . $HOME/app
 COPY ./requirements.txt ~/app/requirements.txt
 RUN pip install -r requirements.txt
 COPY . .
-CMD ["chainlit", "run", "talking_app.py", "--port", "7860"]
+CMD ["chainlit", "run", "app.py", "--port", "7860"]

app.py

@@ -1,8 +1,9 @@
-import chainlit as cl
 import sys
 
 sys.path.append(".")
 
+import chainlit as cl
+
 from earnings_app import extract_information
 
 @cl.author_rename
@@ -13,15 +14,21 @@ def rename(orig_author: str):
     return rename_dict.get(orig_author, orig_author)
 
 @cl.on_chat_start
-async def init():
+async def start():
+    """
+    This is called when the Chainlit chat is started!
+    """
     cl.user_session.set("chain", extract_information())
-    await cl.Message(content="Welcome to Earnings chat!").send()
+    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!
+    """
     chain = cl.user_session.get("chain")
-    res = chain.chat(message)
+    res = await chain.achat(message)
     # res = await chain.aiinvoke({"input": message})
     # res = res["text"]
-    out = ''.join(str(res))
+    out = str(res)
     await cl.Message(content=out).send()

earnings_app.py

@@ -171,16 +171,16 @@ def extract_information():
     return agent
 
 
-if __name__ == "__main__":
-    text = "Who is the CEO of MSFT."
-    chain = extract_information()
-    print(str(chain.chat(text)))
+# if __name__ == "__main__":
+#     text = "Who is the CEO of MSFT."
+#     chain = extract_information()
+#     print(str(chain.chat(text)))
 
-    async def extract_information_async(message: str):
-        return str(chain.chat(text))
+#     async def extract_information_async(message: str):
+#         return str(chain.chat(text))
 
-    async def main():
-        res = await extract_information_async(text)
-        print(res)
+#     async def main():
+#         res = await extract_information_async(text)
+#         print(res)
 
     asyncio.run(main())