vedana-backoffice 0.6.1__py3-none-any.whl → 0.6.3__py3-none-any.whl

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/ui.py

@@ -5,7 +5,7 @@ from typing import Any
 import reflex as rx
 
 from vedana_backoffice.states.chat import ChatState
-from vedana_backoffice.states.common import AppVersionState, TelegramBotState
+from vedana_backoffice.states.common import AppVersionState, DebugState, TelegramBotState
 
 
 def telegram_link_box() -> rx.Component:
@@ -36,44 +36,122 @@ def data_model_reload_btn() -> rx.Component:
     )
 
 
-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",
+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(
-                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.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",
                 ),
-                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",
+                justify="between",
                 align="center",
+                width="100%",
             ),
-            justify="between",
-            align="center",
             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,
         ),
-        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",
+        api_key_setup_dialog(),
     )
 
 

{vedana_backoffice-0.6.1.dist-info → vedana_backoffice-0.6.3.dist-info}/METADATA

@@ -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.dist-info → vedana_backoffice-0.6.3.dist-info}/RECORD

@@ -3,7 +3,7 @@ vedana_backoffice/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 vedana_backoffice/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vedana_backoffice/start_services.py,sha256=94k2xrgIYXWhRfurR7XM-uMoXXL09RG0HFzyRRh9Nrs,974
 vedana_backoffice/state.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vedana_backoffice/ui.py,sha256=yKPAwUJZ_8cYzswqCV7nUxihvcrHw_DszeI1lbTaTbM,3618
+vedana_backoffice/ui.py,sha256=3Banuy-WT8U3sQOiQw2_UfWvkJjs_aUFDAyD4_c97mk,6463
 vedana_backoffice/util.py,sha256=RVN9ezjNkSnrlT3pBIrSvQQLyf-TZ2WzP_AYxRGks_Q,2209
 vedana_backoffice/vedana_backoffice.py,sha256=WlwF8hTBmkYOlgj9vE3k58MqDcj8FIbMNXZRsUXnjKg,1224
 vedana_backoffice/components/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -19,12 +19,12 @@ vedana_backoffice/pages/jims_thread_list_page.py,sha256=ZogjE7XQW2Zuyzcy4mod8qQE
 vedana_backoffice/pages/main_dashboard.py,sha256=odi4V5Bf-qukPLN9f8whgaYWikL6D7v5kzqB4rm3T10,21350
 vedana_backoffice/states/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vedana_backoffice/states/chat.py,sha256=xZwnqhzVIhnfGRPAWPcgZg9_boM3Crvmu-gieh1QMfs,13980
-vedana_backoffice/states/common.py,sha256=wI3ZCDeuV4_pkVQ-Q_Z0lgC8TQPs_eIT2nPqD-5rtjs,1977
+vedana_backoffice/states/common.py,sha256=ArFkK_A8yN0cpQtyK6CWSLj17dNxYbhN-ay59E_gYPs,3707
 vedana_backoffice/states/etl.py,sha256=1eLahWMm0Zd3oBDpOAJ_rfqHMPgRIJ_InwYzhm52hoo,68261
 vedana_backoffice/states/eval.py,sha256=nDfDL5ifL_q2vWWBEPwtWomUG_dhJxOMlIWOndOtjxM,76897
 vedana_backoffice/states/jims.py,sha256=gM0iZInD40-AebWY9gOzn5oJjAx7wyA_QliTluzMoZ8,19927
 vedana_backoffice/states/main_dashboard.py,sha256=9AcWo2nl5L-hTymxr-Stbu2s5nTGl9CVxTA9mHu8Rt4,32388
-vedana_backoffice-0.6.1.dist-info/METADATA,sha256=fWD47CYK9zRHd2_ej_NTcr5-_4cdtNrgbi_NeRqU8RI,341
-vedana_backoffice-0.6.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-vedana_backoffice-0.6.1.dist-info/entry_points.txt,sha256=WyUmp5EmDq_dB9RjzEim4L-XmDhPDxYI7ptab4A1EJc,87
-vedana_backoffice-0.6.1.dist-info/RECORD,,
+vedana_backoffice-0.6.3.dist-info/METADATA,sha256=uWKJRkWmGlBHkMpIJ9HtHuQ_iuyc2Ix6XR0jX2SZliw,341
+vedana_backoffice-0.6.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+vedana_backoffice-0.6.3.dist-info/entry_points.txt,sha256=WyUmp5EmDq_dB9RjzEim4L-XmDhPDxYI7ptab4A1EJc,87
+vedana_backoffice-0.6.3.dist-info/RECORD,,