Spaces:
Running
Running
| import chainlit as cl | |
| from typing import Optional | |
| import time | |
| # Store conversation history | |
| conversation_memory = [] | |
| async def start(): | |
| """Initializes the chat session""" | |
| # Send an initial message | |
| await cl.Message( | |
| content="👋 Hello! I'm your AI assistant. How can I help you today?", | |
| author="Assistant" | |
| ).send() | |
| # Set some session variables | |
| cl.user_session.set("conversation_started", True) | |
| async def main(message: cl.Message): | |
| """Main message handler""" | |
| # Simulate some processing time | |
| with cl.Step("Processing...") as step: | |
| time.sleep(1) # Simulated delay | |
| step.output = "Processed message" | |
| # Store message in conversation history | |
| conversation_memory.append({ | |
| "role": "user", | |
| "content": message.content | |
| }) | |
| # Create a response | |
| response = f"I received your message: '{message.content}'. This is a demo response." | |
| # Store response in conversation history | |
| conversation_memory.append({ | |
| "role": "assistant", | |
| "content": response | |
| }) | |
| # Send response with typing effect | |
| await cl.Message( | |
| content=response, | |
| author="Assistant" | |
| ).send() | |
| def auth_callback(username: str, password: str) -> Optional[cl.User]: | |
| """Basic authentication handler""" | |
| # This is a simple example - in production, use proper authentication | |
| if username == "demo" and password == "password": | |
| return cl.User(identifier="demo", metadata={"role": "user"}) | |
| return None | |
| async def end(): | |
| """Cleanup when chat ends""" | |
| await cl.Message(content="👋 Thank you for chatting! Goodbye!").send() | |
| # Custom action handler example | |
| async def on_action(action): | |
| """Handles custom feedback action""" | |
| await cl.Message(content=f"Received feedback: {action.value}").send() | |