updated app.py to limit to 3 tries

app.py

@@ -36,6 +36,18 @@ async def start_chat():
 @cl.on_message  # marks a function that should be run each time the chatbot receives a message from a user
 async def main(message: cl.Message):
     settings = cl.user_session.get("settings")
+   
+    # Increment message count
+    count = cl.user_session.get("message_count") or 0
+    count += 1
+    cl.user_session.set("message_count", count)
+
+    # Stop after 3 messages
+    if count > 3:
+        await cl.Message(
+            content="⚠️ This demo is limited to 3 messages per session."
+        ).send()
+        return
 
     client = AsyncOpenAI()