vedana-backoffice 0.6.1__tar.gz → 0.6.3__tar.gz

{vedana_backoffice-0.6.1 → vedana_backoffice-0.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vedana-backoffice
-Version: 0.6.1
+Version: 0.6.3
 Summary: Reflex-based admin/backoffice UI for Vedana
 Author-email: Andrey Tatarinov <a@tatarinov.co>, Timur Sheydaev <tsheyd@epoch8.co>
 Requires-Python: >=3.12

{vedana_backoffice-0.6.1 → vedana_backoffice-0.6.3}/src/vedana_backoffice/states/common.py

@@ -9,6 +9,9 @@ from vedana_core.app import VedanaApp, make_vedana_app
 vedana_app: VedanaApp | None = None
 
 EVAL_ENABLED = bool(os.environ.get("GRIST_TEST_SET_DOC_ID"))
+DEBUG_MODE = os.environ.get("DEBUG", "").lower() in ("true", "1")
+HAS_OPENAI_KEY = bool(os.environ.get("OPENAI_API_KEY"))
+HAS_OPENROUTER_KEY = bool(os.environ.get("OPENROUTER_API_KEY"))
 
 
 async def get_vedana_app():
@@ -40,6 +43,48 @@ class MemLogger(logging.Logger):
 class AppVersionState(rx.State):
     version: str = f"`{os.environ.get('VERSION', 'unspecified_version')}`"  # md-formatted
     eval_enabled: bool = EVAL_ENABLED
+    debug_mode: bool = DEBUG_MODE
+
+
+class DebugState(rx.State):
+    """State for debug mode API key setup."""
+
+    debug_mode: bool = DEBUG_MODE
+    needs_api_key: bool = DEBUG_MODE and not HAS_OPENAI_KEY and not HAS_OPENROUTER_KEY
+    show_api_key_dialog: bool = False
+    api_key: str = ""
+    api_key_type: str = "openai"  # "openai" or "openrouter"
+    api_key_saved: bool = False
+
+    def check_and_show_dialog(self) -> None:
+        """Called on app mount to show dialog if needed."""
+        if self.needs_api_key and not self.api_key_saved:
+            self.show_api_key_dialog = True
+
+    def set_api_key(self, value: str) -> None:
+        self.api_key = value
+
+    def set_api_key_type(self, value: str) -> None:
+        self.api_key_type = value
+
+    def save_api_key(self) -> None:
+        """Save the API key and close dialog."""
+        if self.api_key.strip() and self.debug_mode:  # extra check for debug mode
+            self.api_key_saved = True
+            self.show_api_key_dialog = False
+            # Set environment variable so it's available throughout the app
+            if self.api_key_type == "openai":
+                os.environ["OPENAI_API_KEY"] = self.api_key.strip()
+            else:
+                os.environ["OPENROUTER_API_KEY"] = self.api_key.strip()
+
+    def close_dialog(self) -> None:
+        """Close dialog without saving."""
+        self.show_api_key_dialog = False
+
+    def open_dialog(self) -> None:
+        """Manually open the dialog to change API key."""
+        self.show_api_key_dialog = True
 
 
 class TelegramBotState(rx.State):

vedana_backoffice-0.6.3/src/vedana_backoffice/ui.py

@@ -0,0 +1,193 @@
+from __future__ import annotations
+
+from typing import Any
+
+import reflex as rx
+
+from vedana_backoffice.states.chat import ChatState
+from vedana_backoffice.states.common import AppVersionState, DebugState, TelegramBotState
+
+
+def telegram_link_box() -> rx.Component:
+    return rx.box(
+        rx.cond(
+            TelegramBotState.has_bot,
+            rx.link(
+                rx.hstack(
+                    rx.text("Telegram", font_size="1.1em"),
+                    # rx.icon("external-link", stroke_width=1, size=10),
+                    # spacing="1",
+                ),
+                href=TelegramBotState.bot_url,
+                is_external=True,
+            ),
+        ),
+        on_mount=TelegramBotState.load_bot_info,
+    )
+
+
+def data_model_reload_btn() -> rx.Component:
+    return rx.button(
+        "Reload Data Model",
+        variant="soft",
+        color_scheme="blue",
+        on_click=ChatState.reload_data_model,
+        loading=ChatState.is_refreshing_dm,
+    )
+
+
+def debug_badge() -> rx.Component:
+    """Red badge indicating debug mode is active. Clickable to open API key dialog."""
+    return rx.cond(
+        AppVersionState.debug_mode,
+        rx.tooltip(
+            rx.badge(
+                "DEBUG MODE",
+                color_scheme="red",
+                variant="solid",
+                size="2",
+                style={
+                    "font_weight": "bold",
+                    "text_transform": "uppercase",
+                    "cursor": "pointer",
+                },
+                on_click=DebugState.open_dialog,
+            ),
+            content="Debug mode enabled! Some features are not for production use. Click to reset LLM API_KEY"
+        ),
+        rx.fragment(),
+    )
+
+
+def api_key_setup_dialog() -> rx.Component:
+    """Dialog to prompt user for API key in debug mode when no keys are configured."""
+    return rx.dialog.root(
+        rx.dialog.content(
+            rx.dialog.title("API Key Setup"),
+            rx.dialog.description(
+                "Please provide an OpenAI or OpenRouter API key to use the chat functionality.",
+                margin_bottom="1em",
+            ),
+            rx.vstack(
+                rx.select(
+                    ["openai", "openrouter"],
+                    value=DebugState.api_key_type,
+                    on_change=DebugState.set_api_key_type,
+                    placeholder="Select API provider",
+                ),
+                rx.input(
+                    placeholder="Enter your API key",
+                    value=DebugState.api_key,
+                    on_change=DebugState.set_api_key,
+                    type="password",
+                    width="100%",
+                ),
+                rx.hstack(
+                    rx.dialog.close(
+                        rx.button(
+                            "Skip",
+                            variant="soft",
+                            color_scheme="gray",
+                            on_click=DebugState.close_dialog,
+                        ),
+                    ),
+                    rx.button(
+                        "Save",
+                        color_scheme="blue",
+                        on_click=DebugState.save_api_key,
+                    ),
+                    justify="end",
+                    spacing="3",
+                    width="100%",
+                ),
+                spacing="4",
+                width="100%",
+            ),
+            style={"max_width": "450px"},
+        ),
+        open=DebugState.show_api_key_dialog,
+    )
+
+
+def app_header() -> rx.Component:
+    return rx.fragment(
+        rx.box(
+            rx.hstack(
+                rx.hstack(
+                    rx.link("Vedana Backoffice", href="/", font_weight="bold", font_size="1.25em"),
+                    rx.markdown(AppVersionState.version),  # type: ignore[operator]
+                    debug_badge(),
+                    align="center",
+                    spacing="3",
+                ),
+                rx.hstack(
+                    data_model_reload_btn(),
+                    rx.link("ETL", href="/etl", font_size="1.1em"),
+                    rx.cond(
+                        AppVersionState.eval_enabled,
+                        rx.link("Eval", href="/eval", font_size="1.1em"),
+                        rx.fragment(),
+                    ),
+                    rx.link("Chat", href="/chat", font_size="1.1em"),
+                    rx.link("JIMS", href="/jims", font_size="1.1em"),
+                    telegram_link_box(),
+                    rx.color_mode.button(),  # type: ignore[attr-defined]
+                    spacing="6",
+                    align="center",
+                ),
+                justify="between",
+                align="center",
+                width="100%",
+            ),
+            width="100%",
+            padding="0.5em 1.25em",
+            border_bottom="1px solid #e5e7eb",
+            position="sticky",
+            top="0",
+            background_color=rx.color_mode_cond(light="white", dark="black"),
+            style={
+                "backdrop-filter": "blur(10px)",  # enables non-transparent background
+            },
+            z_index="10",
+            on_mount=DebugState.check_and_show_dialog,
+        ),
+        api_key_setup_dialog(),
+    )
+
+
+def themed_data_table(
+    *,
+    data: rx.Var | list[Any],
+    columns: rx.Var | list[str],
+    width: str | rx.Var = "fit-content",
+    max_width: str | rx.Var = "100%",
+    **kwargs: Any,
+) -> rx.Component:
+    """Wrap rx.data_table with shared styling so it matches the app theme."""
+
+    default_kwargs = {"pagination": True, "search": True, "sort": True}
+    table_kwargs = {**default_kwargs, **kwargs}
+
+    table_style = table_kwargs.pop("style", {})
+    table_style = {"width": "fit-content", "minWidth": "100%", **table_style}
+
+    container_style: dict[str, Any] = {
+        "width": width,
+        "maxWidth": max_width,
+        "minWidth": "fit-content",
+    }
+
+    return rx.box(
+        rx.data_table(data=data, columns=columns, style=table_style, **table_kwargs),
+        class_name="datatable-surface",
+        style=container_style,
+    )
+
+
+def breadcrumbs(items: list[tuple[str, str]]) -> rx.Component:
+    parts: list[rx.Component] = []
+    for idx, (label, href) in enumerate(items):
+        parts.append(rx.link(label, href=href))
+        if idx < len(items) - 1:
+            parts.append(rx.text("→", color="#9ca3af"))
+    return rx.hstack(*parts, spacing="2", padding_y="0.5em")

vedana_backoffice-0.6.1/src/vedana_backoffice/ui.py

@@ -1,115 +0,0 @@
-from __future__ import annotations
-
-from typing import Any
-
-import reflex as rx
-
-from vedana_backoffice.states.chat import ChatState
-from vedana_backoffice.states.common import AppVersionState, TelegramBotState
-
-
-def telegram_link_box() -> rx.Component:
-    return rx.box(
-        rx.cond(
-            TelegramBotState.has_bot,
-            rx.link(
-                rx.hstack(
-                    rx.text("Telegram", font_size="1.1em"),
-                    # rx.icon("external-link", stroke_width=1, size=10),
-                    # spacing="1",
-                ),
-                href=TelegramBotState.bot_url,
-                is_external=True,
-            ),
-        ),
-        on_mount=TelegramBotState.load_bot_info,
-    )
-
-
-def data_model_reload_btn() -> rx.Component:
-    return rx.button(
-        "Reload Data Model",
-        variant="soft",
-        color_scheme="blue",
-        on_click=ChatState.reload_data_model,
-        loading=ChatState.is_refreshing_dm,
-    )
-
-
-def app_header() -> rx.Component:
-    return rx.box(
-        rx.hstack(
-            rx.hstack(
-                rx.link("Vedana Backoffice", href="/", font_weight="bold", font_size="1.25em"),
-                rx.markdown(AppVersionState.version),  # type: ignore[operator]
-                align="center",
-                spacing="3",
-            ),
-            rx.hstack(
-                data_model_reload_btn(),
-                rx.link("ETL", href="/etl", font_size="1.1em"),
-                rx.cond(
-                    AppVersionState.eval_enabled,
-                    rx.link("Eval", href="/eval", font_size="1.1em"),
-                    rx.fragment(),
-                ),
-                rx.link("Chat", href="/chat", font_size="1.1em"),
-                rx.link("JIMS", href="/jims", font_size="1.1em"),
-                telegram_link_box(),
-                rx.color_mode.button(),  # type: ignore[attr-defined]
-                spacing="6",
-                align="center",
-            ),
-            justify="between",
-            align="center",
-            width="100%",
-        ),
-        width="100%",
-        padding="0.5em 1.25em",
-        border_bottom="1px solid #e5e7eb",
-        position="sticky",
-        top="0",
-        background_color=rx.color_mode_cond(light="white", dark="black"),
-        style={
-            "backdrop-filter": "blur(10px)",  # enables non-transparent background
-        },
-        z_index="10",
-    )
-
-
-def themed_data_table(
-    *,
-    data: rx.Var | list[Any],
-    columns: rx.Var | list[str],
-    width: str | rx.Var = "fit-content",
-    max_width: str | rx.Var = "100%",
-    **kwargs: Any,
-) -> rx.Component:
-    """Wrap rx.data_table with shared styling so it matches the app theme."""
-
-    default_kwargs = {"pagination": True, "search": True, "sort": True}
-    table_kwargs = {**default_kwargs, **kwargs}
-
-    table_style = table_kwargs.pop("style", {})
-    table_style = {"width": "fit-content", "minWidth": "100%", **table_style}
-
-    container_style: dict[str, Any] = {
-        "width": width,
-        "maxWidth": max_width,
-        "minWidth": "fit-content",
-    }
-
-    return rx.box(
-        rx.data_table(data=data, columns=columns, style=table_style, **table_kwargs),
-        class_name="datatable-surface",
-        style=container_style,
-    )
-
-
-def breadcrumbs(items: list[tuple[str, str]]) -> rx.Component:
-    parts: list[rx.Component] = []
-    for idx, (label, href) in enumerate(items):
-        parts.append(rx.link(label, href=href))
-        if idx < len(items) - 1:
-            parts.append(rx.text("→", color="#9ca3af"))
-    return rx.hstack(*parts, spacing="2", padding_y="0.5em")