Update app.py

app.py

@@ -41,23 +41,23 @@ def initialize_chainlit():
         )
     )
 
-    # Setup chainlit callbacks
-    @cl.on_chat_start
-    async def on_chat_start():
-        prompt = ChatPromptTemplate.from_messages([("human", "{question}")])
-        runnable = prompt | llm | StrOutputParser()
-        cl.user_session.set("runnable", runnable)
+# Setup chainlit callbacks
+@cl.on_chat_start
+async def on_chat_start():
+    prompt = ChatPromptTemplate.from_messages([("human", "{question}")])
+    runnable = prompt | llm | StrOutputParser()
+    cl.user_session.set("runnable", runnable)
 
-    @cl.on_message
-    async def on_message(message: cl.Message):
-        runnable = cl.user_session.get("runnable")  # type: Runnable
-        msg = cl.Message(content="")
-        async for chunk in runnable.astream(
-            {"question": message.content},
-            config=RunnableConfig(callbacks=[cl.LangchainCallbackHandler()]),
-        ):
-            await msg.stream_token(chunk)
-        await msg.send()
+@cl.on_message
+async def on_message(message: cl.Message):
+    runnable = cl.user_session.get("runnable")  # type: Runnable
+    msg = cl.Message(content="")
+    async for chunk in runnable.astream(
+        {"question": message.content},
+        config=RunnableConfig(callbacks=[cl.LangchainCallbackHandler()]),
+    ):
+        await msg.stream_token(chunk)
+    await msg.send()
 
 @app.route('/')
 def home():