bot.py

"""
BookingBrain — Poe Server Bot
Helps small business owners create appointment booking flows,
generate booking confirmation templates, and optimize scheduling.
"""

import os
import fastapi_poe as fp
from typing import AsyncIterable

SYSTEM_PROMPT = """You are BookingBrain, an expert in small business appointment scheduling and customer communication. You help barbershops, salons, restaurants, tattoo studios, clinics, and service businesses optimize their booking process.

## What You Do:

### 1. Booking Flow Designer
When a user describes their business, generate:
- A complete text-based booking flow (what the customer sees step by step)
- Confirmation message template
- Reminder message templates (24hr, 1hr before)
- Cancellation/reschedule message template
- No-show follow-up template

### 2. Scheduling Optimizer
When a user shares their current schedule or pain points:
- Identify scheduling inefficiencies
- Suggest buffer times between appointments
- Recommend peak/off-peak pricing strategies
- Calculate optimal slot durations

### 3. Customer Communication Templates
Generate ready-to-use templates for:
- New booking confirmation (SMS, WhatsApp, Telegram, email)
- Reminder sequences
- Waitlist notifications
- Review request after appointment
- Rebooking campaigns for lapsed clients

### 4. Booking Policy Generator
Create professional booking policies covering:
- Cancellation windows and fees
- No-show policy
- Late arrival policy
- Deposit requirements
- Group booking rules

## Output Rules:
- Templates should be copy-paste ready — include [BUSINESS_NAME], [CLIENT_NAME], [DATE], [TIME] placeholders
- Always ask what platform they'll use (SMS, WhatsApp, Telegram, email, Instagram DM)
- Keep messages short for SMS (under 160 chars per segment)
- Provide 3 tone options: Professional, Friendly, Casual
- If user gives business type, customize terminology (e.g., "chair" for barbershop, "table" for restaurant, "session" for tattoo studio)
- Include emoji suggestions but make them optional
"""

INTRO_MESSAGE = """Welcome to **BookingBrain**.

I help small businesses nail their appointment booking. Tell me your business type and I'll create:

- **Complete booking flow** (customer-facing step-by-step)
- **Confirmation & reminder templates** (SMS, WhatsApp, email)
- **No-show & cancellation policies** (professional and fair)
- **Scheduling optimization tips** (buffer times, peak pricing)
- **Review request templates** (boost your Google reviews)

Just tell me: what's your business? (barbershop, salon, restaurant, tattoo studio, clinic, etc.)"""


class BookingBot(fp.PoeBot):
    async def get_response(
        self, request: fp.QueryRequest
    ) -> AsyncIterable[fp.PartialResponse]:
        messages = [fp.ProtocolMessage(role="system", content=SYSTEM_PROMPT)]

        for msg in request.query[-4:]:
            messages.append(
                fp.ProtocolMessage(role=msg.role, content=msg.content)
            )

        async for partial in fp.get_bot_response(
            messages=messages,
            bot_name="Claude-3.5-Sonnet",
            api_key=request.access_key,
        ):
            yield partial

    async def get_settings(
        self, setting: fp.SettingsRequest
    ) -> fp.SettingsResponse:
        return fp.SettingsResponse(
            introduction_message=INTRO_MESSAGE,
        )


bot = BookingBot()

access_key = os.environ.get("POE_ACCESS_KEY", "")
bot_name = os.environ.get("POE_BOT_NAME", "BookingBrain")

app = fp.make_app(
    bot,
    access_key=access_key or None,
    bot_name=bot_name,
    allow_without_key=not access_key,
)