No conversations yet
+{% endif %} diff --git a/cheddahbot/templates/partials/loop_status.html b/cheddahbot/templates/partials/loop_status.html new file mode 100644 index 0000000..1914730 --- /dev/null +++ b/cheddahbot/templates/partials/loop_status.html @@ -0,0 +1,6 @@ +{% for name, info in loops.items() %} +| Task | +Customer | +Status | +Due | +
|---|---|---|---|
| + {% if task.url %}{{ task.name }} + {% else %}{{ task.name }}{% endif %} + | +{{ task.custom_fields.get('Client', 'N/A') if task.custom_fields else 'N/A' }} | +{{ task.status }} | +{{ task.due_display or '-' }} | +
No tasks
+{% endif %} diff --git a/cheddahbot/web/__init__.py b/cheddahbot/web/__init__.py new file mode 100644 index 0000000..78e286c --- /dev/null +++ b/cheddahbot/web/__init__.py @@ -0,0 +1,57 @@ +"""HTMX + FastAPI web frontend for CheddahBot.""" + +from __future__ import annotations + +import logging +from pathlib import Path +from typing import TYPE_CHECKING + +from fastapi import FastAPI +from fastapi.templating import Jinja2Templates +from starlette.staticfiles import StaticFiles + +if TYPE_CHECKING: + from ..agent_registry import AgentRegistry + from ..config import Config + from ..db import Database + from ..llm import LLMAdapter + from ..notifications import NotificationBus + from ..scheduler import Scheduler + +log = logging.getLogger(__name__) + +_TEMPLATE_DIR = Path(__file__).resolve().parent.parent / "templates" +_STATIC_DIR = Path(__file__).resolve().parent.parent / "static" + +templates = Jinja2Templates(directory=str(_TEMPLATE_DIR)) + + +def mount_web_app( + app: FastAPI, + registry: AgentRegistry, + config: Config, + llm: LLMAdapter, + notification_bus: NotificationBus | None = None, + scheduler: Scheduler | None = None, + db: Database | None = None, +): + """Mount all web routes and static files onto the FastAPI app.""" + # Wire dependencies into route modules + from . import routes_chat, routes_pages, routes_sse + from .routes_chat import router as chat_router + from .routes_pages import router as pages_router + from .routes_sse import router as sse_router + + routes_pages.setup(registry, config, llm, templates, db=db, scheduler=scheduler) + routes_chat.setup(registry, config, llm, db, templates) + routes_sse.setup(notification_bus, scheduler, db) + + app.include_router(chat_router) + app.include_router(sse_router) + # Pages router last (it has catch-all GET /) + app.include_router(pages_router) + + # Static files + app.mount("/static", StaticFiles(directory=str(_STATIC_DIR)), name="static") + + log.info("Web UI mounted (templates: %s, static: %s)", _TEMPLATE_DIR, _STATIC_DIR) diff --git a/cheddahbot/web/routes_chat.py b/cheddahbot/web/routes_chat.py new file mode 100644 index 0000000..371dd8f --- /dev/null +++ b/cheddahbot/web/routes_chat.py @@ -0,0 +1,270 @@ +"""Chat routes: send messages, stream responses, manage conversations.""" + +from __future__ import annotations + +import asyncio +import logging +import tempfile +import time +from pathlib import Path +from typing import TYPE_CHECKING + +from fastapi import APIRouter, Form, Request, UploadFile +from fastapi.responses import HTMLResponse +from fastapi.templating import Jinja2Templates +from sse_starlette.sse import EventSourceResponse + +if TYPE_CHECKING: + from ..agent_registry import AgentRegistry + from ..config import Config + from ..db import Database + from ..llm import LLMAdapter + +log = logging.getLogger(__name__) + +router = APIRouter(prefix="/chat") + +_registry: AgentRegistry | None = None +_config: Config | None = None +_llm: LLMAdapter | None = None +_db: Database | None = None +_templates: Jinja2Templates | None = None + +# Pending responses: conv_id -> {text, files, timestamp} +_pending: dict[str, dict] = {} + + +def setup(registry, config, llm, db, templates): + global _registry, _config, _llm, _db, _templates + _registry = registry + _config = config + _llm = llm + _db = db + _templates = templates + + +def _get_agent(name: str): + if _registry: + return _registry.get(name) or _registry.default + return None + + +def _cleanup_pending(): + """Remove pending entries older than 60s.""" + now = time.time() + expired = [k for k, v in _pending.items() if now - v["timestamp"] > 60] + for k in expired: + del _pending[k] + + +@router.post("/send") +async def send_message( + request: Request, + text: str = Form(""), + agent_name: str = Form("default"), + conv_id: str = Form(""), + files: list[UploadFile] | None = None, +): + """Accept user message, return user bubble HTML + trigger SSE stream.""" + _cleanup_pending() + + agent = _get_agent(agent_name) + if not agent: + return HTMLResponse("ClickUp not configured
') + + try: + from ..api import get_tasks + data = await get_tasks() + all_tasks = data.get("tasks", []) + except Exception as e: + log.error("Pipeline data fetch failed: %s", e) + return HTMLResponse(f'Error: {e}
') + + # Group by work category, then by status + pipeline_statuses = [ + "to do", "automation underway", "outline review", "internal review", "error", + ] + categories = {} # category -> {status -> [tasks]} + for t in all_tasks: + cat = t.get("task_type") or "Other" + status = t.get("status", "unknown") + + # Only show tasks in pipeline-relevant statuses + if status not in pipeline_statuses: + continue + + if cat not in categories: + categories[cat] = {} + categories[cat].setdefault(status, []).append(t) + + # Build HTML + html_parts = [] + + # Status summary counts + total_counts = {} + for cat_data in categories.values(): + for status, tasks in cat_data.items(): + total_counts[status] = total_counts.get(status, 0) + len(tasks) + + if total_counts: + html_parts.append('| Task | Customer | Status | Due | ' + '
|---|---|---|---|
| {name_cell} | {customer} | ' + f'{status} | ' + f'{due_display} |
No active pipeline tasks
') + + return HTMLResponse('\n'.join(html_parts)) diff --git a/cheddahbot/web/routes_sse.py b/cheddahbot/web/routes_sse.py new file mode 100644 index 0000000..9936149 --- /dev/null +++ b/cheddahbot/web/routes_sse.py @@ -0,0 +1,94 @@ +"""SSE routes for live dashboard updates.""" + +from __future__ import annotations + +import asyncio +import json +import logging +from datetime import datetime +from typing import TYPE_CHECKING + +from fastapi import APIRouter +from sse_starlette.sse import EventSourceResponse + +if TYPE_CHECKING: + from ..db import Database + from ..notifications import NotificationBus + from ..scheduler import Scheduler + +log = logging.getLogger(__name__) + +router = APIRouter(prefix="/sse") + +_notification_bus: NotificationBus | None = None +_scheduler: Scheduler | None = None +_db: Database | None = None + + +def setup(notification_bus, scheduler, db): + global _notification_bus, _scheduler, _db + _notification_bus = notification_bus + _scheduler = scheduler + _db = db + + +@router.get("/notifications") +async def sse_notifications(): + """Stream new notifications as they arrive.""" + listener_id = f"sse-notif-{id(asyncio.current_task())}" + + # Subscribe to notification bus + queue: asyncio.Queue = asyncio.Queue() + loop = asyncio.get_event_loop() + + if _notification_bus: + def on_notify(msg, cat): + loop.call_soon_threadsafe( + queue.put_nowait, {"message": msg, "category": cat} + ) + _notification_bus.subscribe(listener_id, on_notify) + + async def generate(): + try: + while True: + try: + notif = await asyncio.wait_for(queue.get(), timeout=30) + yield { + "event": "notification", + "data": json.dumps(notif), + } + except TimeoutError: + yield {"event": "heartbeat", "data": ""} + finally: + if _notification_bus: + _notification_bus.unsubscribe(listener_id) + + return EventSourceResponse(generate()) + + +@router.get("/loops") +async def sse_loops(): + """Push loop timestamps + active executions every 15s.""" + async def generate(): + while True: + data = {"loops": {}, "executions": {}} + if _scheduler: + ts = _scheduler.get_loop_timestamps() + data["loops"] = ts + # Serialize active executions (datetime -> str) + raw_exec = _scheduler.get_active_executions() + execs = {} + for tid, info in raw_exec.items(): + execs[tid] = { + "name": info.get("name", ""), + "tool": info.get("tool", ""), + "started_at": info["started_at"].isoformat() + if isinstance(info.get("started_at"), datetime) + else str(info.get("started_at", "")), + "thread": info.get("thread", ""), + } + data["executions"] = execs + yield {"event": "loops", "data": json.dumps(data)} + await asyncio.sleep(15) + + return EventSourceResponse(generate()) diff --git a/pyproject.toml b/pyproject.toml index 03597d1..ad3e210 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -16,6 +16,9 @@ dependencies = [ "edge-tts>=6.1", "python-docx>=1.2.0", "openpyxl>=3.1.5", + "jinja2>=3.1.6", + "python-multipart>=0.0.22", + "sse-starlette>=3.3.3", ] [build-system] diff --git a/uv.lock b/uv.lock index 6aa1314..d485da0 100644 --- a/uv.lock +++ b/uv.lock @@ -325,13 +325,16 @@ dependencies = [ { name = "edge-tts" }, { name = "gradio" }, { name = "httpx" }, + { name = "jinja2" }, { name = "numpy" }, { name = "openai" }, { name = "openpyxl" }, { name = "python-docx" }, { name = "python-dotenv" }, + { name = "python-multipart" }, { name = "pyyaml" }, { name = "sentence-transformers" }, + { name = "sse-starlette" }, ] [package.dev-dependencies] @@ -357,13 +360,16 @@ requires-dist = [ { name = "edge-tts", specifier = ">=6.1" }, { name = "gradio", specifier = ">=5.0" }, { name = "httpx", specifier = ">=0.27" }, + { name = "jinja2", specifier = ">=3.1.6" }, { name = "numpy", specifier = ">=1.24" }, { name = "openai", specifier = ">=1.30" }, { name = "openpyxl", specifier = ">=3.1.5" }, { name = "python-docx", specifier = ">=1.2.0" }, { name = "python-dotenv", specifier = ">=1.0" }, + { name = "python-multipart", specifier = ">=0.0.22" }, { name = "pyyaml", specifier = ">=6.0" }, { name = "sentence-transformers", specifier = ">=3.0" }, + { name = "sse-starlette", specifier = ">=3.3.3" }, ] [package.metadata.requires-dev] @@ -2556,6 +2562,19 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/46/2c/1462b1d0a634697ae9e55b3cecdcb64788e8b7d63f54d923fcd0bb140aed/soupsieve-2.8.3-py3-none-any.whl", hash = "sha256:ed64f2ba4eebeab06cc4962affce381647455978ffc1e36bb79a545b91f45a95", size = 37016, upload-time = "2026-01-20T04:27:01.012Z" }, ] +[[package]] +name = "sse-starlette" +version = "3.3.3" +source = { registry = "https://pypi.org/simple" } +dependencies = [ + { name = "anyio" }, + { name = "starlette" }, +] +sdist = { url = "https://files.pythonhosted.org/packages/14/2f/9223c24f568bb7a0c03d751e609844dce0968f13b39a3f73fbb3a96cd27a/sse_starlette-3.3.3.tar.gz", hash = "sha256:72a95d7575fd5129bd0ae15275ac6432bb35ac542fdebb82889c24bb9f3f4049", size = 32420, upload-time = "2026-03-17T20:05:55.529Z" } +wheels = [ + { url = "https://files.pythonhosted.org/packages/78/e2/b8cff57a67dddf9a464d7e943218e031617fb3ddc133aeeb0602ff5f6c85/sse_starlette-3.3.3-py3-none-any.whl", hash = "sha256:c5abb5082a1cc1c6294d89c5290c46b5f67808cfdb612b7ec27e8ba061c22e8d", size = 14329, upload-time = "2026-03-17T20:05:54.35Z" }, +] + [[package]] name = "starlette" version = "0.52.1" @@ -2722,6 +2741,12 @@ wheels = [ { url = "https://files.pythonhosted.org/packages/0f/8b/4b61d6e13f7108f36910df9ab4b58fd389cc2520d54d81b88660804aad99/torch-2.10.0-2-cp311-none-macosx_11_0_arm64.whl", hash = "sha256:418997cb02d0a0f1497cf6a09f63166f9f5df9f3e16c8a716ab76a72127c714f", size = 79423467, upload-time = "2026-02-10T21:44:48.711Z" }, { url = "https://files.pythonhosted.org/packages/d3/54/a2ba279afcca44bbd320d4e73675b282fcee3d81400ea1b53934efca6462/torch-2.10.0-2-cp312-none-macosx_11_0_arm64.whl", hash = "sha256:13ec4add8c3faaed8d13e0574f5cd4a323c11655546f91fbe6afa77b57423574", size = 79498202, upload-time = "2026-02-10T21:44:52.603Z" }, { url = "https://files.pythonhosted.org/packages/ec/23/2c9fe0c9c27f7f6cb865abcea8a4568f29f00acaeadfc6a37f6801f84cb4/torch-2.10.0-2-cp313-none-macosx_11_0_arm64.whl", hash = "sha256:e521c9f030a3774ed770a9c011751fb47c4d12029a3d6522116e48431f2ff89e", size = 79498254, upload-time = "2026-02-10T21:44:44.095Z" }, + { url = "https://files.pythonhosted.org/packages/36/ab/7b562f1808d3f65414cd80a4f7d4bb00979d9355616c034c171249e1a303/torch-2.10.0-3-cp311-cp311-manylinux_2_28_x86_64.whl", hash = "sha256:ac5bdcbb074384c66fa160c15b1ead77839e3fe7ed117d667249afce0acabfac", size = 915518691, upload-time = "2026-03-11T14:15:43.147Z" }, + { url = "https://files.pythonhosted.org/packages/b3/7a/abada41517ce0011775f0f4eacc79659bc9bc6c361e6bfe6f7052a6b9363/torch-2.10.0-3-cp312-cp312-manylinux_2_28_x86_64.whl", hash = "sha256:98c01b8bb5e3240426dcde1446eed6f40c778091c8544767ef1168fc663a05a6", size = 915622781, upload-time = "2026-03-11T14:17:11.354Z" }, + { url = "https://files.pythonhosted.org/packages/ab/c6/4dfe238342ffdcec5aef1c96c457548762d33c40b45a1ab7033bb26d2ff2/torch-2.10.0-3-cp313-cp313-manylinux_2_28_x86_64.whl", hash = "sha256:80b1b5bfe38eb0e9f5ff09f206dcac0a87aadd084230d4a36eea5ec5232c115b", size = 915627275, upload-time = "2026-03-11T14:16:11.325Z" }, + { url = "https://files.pythonhosted.org/packages/d8/f0/72bf18847f58f877a6a8acf60614b14935e2f156d942483af1ffc081aea0/torch-2.10.0-3-cp313-cp313t-manylinux_2_28_x86_64.whl", hash = "sha256:46b3574d93a2a8134b3f5475cfb98e2eb46771794c57015f6ad1fb795ec25e49", size = 915523474, upload-time = "2026-03-11T14:17:44.422Z" }, + { url = "https://files.pythonhosted.org/packages/f4/39/590742415c3030551944edc2ddc273ea1fdfe8ffb2780992e824f1ebee98/torch-2.10.0-3-cp314-cp314-manylinux_2_28_x86_64.whl", hash = "sha256:b1d5e2aba4eb7f8e87fbe04f86442887f9167a35f092afe4c237dfcaaef6e328", size = 915632474, upload-time = "2026-03-11T14:15:13.666Z" }, + { url = "https://files.pythonhosted.org/packages/b6/8e/34949484f764dde5b222b7fe3fede43e4a6f0da9d7f8c370bb617d629ee2/torch-2.10.0-3-cp314-cp314t-manylinux_2_28_x86_64.whl", hash = "sha256:0228d20b06701c05a8f978357f657817a4a63984b0c90745def81c18aedfa591", size = 915523882, upload-time = "2026-03-11T14:14:46.311Z" }, { url = "https://files.pythonhosted.org/packages/78/89/f5554b13ebd71e05c0b002f95148033e730d3f7067f67423026cc9c69410/torch-2.10.0-cp311-cp311-manylinux_2_28_aarch64.whl", hash = "sha256:3282d9febd1e4e476630a099692b44fdc214ee9bf8ee5377732d9d9dfe5712e4", size = 145992610, upload-time = "2026-01-21T16:25:26.327Z" }, { url = "https://files.pythonhosted.org/packages/ae/30/a3a2120621bf9c17779b169fc17e3dc29b230c29d0f8222f499f5e159aa8/torch-2.10.0-cp311-cp311-manylinux_2_28_x86_64.whl", hash = "sha256:a2f9edd8dbc99f62bc4dfb78af7bf89499bca3d753423ac1b4e06592e467b763", size = 915607863, upload-time = "2026-01-21T16:25:06.696Z" }, { url = "https://files.pythonhosted.org/packages/6f/3d/c87b33c5f260a2a8ad68da7147e105f05868c281c63d65ed85aa4da98c66/torch-2.10.0-cp311-cp311-win_amd64.whl", hash = "sha256:29b7009dba4b7a1c960260fc8ac85022c784250af43af9fb0ebafc9883782ebd", size = 113723116, upload-time = "2026-01-21T16:25:21.916Z" },