pylogue 0.2.1__py3-none-any.whl → 0.3.29__py3-none-any.whl

pylogue/shell.py

@@ -0,0 +1,342 @@
+"""
+Chat app wrapper with multiple local histories.
+Run: python -m scripts.examples.chat_app_with_histories.main
+"""
+
+from dataclasses import dataclass
+from datetime import datetime, timezone
+import json
+import os
+from pathlib import Path
+from uuid import uuid4
+
+from fasthtml.common import *
+from fastsql import Database
+from monsterui.all import (
+    Button,
+    ButtonT,
+    Container,
+    ContainerT,
+    FastHTML as MUFastHTML,
+    TextPresets,
+    Theme,
+    UkIcon,
+)
+
+from pylogue.core import (
+    EchoResponder,
+    IMPORT_PREFIX,
+    _register_google_auth_routes,
+    _session_cookie_name,
+    google_oauth_config_from_env,
+    get_core_headers,
+    register_core_static,
+    register_ws_routes,
+    render_cards,
+    render_input,
+)
+from starlette.requests import Request
+from starlette.responses import FileResponse
+from starlette.responses import JSONResponse
+from starlette.responses import RedirectResponse
+
+PROJECT_ROOT = Path(__file__).resolve().parents[2]
+CHAT_APP_DIR = PROJECT_ROOT / "scripts" / "examples" / "chat_app_with_histories"
+STATIC_DIR = CHAT_APP_DIR / "static"
+DB_PATH = CHAT_APP_DIR / "chat_app.db"
+db = Database(f"sqlite:///{DB_PATH}")
+
+
+@dataclass
+class Chat:
+    id: str
+    title: str
+    created_at: str
+    updated_at: str
+    payload: str = ""
+
+
+chats = db.create(Chat, pk="id")
+
+
+def _utc_iso() -> str:
+    return datetime.now(timezone.utc).replace(microsecond=0).isoformat()
+
+
+def app_factory(
+    responder=None,
+    responder_factory=None,
+    db_path: Path | str | None = None,
+    sidebar_title: str = "Pylogue",
+    hero_title: str = "Fast HTML + Pylogue Core",
+    hero_subtitle: str = (
+        "One UI wraps multiple Pylogue chat sessions. Pick a chat on the left, "
+        "start a new one, or return to previous conversations instantly."
+    ),
+) -> MUFastHTML:
+    resolved_db_path = Path(db_path) if db_path is not None else DB_PATH
+    local_db = Database(f"sqlite:///{resolved_db_path}")
+    if responder_factory is None:
+        responder = responder or EchoResponder()
+    headers = list(get_core_headers(include_markdown=True))
+    headers.extend(
+        [
+            Link(
+                rel="stylesheet",
+                href="https://fonts.googleapis.com/css2?family=Sora:wght@300;400;500;600;700&display=swap",
+            ),
+            Link(rel="stylesheet", href="/static/chat_app.css"),
+            Script(src="/static/chat_app.js", type="module"),
+        ]
+    )
+
+    oauth_cfg = google_oauth_config_from_env()
+    auth_required = bool(oauth_cfg and oauth_cfg.auth_required)
+    session_secret = (
+        oauth_cfg.session_secret
+        if oauth_cfg and oauth_cfg.session_secret
+        else os.getenv("PYLOGUE_SESSION_SECRET")
+    )
+    app_kwargs = {"exts": "ws", "hdrs": tuple(headers), "pico": False}
+    app_kwargs["session_cookie"] = _session_cookie_name()
+    if session_secret:
+        app_kwargs["secret_key"] = session_secret
+    app = MUFastHTML(**app_kwargs)
+    register_core_static(app)
+    auth_paths = _register_google_auth_routes(app, oauth_cfg) if oauth_cfg else None
+
+    def _is_authorized(request: Request) -> bool:
+        if not auth_required:
+            return True
+        auth = request.session.get("auth")
+        return isinstance(auth, dict)
+
+    @app.route("/static/chat_app.css")
+    def _chat_app_css():
+        return FileResponse(STATIC_DIR / "chat_app.css")
+
+    @app.route("/static/chat_app.js")
+    def _chat_app_js():
+        return FileResponse(STATIC_DIR / "chat_app.js")
+
+    @app.route("/api/chats", methods=["GET"])
+    def list_chats(request: Request):
+        if not _is_authorized(request):
+            return JSONResponse({"error": "Unauthorized"}, status_code=401)
+        items = list(local_db.create(Chat, pk="id")())
+        items.sort(key=lambda c: c.updated_at or c.created_at, reverse=True)
+        return JSONResponse(
+            [
+                {
+                    "id": c.id,
+                    "title": c.title,
+                    "created_at": c.created_at,
+                    "updated_at": c.updated_at,
+                }
+                for c in items
+            ]
+        )
+
+    @app.route("/api/chats", methods=["POST"])
+    async def create_chat(request: Request):
+        if not _is_authorized(request):
+            return JSONResponse({"error": "Unauthorized"}, status_code=401)
+        chats = local_db.create(Chat, pk="id")
+        data = await request.json()
+        chat_id = data.get("id") or str(uuid4())
+        title = data.get("title") or "New chat"
+        now = _utc_iso()
+        payload = data.get("payload")
+        payload_str = json.dumps(payload) if payload is not None else ""
+        chat = Chat(chat_id, title, now, now, payload_str)
+        try:
+            _ = chats[chat_id]
+            chats.update(chat)
+        except Exception:
+            chats.insert(chat)
+        return JSONResponse(
+            {
+                "id": chat.id,
+                "title": chat.title,
+                "created_at": chat.created_at,
+                "updated_at": chat.updated_at,
+            }
+        )
+
+    @app.route("/api/chats/{chat_id}", methods=["GET"])
+    def get_chat(request: Request, chat_id: str):
+        if not _is_authorized(request):
+            return JSONResponse({"error": "Unauthorized"}, status_code=401)
+        chats = local_db.create(Chat, pk="id")
+        try:
+            chat = chats[chat_id]
+        except Exception:
+            return JSONResponse({"cards": []})
+        payload = chat.payload or ""
+        if not payload:
+            return JSONResponse({"cards": []})
+        try:
+            data = json.loads(payload)
+        except json.JSONDecodeError:
+            data = {"cards": []}
+        return JSONResponse(data)
+
+    @app.route("/api/chats/{chat_id}", methods=["POST"])
+    async def save_chat(chat_id: str, request: Request):
+        if not _is_authorized(request):
+            return JSONResponse({"error": "Unauthorized"}, status_code=401)
+        chats = local_db.create(Chat, pk="id")
+        data = await request.json()
+        payload = data.get("payload") or {"cards": []}
+        title = data.get("title") or "New chat"
+        now = _utc_iso()
+        try:
+            existing = chats[chat_id]
+            created_at = existing.created_at
+        except Exception:
+            created_at = data.get("created_at") or now
+        chat = Chat(chat_id, title, created_at, now, json.dumps(payload))
+        try:
+            _ = chats[chat_id]
+            chats.update(chat)
+        except Exception:
+            chats.insert(chat)
+        return JSONResponse(
+            {
+                "id": chat.id,
+                "title": chat.title,
+                "created_at": chat.created_at,
+                "updated_at": chat.updated_at,
+            }
+        )
+
+    @app.route("/api/chats/{chat_id}", methods=["DELETE"])
+    def delete_chat(request: Request, chat_id: str):
+        if not _is_authorized(request):
+            return JSONResponse({"error": "Unauthorized"}, status_code=401)
+        chats = local_db.create(Chat, pk="id")
+        try:
+            chats.delete(chat_id)
+        except Exception:
+            pass
+        return JSONResponse({"deleted": True})
+
+    sessions: dict[int, dict] = {}
+    register_ws_routes(
+        app,
+        responder=responder,
+        responder_factory=responder_factory,
+        sessions=sessions,
+        auth_required=auth_required,
+    )
+
+    def _sidebar(request: Request):
+        show_logout = auth_required and _is_authorized(request)
+        logout_href = auth_paths["logout_path"] if auth_paths else "/logout"
+        return Div(
+            Div(
+                H1(sidebar_title, cls="text-xl font-semibold"),
+                A(
+                    UkIcon("sign-out"),
+                    Span("Logout"),
+                    href=logout_href,
+                    cls=(ButtonT.secondary, "text-xs"),
+                ) if show_logout else None,
+                cls="sidebar-header",
+            ),
+            Button(
+                UkIcon("plus"),
+                Span("New chat"),
+                cls=(ButtonT.secondary, "w-full justify-center"),
+                type="button",
+                id="new-chat-btn",
+            ),
+            Div(id="chat-list", cls="chat-list"),
+            cls="sidebar",
+        )
+
+    def _hero():
+        return Div(
+            Div(
+                H2(hero_title, cls="hero-title"),
+                P(hero_subtitle, cls="hero-sub"),
+                cls="space-y-2",
+            ),
+            Div(
+                Button(
+                    UkIcon("download"),
+                    cls="uk-button uk-button-text copy-chat-btn",
+                    type="button",
+                    aria_label="Download conversation JSON",
+                    title="Download conversation JSON",
+                ),
+                Button(
+                    UkIcon("upload"),
+                    cls="uk-button uk-button-text copy-chat-btn upload-chat-btn",
+                    type="button",
+                    aria_label="Upload conversation JSON",
+                    title="Upload conversation JSON",
+                ),
+                Input(
+                    type="file",
+                    id="chat-upload",
+                    accept="application/json",
+                    cls="sr-only",
+                ),
+                cls="flex gap-2 justify-end",
+            ),
+            cls="hero",
+        )
+
+    def _chat_panel():
+        return Div(
+            Div(render_cards([])),
+            Form(
+                render_input(),
+                Div(
+                    Button("Send", cls=ButtonT.primary, type="submit", id="chat-send-btn"),
+                    P("Cmd/Ctrl+Enter to send", cls="text-xs text-slate-400"),
+                    cls="flex flex-col gap-2 items-stretch",
+                ),
+                id="form",
+                hx_ext="ws",
+                ws_connect="/ws",
+                ws_send=True,
+                hx_target="#cards",
+                hx_swap="outerHTML",
+                cls="flex flex-col sm:flex-row gap-3 items-stretch pt-4",
+            ),
+            cls="chat-panel space-y-4",
+        )
+
+    def _main_panel():
+        return Div(
+            _hero(),
+            _chat_panel(),
+            cls="main-panel space-y-6",
+        )
+
+    def _shell(request: Request):
+        return Div(
+            _sidebar(request),
+            _main_panel(),
+            cls="app-shell",
+        )
+
+    @app.route("/")
+    def home(request: Request):
+        if auth_required and not _is_authorized(request):
+            request.session["next"] = "/"
+            login_path = auth_paths["login_path"] if auth_paths else "/login"
+            return RedirectResponse(login_path, status_code=303)
+        return (
+            Title(hero_title),
+            Meta(name="viewport", content="width=device-width, initial-scale=1.0"),
+            Body(
+                _shell(request),
+                cls="min-h-screen",
+                data_import_prefix=IMPORT_PREFIX,
+            ),
+        )
+
+    return app

