Upload jarvis/ui.py

jarvis/ui.py

@@ -0,0 +1,103 @@
+"""Terminal chat interface for J.A.R.V.I.S. using Rich."""
+import sys
+from typing import Optional
+
+from rich.console import Console
+from rich.panel import Panel
+from rich.text import Text
+from rich.prompt import Prompt
+from rich.markdown import Markdown
+
+from jarvis.config import Config
+from jarvis.memory import Memory
+from jarvis.llm import get_backend, BaseBackend
+from jarvis.pipeline import run_pipeline
+
+
+class TerminalUI:
+    def __init__(self, config: Config, memory: Memory, backend: BaseBackend):
+        self.console = Console()
+        self.config = config
+        self.memory = memory
+        self.backend = backend
+        self.persona_name = config.persona.name
+        self.user_name = config.persona.user_name
+
+    def _header(self):
+        self.console.print()
+        self.console.print(
+            Panel.fit(
+                f"[bold cyan]{self.persona_name}[/bold cyan] — Alvin OS v0.1.0 (MVP)\n"
+                f"[dim]Type 'exit', 'quit', or press Ctrl-C to leave.[/dim]",
+                title="[bold green]Personal AI Assistant[/bold green]",
+                border_style="cyan",
+            )
+        )
+        self.console.print()
+
+    def _print_user(self, text: str):
+        self.console.print(
+            Panel(
+                text,
+                title=f"[bold yellow]{self.user_name}[/bold yellow]",
+                border_style="yellow",
+                width=min(80, self.console.width - 4),
+            )
+        )
+
+    def _print_jarvis(self, text: str, metadata: Optional[dict] = None):
+        # Render markdown in the response
+        md = Markdown(text)
+        meta_text = ""
+        if metadata:
+            meta_text = (
+                f"\n[dim]Contract: {metadata.get('contract')} | "
+                f"Dial: {metadata.get('dial')} | "
+                f"Mix: {metadata.get('archetype_mix')}[/dim]"
+            )
+        self.console.print(
+            Panel(
+                md,
+                title=f"[bold cyan]{self.persona_name}[/bold cyan]",
+                subtitle=meta_text,
+                border_style="cyan",
+                width=min(80, self.console.width - 4),
+            )
+        )
+
+    def _print_system(self, text: str):
+        self.console.print(f"[dim italic]{text}[/dim italic]")
+
+    def run(self):
+        self._header()
+        try:
+            while True:
+                user_input = Prompt.ask(f"[bold yellow]{self.user_name}[/bold yellow]")
+                user_input = user_input.strip()
+                if not user_input:
+                    continue
+                if user_input.lower() in ("exit", "quit", "q"):
+                    self._print_system("Shutting down. Goodbye, sir.")
+                    break
+
+                self._print_user(user_input)
+                self._print_system("Processing...")
+
+                result = run_pipeline(
+                    user_message=user_input,
+                    config=self.config,
+                    memory=self.memory,
+                    backend=self.backend,
+                )
+
+                self._print_jarvis(result["response"], metadata={
+                    "contract": result["contract"],
+                    "dial": result["dial"],
+                    "archetype_mix": result["archetype_mix"],
+                })
+                self.console.print()
+
+        except KeyboardInterrupt:
+            self.console.print()
+            self._print_system("Interrupted. Goodbye, sir.")
+            sys.exit(0)