ugh

app.py

@@ -84,7 +84,7 @@ User Query:
 {query}
 
 Context:
-{context}<|eot_id|>
+#{context}<|eot_id|>
 
 <|start_header_id|>assistant<|end_header_id|>
 """
@@ -133,7 +133,13 @@ async def start_chat():
     )"""
     lcel_rag_chain = {"query": itemgetter("query")} | rag_prompt | hf_llm
 
-    cl.user_session.set("lcel_rag_chain", lcel_rag_chain)
+    try:
+        # Attempt to set up session normally
+        cl.user_session.set("lcel_rag_chain", lcel_rag_chain)
+    except KeyError:
+        print("Reinitializing session due to disconnection.")
+        cl.user_session.clear()
+        cl.user_session.set("lcel_rag_chain", lcel_rag_chain)
 
 @cl.on_message  
 async def main(message: cl.Message):
@@ -145,6 +151,9 @@ async def main(message: cl.Message):
     The LCEL RAG chain is stored in the user session, and is unique to each user session - this is why we can access it here.
     """
     print("entering on_message")
+    msg = cl.Message(content="Processing your request... this may take a moment.")
+    await msg.send()
+
     lcel_rag_chain = cl.user_session.get("lcel_rag_chain")
 
     msg = cl.Message(content="")