Update app.py

app.py

@@ -13,7 +13,7 @@ Code:
 """
 
 
-@cl.langchain_factory(use_async=False)
+@cl.on_chat_start
 def factory():
     system_message_prompt = SystemMessagePromptTemplate.from_template(template)
 
@@ -23,9 +23,17 @@ def factory():
     llm_chain = LLMChain(prompt=prompt, llm=llm, verbose=True,)
 
     cl.user_session.set("llm_chain", llm_chain)
-    return llm_chain
 
 
+
+@cl.on_message
+async def main(message):
+    llm_chain = cl.user_session.get("llm_chain")
+
+    res = await llm_chain.acall(message, callbacks=[cl.AsyncLangchainCallbackHandler()])
+
+    await cl.Message(content=res["text"]).send()
+    
 @cl.langchain_rename   # This will be particularly useful when we want to customize this thing for production.
 def rename(orig_author):
     rename_dict = {
@@ -33,7 +41,3 @@ def rename(orig_author):
     }
     return rename_dict.get(orig_author, orig_author)
 
-
-@cl.on_chat_start
-async def main():
-    await cl.Message(content="Welcome to CodeSmith !! Let's start this").send()