Update to later versions

Enviroment variables simplification

Dockerfile

@@ -1,8 +1,12 @@
-FROM python:3.11-slim
+FROM ghcr.io/astral-sh/uv:0.11.5-python3.12-trixie-slim
+# Using the uv image
 
 WORKDIR /app
 
-RUN pip install uv
+# Setting up dependencies and stuff to be sure compilation is also working
+# RUN apt-get update \
+#  && apt-get install -y --no-install-recommends build-essential python3-dev \
+#  && rm -rf /var/lib/apt/lists/*
 
 COPY pyproject.toml uv.lock ./
 RUN uv sync --frozen --no-dev
@@ -14,4 +18,6 @@ COPY utils/ ./utils/
 
 EXPOSE 8501
 
-CMD ["sh", "-c", "uv run streamlit run app.py --server.port=${PORT:-8501} --server.address=0.0.0.0"]
+ENV PATH="/app/.venv/bin:$PATH"
+
+CMD ["sh", "-c", "streamlit run app.py --server.port=${PORT:-8501} --server.address=0.0.0.0"]

README.md

@@ -96,7 +96,8 @@ No secrets needed! The dependencies are published on PyPI.
 
 ```bash
 docker build -t persona-ui .
-docker run -p 8501:8501 persona-ui
+# Specify your local .env to have things working as expectd
+docker run --env-file .env --rm -p 8501:8501 persona-ui
 ```
 
 ## Configuration
@@ -111,9 +112,6 @@ ARTIFACTS_DIR=...      # Optional: where activations are read from (default: ./a
 
 The app picks up this file automatically via `load_dotenv()` on startup.
 
-You can also override the active NDIF or Hugging Face token from the sidebar
-`API Keys` section. Those inputs only apply for the current session.
-
 ## Saved Artifacts
 
 The Compare and Extract tabs read from / write to:

app.py

@@ -8,33 +8,12 @@ from utils.helpers import DATASET_SOURCES
 load_dotenv()
 DEFAULT_MODEL = os.environ.get("DEFAULT_MODEL", "google/gemma-2-2b-it")
 REMOTE_DEFAULT_MODEL = os.environ.get("REMOTE_DEFAULT_MODEL", "google/gemma-2-9b-it")
-NDIF_API_KEY = os.environ.get("NDIF_API_KEY", "")
 
 
 _TABS = ["Chat", "Compare", "Extract"]
 _TAB_ICONS = [":material/chat:", ":material/search:", ":material/tune:"]
 
 
-def _sync_sidebar_api_key(env_var: str, value: str) -> None:
-    if value:
-        os.environ[env_var] = value
-
-
-def _sidebar_api_keys() -> None:
-    with st.sidebar:
-        st.divider()
-        st.caption("API Keys")
-
-        ndif_api_key = st.text_input(
-            "NDIF API key",
-            value=NDIF_API_KEY,
-            type="password",
-            key="sidebar__ndif_api_key",
-            help="Overrides NDIF_API_KEY for this session.",
-        )
-        _sync_sidebar_api_key("NDIF_API_KEY", ndif_api_key)
-
-
 def _sidebar_controls() -> tuple[bool, str, str, str]:
     from utils.runtime import list_remote_models
 
@@ -96,8 +75,6 @@ def _sidebar_controls() -> tuple[bool, str, str, str]:
             help="Dataset for Chat and Extract.",
         )
 
-    _sidebar_api_keys()
-
     return remote, model_name, dataset_source, active_tab
 
 

pyproject.toml

@@ -3,7 +3,7 @@ name = "persona-ui"
 version = "0.1.0"
 description = "Streamlit UI for persona-vectors"
 readme = "README.md"
-requires-python = ">=3.10"
+requires-python = ">=3.12"
 dependencies = [
     "persona-vectors>=0.1.0",
     "persona-data>=0.1.0",
@@ -13,7 +13,6 @@ dependencies = [
     "transformers>=5.5.0",
 ]
 
-
 [tool.uv.sources]
 # Local development:
 # persona-vectors = { path = "../persona-data", editable = true }