Add pydantic, add theme reactive to app

darrenburns · Sep 10, 2024 · f1cabd4 · f1cabd4
1 parent 35d9068
commit f1cabd4
Show file tree

Hide file tree

Showing 6 changed files with 55 additions and 4 deletions.
diff --git a/elia_chat/app.py b/elia_chat/app.py
@@ -6,6 +6,7 @@
 
 from textual.app import App
 from textual.binding import Binding
+from textual.reactive import Reactive, reactive
 from textual.signal import Signal
 
 from elia_chat.chats_manager import ChatsManager
@@ -52,6 +53,8 @@ def __init__(self, config: LaunchConfig, startup_prompt: str = ""):
         put users into the chat window, rather than going to the home screen.
         """
 
+    theme: Reactive[str] = reactive("elia", init=False)
+
     @property
     def runtime_config(self) -> RuntimeConfig:
         return self._runtime_config

diff --git a/elia_chat/config.py b/elia_chat/config.py
@@ -86,9 +86,7 @@ def get_builtin_anthropic_models() -> list[EliaChatModel]:
             display_name="Claude 3.5 Sonnet",
             provider="Anthropic",
             product="Claude 3.5",
-            description=(
-                "Anthropic's most intelligent model"
-            ),
+            description=("Anthropic's most intelligent model"),
         ),
         EliaChatModel(
             id="elia-claude-3-haiku-20240307",
@@ -172,6 +170,7 @@ class LaunchConfig(BaseModel):
     builtin_models: list[EliaChatModel] = Field(
         default_factory=get_builtin_models, init=False
     )
+    theme: str = Field(default="elia")
 
     @property
     def all_models(self) -> list[EliaChatModel]:

diff --git a/elia_chat/locations.py b/elia_chat/locations.py
@@ -21,3 +21,10 @@ def config_directory() -> Path:
 
 def config_file() -> Path:
     return config_directory() / "config.toml"
+
+
+def theme_directory() -> Path:
+    """Return (possibly creating) the themes directory."""
+    theme_dir = data_directory() / "themes"
+    theme_dir.mkdir(exist_ok=True, parents=True)
+    return theme_dir
diff --git a/elia_chat/themes.py b/elia_chat/themes.py
@@ -0,0 +1,39 @@
+from pydantic import BaseModel, Field
+import yaml
+
+from elia_chat.locations import theme_directory
+
+
+class Theme(BaseModel):
+    name: str = Field(exclude=True)
+    primary: str
+    secondary: str | None = None
+    background: str | None = None
+    surface: str | None = None
+    panel: str | None = None
+    warning: str | None = None
+    error: str | None = None
+    success: str | None = None
+    accent: str | None = None
+    dark: bool = True
+
+
+def load_user_themes() -> dict[str, Theme]:
+    """Load user themes from "~/.config/elia/themes".
+
+    Returns:
+        A dictionary mapping theme names to theme objects.
+    """
+    themes: dict[str, Theme] = {}
+    for path in theme_directory().iterdir():
+        path_suffix = path.suffix
+        if path_suffix == ".yaml" or path_suffix == ".yml":
+            with path.open() as theme_file:
+                theme_content = yaml.load(theme_file, Loader=yaml.FullLoader) or {}
+                try:
+                    themes[theme_content["name"]] = Theme(**theme_content)
+                except KeyError:
+                    raise ValueError(
+                        f"Invalid theme file {path}. A `name` is required."
+                    )
+    return themes
diff --git a/pyproject.toml b/pyproject.toml
@@ -17,6 +17,7 @@ dependencies = [
     "google-generativeai>=0.5.3",
     "pyperclip>=1.8.2",
     "litellm>=1.37.19",
+    "pydantic>=2.9.0",
 ]
 readme = "README.md"
 requires-python = ">= 3.11"

diff --git a/uv.lock b/uv.lock