updated session vars

.gitignore

@@ -205,3 +205,7 @@ cython_debug/
 marimo/_static/
 marimo/_lsp/
 __marimo__/
+
+# Chainlit
+
+.chainlit/

app.py

@@ -1,5 +1,4 @@
 import os
-import contextvars
 import chainlit as cl
 from dotenv import load_dotenv
 from langchain_anthropic import ChatAnthropic
@@ -56,28 +55,40 @@ def get_chain():
 @cl.on_chat_start
 async def on_chat_start():
     """Initialize the chain when a chat session starts."""
+    # Store the chain in user session first (before any messages)
+    chain = get_chain()
+    if chain:
+        cl.user_session.set("chain", chain)
+
     # Check if API key is available
     if not ANTHROPIC_API_KEY:
-        await cl.Message(
-            content="❌ Error: ANTHROPIC_API_KEY not configured. Please set it in Space settings → Variables and secrets."
-        ).send()
+        try:
+            await cl.Message(
+                content="❌ Error: ANTHROPIC_API_KEY not configured. Please set it in Space settings → Variables and secrets."
+            ).send()
+        except LookupError:
+            # ContextVar error - chain will be initialized on first message instead
+            pass
         return
 
-    # Create the chain
-    chain = get_chain()
+    # Create the chain if not already created
     if chain is None:
-        await cl.Message(
-            content="❌ Error: Could not initialize chain. Please check API key configuration."
-        ).send()
+        try:
+            await cl.Message(
+                content="❌ Error: Could not initialize chain. Please check API key configuration."
+            ).send()
+        except LookupError:
+            # ContextVar error - chain will be initialized on first message instead
+            pass
         return
 
-    # Store the chain in user session
-    cl.user_session.set("chain", chain)
-
     # Send welcome message - according to prompt section 1.6, ask ONE question: "What are you working on?"
-    await cl.Message(
-        content="What are you working on?"
-    ).send()
+    try:
+        await cl.Message(content="What are you working on?").send()
+    except LookupError:
+        # ContextVar error - message will be sent on first user message instead
+        # The chain is already stored in user_session, so on_message will work
+        pass
 
 
 @cl.on_message
@@ -86,9 +97,14 @@ async def on_message(message: cl.Message):
     # Retrieve the chain from user session
     chain = cl.user_session.get("chain")
 
+    # If chain is None, try to initialize it (in case on_chat_start had ContextVar issues)
     if chain is None:
-        await cl.Message(content="Error: Chain not initialized. Please refresh the page.").send()
-        return
+        chain = get_chain()
+        if chain:
+            cl.user_session.set("chain", chain)
+        else:
+            await cl.Message(content="Error: Chain not initialized. Please refresh the page.").send()
+            return
 
     # Create a message object for streaming
     msg = cl.Message(content="")

chainlit.md

@@ -0,0 +1,14 @@
+# Welcome to Chainlit! 🚀🤖
+
+Hi there, Developer! 👋 We're excited to have you on board. Chainlit is a powerful tool designed to help you prototype, debug and share applications built on top of LLMs.
+
+## Useful Links 🔗
+
+- **Documentation:** Get started with our comprehensive [Chainlit Documentation](https://docs.chainlit.io) 📚
+- **Discord Community:** Join our friendly [Chainlit Discord](https://discord.gg/k73SQ3FyUh) to ask questions, share your projects, and connect with other developers! 💬
+
+We can't wait to see what you create with Chainlit! Happy coding! 💻😊
+
+## Welcome screen
+
+To modify the welcome screen, edit the `chainlit.md` file at the root of your project. If you do not want a welcome screen, just leave this file empty.