pylogue/static/pylogue-core.css

@@ -0,0 +1,372 @@
+html, body {
+  margin: 0;
+  background: #f8fafc;
+  color: #0f172a;
+}
+
+html {
+  color-scheme: light;
+}
+
+:root, .uk-theme-slate {
+  --background: 0 0% 98%;
+  --foreground: 222 47% 11%;
+}
+
+.chat-panel {
+  background: #ffffff;
+  border: 1px solid #e2e8f0;
+  border-radius: 16px;
+  padding: 20px;
+}
+
+.chat-row-user {
+  background: #e2e8f0;
+  border-left: 3px solid #cbd5f5;
+}
+
+.chat-row-assistant {
+  background: #f8fafc;
+  border-left: 3px solid #e2e8f0;
+}
+
+.chat-row-block {
+  margin: 0;
+  padding: 14px 12px;
+}
+
+.tool-status {
+  display: inline-block;
+  padding: 0.18rem 0.45rem;
+  border-radius: 999px;
+  font-size: 0.78rem;
+  font-weight: 600;
+  letter-spacing: 0.005em;
+  background: rgba(148, 163, 184, 0.15);
+  color: #0f172a;
+}
+
+.tool-status--running {
+  position: relative;
+  overflow: hidden;
+  color: #0f172a;
+}
+
+.tool-status--running::after {
+  content: "";
+  position: absolute;
+  top: 0;
+  left: -150%;
+  width: 150%;
+  height: 100%;
+  background: linear-gradient(
+    120deg,
+    transparent 0%,
+    rgba(34, 197, 94, 0.15),
+    transparent 80%
+  );
+  animation: tool-shimmer 1.4s linear infinite;
+}
+
+.tool-status--done {
+  background: rgba(255, 255, 255, 0.7) 40%;
+  color: #14532d;
+}
+
+@keyframes tool-shimmer {
+  0% { transform: translateX(0); }
+  100% { transform: translateX(100%); }
+}
+
+.marked a {
+  color: #2563eb;
+  text-decoration: underline;
+  text-underline-offset: 3px;
+  text-decoration-thickness: 1.5px;
+}
+
+.marked a:hover {
+  color: #1d4ed8;
+}
+
+.marked table {
+  width: 100%;
+  border-collapse: collapse;
+  margin: 16px 0;
+  font-size: 0.95rem;
+}
+
+.marked thead th {
+  background: #f1f5f9;
+  color: #0f172a;
+  font-weight: 600;
+  border: 1px solid #e2e8f0;
+  padding: 10px 12px;
+  text-align: left;
+}
+
+.marked tbody td {
+  border: 1px solid #e2e8f0;
+  padding: 10px 12px;
+  vertical-align: top;
+}
+
+.marked tbody tr:nth-child(odd) {
+  background: #f8fafc;
+}
+
+.marked tbody tr:hover {
+  background: #eef2ff;
+}
+
+.marked blockquote {
+  margin: 16px 0;
+  padding: 12px 14px;
+  border-left: 3px solid #e2e8f0;
+  background: #f8fafc;
+  color: #475569;
+  border-radius: 8px;
+}
+
+.marked pre {
+  background: #f8fafc;
+  color: #0f172a;
+  padding: 12px 14px;
+  border-radius: 12px;
+  border: 1px solid #e2e8f0;
+  overflow: auto;
+  text-align: left;
+  position: relative;
+}
+
+.marked {
+  font-family: "SFMono-Regular", Menlo, Monaco, Consolas, "Liberation Mono", "Courier New", monospace;
+}
+
+.marked :not(pre) > code {
+  background: #f1f5f9;
+  color: #0f172a;
+  padding: 0.15rem 0.35rem;
+  border-radius: 6px;
+  font-size: 0.95em;
+}
+
+.marked pre code {
+  background: transparent;
+  padding: 0;
+  border-radius: 0;
+  text-align: left;
+}
+
+.marked pre .code-copy-btn {
+  position: absolute;
+  top: 8px;
+  right: 8px;
+  display: inline-flex;
+  align-items: center;
+  justify-content: center;
+  width: 28px;
+  height: 28px;
+  border: 1px solid #e2e8f0;
+  border-radius: 8px;
+  background: rgba(248, 250, 252, 0.9);
+  color: #64748b;
+  opacity: 0;
+  transform: translateY(-2px);
+  transition: opacity 0.15s ease, transform 0.15s ease, color 0.15s ease, border-color 0.15s ease;
+  cursor: pointer;
+  pointer-events: none;
+}
+
+.marked pre:hover .code-copy-btn,
+.marked pre:focus-within .code-copy-btn {
+  opacity: 1;
+  transform: translateY(0);
+  pointer-events: auto;
+}
+
+.marked pre .code-copy-btn svg {
+  width: 16px;
+  height: 16px;
+}
+
+.marked pre .code-copy-btn:hover {
+  color: #0f172a;
+  border-color: #cbd5f5;
+}
+
+.marked pre .code-copy-btn[data-copied="true"] {
+  color: #16a34a;
+  border-color: #86efac;
+  background: #f0fdf4;
+}
+
+.marked .mermaid-container {
+  position: relative;
+  border: 1px solid #e2e8f0;
+  border-radius: 12px;
+  background: #ffffff;
+  padding: 8px;
+  margin: 16px 0;
+  box-shadow: 0 1px 3px rgba(15, 23, 42, 0.08);
+}
+
+.marked .mermaid-wrapper {
+  min-height: 0;
+  height: fit-content;
+  overflow: hidden;
+  position: relative;
+}
+
+.marked .mermaid-wrapper svg {
+  display: block;
+  width: 100%;
+  height: auto;
+  pointer-events: none;
+}
+
+.marked .mermaid-controls {
+  position: absolute;
+  top: 8px;
+  right: 8px;
+  display: inline-flex;
+  gap: 6px;
+  background: rgba(255, 255, 255, 0.95);
+  border: 1px solid #e2e8f0;
+  border-radius: 999px;
+  padding: 4px 6px;
+  z-index: 10;
+}
+
+.marked .mermaid-controls button {
+  border: 0;
+  background: transparent;
+  color: #475569;
+  font-size: 12px;
+  line-height: 1;
+  padding: 4px 6px;
+  border-radius: 999px;
+  cursor: pointer;
+}
+
+.marked .mermaid-controls button:hover {
+  background: #f1f5f9;
+  color: #0f172a;
+}
+
+.marked .mermaid-error {
+  border: 1px dashed #cbd5e1;
+  border-radius: 10px;
+  background: #f8fafc;
+  color: #475569;
+  font-size: 0.9rem;
+  padding: 10px 12px;
+  margin: 8px 0;
+}
+
+.marked ul,
+.marked ol {
+  margin: 12px 0;
+  padding-left: 1.25rem;
+}
+
+.marked hr {
+  border: 0;
+  border-top: 1px solid #e2e8f0;
+  margin: 20px 0;
+}
+
+#cards,
+.chat-panel {
+  overflow-anchor: none;
+}
+
+.marked details.tool-call {
+  background: #f1f5f9;
+  border: 1px solid #e2e8f0;
+  border-radius: 12px;
+  padding: 6px 8px;
+  color: #475569;
+  margin: 14px 0;
+  font-size: 0.85rem;
+}
+
+.marked details.tool-call summary {
+  cursor: pointer;
+  color: #64748b;
+  font-weight: 600;
+  font-size: 0.85rem;
+}
+
+.marked details.tool-call pre {
+  background: #e2e8f0;
+  color: #334155;
+  border-color: #cbd5f5;
+  padding: 8px 10px;
+  font-size: 0.82rem;
+}
+
+.copy-btn {
+  border: 1px solid #e2e8f0;
+  border-radius: 6px;
+  width: 24px;
+  height: 24px;
+  padding: 0;
+  color: #94a3b8;
+}
+
+.copy-btn:hover {
+  color: #475569;
+}
+
+.copy-chat-btn {
+  border: 1px solid #e2e8f0;
+  border-radius: 8px;
+  padding: 4px 10px;
+  color: #475569;
+  gap: 6px;
+  display: inline-flex;
+  align-items: center;
+}
+
+.copy-chat-btn:hover {
+  color: #1f2937;
+}
+
+.copy-chat-btn svg {
+  width: 14px;
+  height: 14px;
+}
+
+.copy-btn svg {
+  width: 12px;
+  height: 12px;
+}
+
+.copy-btn[data-copied="true"] {
+  color: #16a34a;
+  border-color: #86efac;
+  background: #f0fdf4;
+}
+
+.copy-btn .copy-label {
+  display: none;
+}
+
+.copy-chat-btn[data-copied="true"] {
+  color: #16a34a;
+  border-color: #86efac;
+  background: #f0fdf4;
+}
+
+.sr-only {
+  position: absolute;
+  width: 1px;
+  height: 1px;
+  padding: 0;
+  margin: -1px;
+  overflow: hidden;
+  clip: rect(0, 0, 0, 0);
+  white-space: nowrap;
+  border: 0;
+}