PraisonAIUI/examples/python/21-email-channel/app.py

"""Example 20: Email Channel — Integrate email into the AIUI dashboard.

Unique Concept: Email as a dashboard channel — managed via the Channels API.

This example shows how email integrates with PraisonAIUI as a channel,
appearing alongside Telegram/Discord/Slack in the dashboard. The email
channel is managed through the same /api/channels endpoints.

Features:
    • Register email as a channel via /api/channels API
    • Dashboard visibility alongside other messaging platforms
    • Gateway-managed lifecycle (start/stop/restart)
    • Approval integration for draft-before-send workflow
    • Full feature dashboard with email channel status

Architecture:
    ┌──────────────────────────────────────────────────────────┐
    │               AIUIGateway (port 8084)                    │
    │                                                          │
    │  ┌────────────┐  ┌──────────────────┐  ┌─────────────┐  │
    │  │  Dashboard  │  │  Channel APIs    │  │  Email Bot   │  │
    │  │  + Email    │  │  /api/channels   │  │  IMAP/SMTP   │  │
    │  │  Status     │  │  CRUD + Toggle   │  │  polling     │  │
    │  └────────────┘  └──────────────────┘  └─────────────┘  │
    └──────────────────────────────────────────────────────────┘

Requires:
    pip install praisonaiui praisonai[email] praisonaiagents
    export OPENAI_API_KEY=sk-...
    export [email protected]
    export EMAIL_APP_PASSWORD=your_app_password

Run:
    aiui run app.py
    # Dashboard at http://localhost:8084
"""

import asyncio
import os
import sys

# ── Imports ─────────────────────────────────────────────────────────

sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "..", "..", "src"))
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))

from _shared.console import icon, safe_print
from praisonaiui.server import create_app
import uvicorn

try:
    from praisonaiui.features.channels import ChannelsFeature
    from praisonaiui.features import get_feature
    CHANNELS_OK = True
except ImportError:
    CHANNELS_OK = False


# ── Seed email channel via API ──────────────────────────────────────

def seed_email_channel():
    """Register an email channel in the dashboard.
    
    This demonstrates how to programmatically add an email channel
    to the AIUI dashboard, which is the same as using the dashboard UI
    or calling POST /api/channels directly.
    """
    from starlette.testclient import TestClient

    app = create_app()
    client = TestClient(app)

    # Check if email channel already exists
    r = client.get("/api/channels")
    channels = r.json().get("channels", [])
    email_exists = any(
        ch.get("platform") == "email"
        for ch in channels
    )

    if not email_exists:
        # Register email as a channel
        r = client.post("/api/channels", json={
            "name": "Support Email",
            "platform": "email",
            "config": {
                "email_address": os.getenv("EMAIL_ADDRESS", "[email protected]"),
                "imap_server": os.getenv("EMAIL_IMAP_SERVER", "imap.gmail.com"),
                "smtp_server": os.getenv("EMAIL_SMTP_SERVER", "smtp.gmail.com"),
                "app_password_ref": "env:EMAIL_APP_PASSWORD",
                "polling_interval": 30,
            },
        })
        if r.status_code in (200, 201):
            channel_id = r.json().get("id", "unknown")
            safe_print(f"   {icon('✅', '[OK]')} Email channel registered: {channel_id}")
        else:
            safe_print(f"   {icon('⚠️', '[WARN]')}  Channel registration: {r.status_code} - {r.json()}")
    else:
        safe_print(f"   {icon('✓', '[OK]')} Email channel already exists")

    # Also seed other channels for comparison in dashboard
    for name, platform in [("Discord #general", "discord"), ("Telegram Support", "telegram")]:
        existing = any(ch.get("platform") == platform for ch in channels)
        if not existing:
            client.post("/api/channels", json={
                "name": name,
                "platform": platform,
                "config": {},
            })
            safe_print(f"   {icon('✓', '[OK]')} Seeded {platform} channel")


# ── Verify email platform support ──────────────────────────────────

def verify_email_support():
    """Check if 'email' is in the supported platforms list."""
    from starlette.testclient import TestClient

    app = create_app()
    client = TestClient(app)

    r = client.get("/api/channels/platforms")
    if r.status_code == 200:
        platforms = r.json().get("platforms", [])
        email_supported = "email" in platforms
        safe_print(f"   Supported platforms: {platforms}")
        safe_print(f"   Email supported: {icon('✅', '[OK]') + ' Yes' if email_supported else icon('❌', '[X]') + ' No - add email to SUPPORTED_PLATFORMS'}")
        return email_supported
    return False


# ── Test channel lifecycle ──────────────────────────────────────────

def test_channel_api():
    """Test the full channel lifecycle for email."""
    from starlette.testclient import TestClient

    app = create_app()
    client = TestClient(app)

    safe_print(f"\n{icon('📋', '[TEST]')} Channel API Test")
    safe_print("=" * 40)

    # List channels
    r = client.get("/api/channels")
    channels = r.json().get("channels", [])
    safe_print(f"   GET /api/channels: {len(channels)} channels")
    for ch in channels:
        ch_icon = icon("📧", "[email]") if ch.get("platform") == "email" else icon("💬", "[chat]")
        safe_print(f"      {ch_icon} {ch.get('name', '?')} ({ch.get('platform', '?')})")

    # Get email channel status
    email_ch = next((ch for ch in channels if ch.get("platform") == "email"), None)
    if email_ch:
        ch_id = email_ch.get("id")
        r = client.get(f"/api/channels/{ch_id}/status")
        safe_print(f"   GET /api/channels/{ch_id}/status: {r.json()}")

    safe_print()


# ── Main ────────────────────────────────────────────────────────────

if __name__ == "__main__":
    safe_print(f"{icon('📧', '[EMAIL]')} PraisonAIUI - Email Channel Integration")
    safe_print("=" * 50)
    safe_print()

    if "--test" in sys.argv:
        # Test mode: verify APIs + seed data
        safe_print(f"{icon('🔍', '[VERIFY]')} Verifying email platform support...")
        verify_email_support()
        safe_print()
        safe_print(f"{icon('🌱', '[SEED]')} Seeding email channel...")
        seed_email_channel()
        test_channel_api()
    else:
        # Run the dashboard
        safe_print(f"{icon('🌱', '[SEED]')} Seeding channels...")
        seed_email_channel()
        safe_print()

        app = create_app()
        safe_print(f"{icon('✅', '[OK]')} Dashboard at http://localhost:8084")
        safe_print("   Channels: http://localhost:8084/api/channels")
        safe_print("   Platforms: http://localhost:8084/api/channels/platforms")
        host = os.getenv("HOST", "127.0.0.1")
        uvicorn.run(app, host=host, port=8084, log_level="info")