anywidget-vector 0.2.8__tar.gz → 0.3.1__tar.gz

anywidget_vector-0.3.1/CHANGELOG.md

@@ -0,0 +1,129 @@
+# Changelog
+
+## 0.3.1 (2026-03-16)
+
+### New Features
+
+- **Lasso selection**: New lasso mode for freehand point selection in the 3D canvas, with point-in-polygon hit testing
+- **Selection highlighting**: Selected points show wireframe highlight rings and non-selected points are dimmed to 15% brightness
+- **Auto-sizing**: Width defaults to 100% (fill container), height defaults to 1:1 aspect ratio. Both can still be set to fixed pixel values
+
+### Bug Fixes
+
+- **Properties panel**: Panel is now always created (previously skipped when `show_properties=False`), so the toolbar button always works. Starts closed unless `show_properties=True`
+- **Distance metric feedback**: Changing the metric now auto-enables k=5 connections when a point is selected, giving immediate visual feedback
+
+### Improvements
+
+- Reduced scroll zoom sensitivity (`zoomSpeed` halved to 0.5)
+- Search filter dimming increased from 12% to 6% opacity for clearer visual contrast
+- Camera no longer resets on every point update (only on initial load)
+- Cross-widget filtering: example updated so VectorSpace lasso selection feeds back into ParallelCoordinates dimming via marimo reactivity
+
+## 0.3.0 (2026-03-16)
+
+### New Features
+
+- **Distance metrics sidebar**: Distance section with metric selector (euclidean, cosine, manhattan, dot product), configurable k-neighbors (0-50), and live nearest-neighbor display for the selected point
+- **Dimension mapping display**: Sidebar now shows active visual mapping fields: color field, size range, shape field and unique shape count
+
+### Improvements
+
+- Improved sidebar styling and layout
+
+## 0.2.8 (2026-03-15)
+
+### New Features
+
+- **6D parallel coordinates example**: Fashion-MNIST demo with PCA across 6+ dimensions, interactive axis mapping, and shape mapping to categorical labels
+
+## 0.2.7 (2026-03-15)
+
+### New Features
+
+- **Dark/light mode toggle**: `dark_mode` trait with real-time CSS color scheme switching and theme-aware styling across all panels
+
+## 0.2.6 (2026-03-15)
+
+### New Features
+
+- **NumPy array support**: `add_numpy()` method for adding points from arrays with automatic PCA projection for high-dimensional data (D > 3)
+- **Dimensionality reduction**: `project()` method with PCA, t-SNE and UMAP support; `set_vectors()` for storing high-dimensional vectors separately
+- **Example scripts**: Projection demo and numpy integration examples
+
+### Improvements
+
+- High-dimensional vector storage without syncing to JavaScript
+- Coordinate normalization after projection
+- Added demo screenshot to README
+
+## 0.2.5 (2026-03-04)
+
+### Bug Fixes
+
+- Fixed lasso selection stability issues
+- Removed problematic custom traitlets implementation
+
+### Improvements
+
+- Added keyboard shortcuts: F to fit view, Escape to clear selection
+- Better visual feedback for selection modes
+- Cleaner toolbar layout with integrated mode controls
+
+## 0.2.4 (2026-03-02)
+
+### New Features
+
+- **Zoom and pan controls**: Zoom in, zoom out and fit-to-view buttons in the toolbar
+- **Box selection mode**: New selection mode alongside click and multi-select, with visual feedback during selection
+- **HTML export**: `to_html()` and `save_html()` methods for self-contained HTML output with embedded data and Three.js
+
+### Improvements
+
+- Improved camera fitting algorithm for better initial view framing
+- Updated README with comprehensive usage examples
+- Updated dependencies
+
+## 0.2.3 (2026-02-12)
+
+### New Features
+
+- **Sidebar explorer**: Three collapsible sections: collections browser, dimensions explorer (point count, dimensionality, axis ranges) and clusters explorer (grouped by color field with sizes)
+- **Browser-side Grafeo backend**: JavaScript client for Grafeo server and WASM-embedded queries
+- **Demo module**: Reference implementation showing various widget features
+
+### Improvements
+
+- Consistency pass across all backend modules
+- Ruff configuration tuned for marimo compatibility
+
+## 0.2.2 (2026-02-07)
+
+### New Features
+
+- **Streaming factory methods**: `from_numpy()`, `from_umap()` and `from_arrays()` constructors for various data formats
+
+### Improvements
+
+- Added Apache 2.0 license
+- Expanded test suite with 194 additional test cases
+- ESM bundle aggregation for widget JavaScript
+
+## 0.2.1 (2026-02-03)
+
+### New Features
+
+- **Multi-backend architecture**: Six backends: Qdrant, Pinecone, Weaviate (browser-side REST), Chroma, LanceDB, Grafeo (Python-side)
+- **Modular UI**: Separated rendering into canvas, settings, properties and toolbar components
+- **Backend factory methods**: `from_qdrant()`, `from_pinecone()`, `from_weaviate()`, `from_chroma()`, `from_lancedb()`, `from_grafeo()`
+- **Event system**: `on_click()`, `on_hover()` and `on_selection()` decorators with callback signatures
+- **CI/CD**: GitHub Actions pipelines for testing and PyPI publishing
+
+### Improvements
+
+- Refactored monolithic widget into modular component architecture
+- Added README with usage examples
+
+## 0.1.0 (2026-02-03)
+
+- Initial release: Three.js-based interactive 3D vector visualization with instanced rendering, six shape types, continuous and categorical color mapping, orbit controls, hover tooltips, axes/grid display and raycast-based selection

{anywidget_vector-0.2.8 → anywidget_vector-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: anywidget-vector
-Version: 0.2.8
+Version: 0.3.1
 Summary: Interactive 3D vector visualization with query UI for vector databases
 Project-URL: Homepage, https://grafeo.dev/
 Project-URL: Repository, https://github.com/GrafeoDB/anywidget-vector
@@ -33,10 +33,10 @@ Requires-Dist: chromadb>=0.4; extra == 'chroma'
 Provides-Extra: demo
 Provides-Extra: dev
 Requires-Dist: marimo>=0.20; extra == 'dev'
-Requires-Dist: prek>=0.3.2; extra == 'dev'
+Requires-Dist: prek>=0.3; extra == 'dev'
 Requires-Dist: pytest>=9.0.2; extra == 'dev'
 Requires-Dist: ruff>=0.15; extra == 'dev'
-Requires-Dist: ty>=0.0.20; extra == 'dev'
+Requires-Dist: ty>=0.0; extra == 'dev'
 Provides-Extra: embed
 Provides-Extra: grafeo
 Requires-Dist: grafeo>=0.5; extra == 'grafeo'

{anywidget_vector-0.2.8 → anywidget_vector-0.3.1}/examples/fashion-mnist-parallel-coords-6d.py

@@ -1,3 +1,8 @@
+# Based on the Fashion MNIST Parallel Coordinates notebook by the marimo team:
+# https://github.com/marimo-team/gallery-examples/blob/main/notebooks/wigglystuff/fashion-mnist-parallel-coords.py
+#
+# Extended with anywidget-vector 3D scatter view and 6D dimension mapping.
+#
 # /// script
 # requires-python = ">=3.12"
 # dependencies = [
@@ -8,7 +13,7 @@
 #     "wigglystuff==0.2.37",
 #     "matplotlib==3.10.8",
 #     "pandas==3.0.1",
-#     "anywidget-vector==0.2.7",
+#     "anywidget-vector==0.3.1",
 # ]
 # ///
 
@@ -29,16 +34,15 @@ def _():
     from wigglystuff import ParallelCoordinates
 
     from anywidget_vector import VectorSpace
-
     return PCA, ParallelCoordinates, VectorSpace, fetch_openml, mo, np, pl, plt
 
 
 @app.cell(hide_code=True)
 def _(mo):
     mo.md(r"""
-    # Fashion MNIST: Parallel Coordinates + 3D Vector View
+    # Fashion MNIST: Parallel Coordinates + 6D Vector View
 
-    Brush the parallel coordinates axes to filter. The 3D scatter updates in real time.
+    Brush the parallel coordinates axes to filter. The 3D (xyz)+3D(color/size/shape) scatter updates in real time.
     Use the dropdowns to map PCA dimensions to visual channels.
     """)
     return
@@ -62,18 +66,9 @@ def _(fetch_openml, np):
         8: "Bag",
         9: "Ankle boot",
     }
-    LABEL_COLORS = {
-        "T-shirt/top": "#6366f1",
-        "Trouser": "#f59e0b",
-        "Pullover": "#10b981",
-        "Dress": "#ef4444",
-        "Coat": "#8b5cf6",
-        "Sandal": "#06b6d4",
-        "Shirt": "#f97316",
-        "Sneaker": "#84cc16",
-        "Bag": "#ec4899",
-        "Ankle boot": "#14b8a6",
-    }
+    # d3 schemeCategory10, assigned alphabetically to match wigglystuff
+    _d3 = ["#1f77b4", "#ff7f0e", "#2ca02c", "#d62728", "#9467bd", "#8c564b", "#e377c2", "#7f7f7f", "#bcbd22", "#17becf"]
+    LABEL_COLORS = {_name: _d3[_i] for _i, _name in enumerate(sorted(label_names.values()))}
     SHAPE_NAMES = ["sphere", "cube", "cone", "octahedron", "cylinder", "tetrahedron"]
     return LABEL_COLORS, SHAPE_NAMES, images, label_names, labels
 
@@ -81,7 +76,7 @@ def _(fetch_openml, np):
 @app.cell
 def _(mo):
     n_samples_slider = mo.ui.slider(start=500, stop=5000, step=500, value=2000, label="Samples")
-    n_components_slider = mo.ui.slider(start=3, stop=15, step=1, value=8, label="PCA dims")
+    n_components_slider = mo.ui.slider(start=3, stop=15, step=1, value=5, label="PCA dims")
     mo.hstack([n_samples_slider, n_components_slider])
     return n_components_slider, n_samples_slider
 
@@ -110,8 +105,8 @@ def _(
 
 
 @app.cell
-def _(ParallelCoordinates, df, mo):
-    widget = mo.ui.anywidget(ParallelCoordinates(df, color_by="label"))
+def _(LABEL_COLORS, ParallelCoordinates, df, mo):
+    widget = mo.ui.anywidget(ParallelCoordinates(df, color_by="label", color_map=LABEL_COLORS))
     widget
     return (widget,)
 
@@ -122,9 +117,9 @@ def _(mo, n_components_slider):
     x_dim = mo.ui.dropdown(options=pcs, value="PC1", label="X")
     y_dim = mo.ui.dropdown(options=pcs, value="PC2", label="Y")
     z_dim = mo.ui.dropdown(options=pcs, value="PC3", label="Z")
-    size_dim = mo.ui.dropdown(options=["none", *pcs], value="none", label="Size")
+    size_dim = mo.ui.dropdown(options=["none", *pcs], value="PC4", label="Size")
     color_dim = mo.ui.dropdown(options=["label", *pcs], value="label", label="Color")
-    shape_dim = mo.ui.dropdown(options=["label", "none"], value="none", label="Shape")
+    shape_dim = mo.ui.dropdown(options=["label", "none", *pcs], value="PC5", label="Shape")
     mo.hstack([x_dim, y_dim, z_dim, size_dim, color_dim, shape_dim], gap=0.5)
     return color_dim, shape_dim, size_dim, x_dim, y_dim, z_dim
 
@@ -140,6 +135,7 @@ def _(
     label_names,
     labels,
     mo,
+    np,
     shape_dim,
     size_dim,
     x_dim,
@@ -152,8 +148,23 @@ def _(
     _xi, _yi, _zi = _pc(x_dim.value), _pc(y_dim.value), _pc(z_dim.value)
     _color_pc = None if color_dim.value == "label" else _pc(color_dim.value)
     _size_pc = None if size_dim.value == "none" else _pc(size_dim.value)
+    _shape_mode = shape_dim.value  # "label", "none", or a PC name
+
+    # Shape mapping: label-based or PC-based (binned into 6 shape buckets)
     _unique_labels = sorted(set(label_names.values()))
     _shape_map = {_n: SHAPE_NAMES[_j % len(SHAPE_NAMES)] for _j, _n in enumerate(_unique_labels)}
+    _shape_bins = None
+    if _shape_mode not in ("label", "none"):
+        _shape_col = components[:, _pc(_shape_mode)]
+        _quantiles = np.percentile(_shape_col, np.linspace(0, 100, len(SHAPE_NAMES) + 1))
+        _shape_bins = list(zip(_quantiles[:-1], _quantiles[1:], SHAPE_NAMES, strict=False))
+        _shape_map = {_s: _s for _s in SHAPE_NAMES}
+
+    def _get_shape_bin(_val):
+        for _lo, _hi, _s in _shape_bins:
+            if _val <= _hi:
+                return _s
+        return SHAPE_NAMES[-1]
 
     vs_points = []
     for _i in range(len(components)):
@@ -171,8 +182,10 @@ def _(
             _p["color_val"] = float(components[_i, _color_pc])
         if _size_pc is not None:
             _p["size_val"] = float(components[_i, _size_pc])
-        if shape_dim.value == "label":
+        if _shape_mode == "label":
             _p["shape_cat"] = _name
+        elif _shape_bins is not None:
+            _p["shape_cat"] = _get_shape_bin(components[_i, _pc(_shape_mode)])
         vs_points.append(_p)
 
     _vs_kwargs = {
@@ -183,17 +196,14 @@ def _(
         _vs_kwargs["color_scale"] = "viridis"
     if _size_pc is not None:
         _vs_kwargs["size_field"] = "size_val"
-        _vs_kwargs["size_range"] = [0.01, 0.06]
-    if shape_dim.value == "label":
+        _vs_kwargs["size_range"] = [0.01, 0.03]
+    if _shape_mode != "none":
         _vs_kwargs["shape_field"] = "shape_cat"
         _vs_kwargs["shape_map"] = _shape_map
 
     vs_widget = VectorSpace(
         points=vs_points,
-        width=1200,
-        height=500,
-        dark_mode=False,
-        background="#fafafa",
+        height=900,
         show_toolbar=True,
         show_settings=True,
         show_properties=False,
@@ -206,36 +216,73 @@ def _(
 
 @app.cell
 def _(LABEL_COLORS, color_dim, mo, vs_points, vs_widget, widget):
-    _filtered = set(widget.widget.filtered_indices)
-    _dim_color = "#d1d5db"
+    # ParallelCoordinates -> VectorSpace (one-way)
+    # Uses vs_widget (raw) to avoid triggering vs-dependent cells
+    _par_selected = list(widget.widget.selected_indices or [])
+    _par_filtered = set(widget.widget.filtered_indices)
+    _active_set = set(_par_selected) if _par_selected else _par_filtered
     _updated = []
     for _i, _p in enumerate(vs_points):
+        if _i not in _active_set:
+            continue
         _new_p = dict(_p)
-        if _i in _filtered:
-            if color_dim.value == "label":
-                _new_p["color"] = LABEL_COLORS[_p["label"]]
-            else:
-                _new_p.pop("color", None)
+        if color_dim.value == "label":
+            _new_p["color"] = LABEL_COLORS[_p["label"]]
         else:
-            _new_p["color"] = _dim_color
+            _new_p.pop("color", None)
         _updated.append(_new_p)
     vs_widget.points = _updated
-    mo.md(f"**{len(_filtered)}** / {len(vs_points)} points selected")
+    mo.md(f"**{len(_updated)}** / {len(vs_points)} points selected")
+    return
+
+
+@app.cell
+def _(LABEL_COLORS, ParallelCoordinates, df, mo, vs):
+    # VectorSpace -> filtered ParallelCoordinates (reactive via vs.value)
+    _selected = list(vs.widget.selected_points or [])
+    if _selected:
+        _indices = sorted(int(sid.split("_")[1]) for sid in _selected)
+        _filtered_df = df[_indices]
+        _par = ParallelCoordinates(_filtered_df, color_by="label", color_map=LABEL_COLORS)
+        mo.vstack([
+            mo.md(f"**{len(_indices)}** points selected in 3D view"),
+            mo.ui.anywidget(_par),
+        ])
+    else:
+        mo.md("*Lasso or box-select points in the 3D view to filter*")
     return
 
 
 @app.cell
-def _(idx, images, label_names, labels, np, plt, widget):
+def _(LABEL_COLORS, idx, images, label_names, labels, np, plt, vs, widget):
+    _selected_ids = set(vs.widget.selected_points or [])
     _filtered = widget.widget.filtered_indices
-    _sample_idx = np.array(_filtered[:10]) if len(_filtered) >= 10 else np.array(_filtered)
 
-    _fig, _axes = plt.subplots(1, len(_sample_idx), figsize=(2 * len(_sample_idx), 2))
+    if _selected_ids:
+        _show = [
+            _i for _i in range(len(idx))
+            if f"p_{_i}" in _selected_ids
+        ][:10]
+    else:
+        _show = list(_filtered[:10])
+
+    _sample_idx = np.array(_show) if len(_show) > 0 else np.array([0])
+
+    _fig, _axes = plt.subplots(1, len(_sample_idx), figsize=(2 * len(_sample_idx), 2.4))
     if len(_sample_idx) == 1:
         _axes = [_axes]
     for _ax, _si in zip(_axes, _sample_idx, strict=False):
+        _name = label_names[labels[idx[_si]]]
         _ax.imshow(images[idx[_si]].reshape(28, 28), cmap="gray")
-        _ax.set_title(label_names[labels[idx[_si]]], fontsize=9)
+        _ax.set_title(_name, fontsize=9)
         _ax.axis("off")
+        if f"p_{_si}" in _selected_ids:
+            _color = LABEL_COLORS.get(_name, "#0880ea")
+            for _spine in _ax.spines.values():
+                _spine.set_visible(True)
+                _spine.set_color(_color)
+                _spine.set_linewidth(3)
+            _ax.set_title(_name, fontsize=9, fontweight="bold", color=_color)
     plt.tight_layout()
     _fig
     return

{anywidget_vector-0.2.8 → anywidget_vector-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "anywidget-vector"
-version = "0.2.8"
+version = "0.3.1"
 description = "Interactive 3D vector visualization with query UI for vector databases"
 readme = "README.md"
 license = "Apache-2.0"
@@ -26,10 +26,10 @@ dependencies = [
 
 [project.optional-dependencies]
 dev = [
-    "prek>=0.3.2",
+    "prek>=0.3",
     "pytest>=9.0.2",
     "ruff>=0.15",
-    "ty>=0.0.20",
+    "ty>=0.0",
     "marimo>=0.20",
 ]
 pandas = ["pandas>=2.0"]

{anywidget_vector-0.2.8 → anywidget_vector-0.3.1}/src/anywidget_vector/static/icons.js

@@ -16,4 +16,5 @@ export const ICONS = {
   zoomFit: '<svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2"><path d="M15 3h6v6M9 21H3v-6M21 3l-7 7M3 21l7-7"/></svg>',
   cursor: '<svg viewBox="0 0 24 24" fill="currentColor" stroke="none"><path d="M4 2l12 14h-5l-2.3 5.5L6 20.3 8.3 15H4V2z"/></svg>',
   boxSelect: '<svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2" stroke-dasharray="3 2"><rect x="3" y="3" width="18" height="18" rx="1"/></svg>',
+  lasso: '<svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2"><ellipse cx="12" cy="10" rx="9" ry="7"/><path d="M12 17v3M12 20l-2 2"/></svg>',
 };

{anywidget_vector-0.2.8 → anywidget_vector-0.3.1}/src/anywidget_vector/ui/__init__.py

@@ -122,16 +122,55 @@ async function executeBackendQuery(model) {{
 function render({{ model, el }}) {{
   const wrapper = document.createElement("div");
   wrapper.className = "avs-wrapper";
-  if (model.get("dark_mode")) wrapper.classList.add("avs-dark");
-  wrapper.style.width = model.get("width") + "px";
-  wrapper.style.height = model.get("height") + "px";
+
+  const _w = model.get("width");
+  const _h = model.get("height");
+  wrapper.style.width = _w > 0 ? _w + "px" : "100%";
+  if (_h > 0) {{
+    wrapper.style.height = _h + "px";
+  }} else {{
+    wrapper.style.aspectRatio = "1 / 1";
+  }}
+
   el.appendChild(wrapper);
 
-  model.on("change:dark_mode", () => {{
-    const dark = model.get("dark_mode");
+  // Auto-detect host theme (marimo uses Tailwind class="dark" on <html>)
+  function detectHostDark() {{
+    const html = document.documentElement;
+    if (html.classList.contains("dark")) return true;
+    if (html.dataset.theme === "dark") return true;
+    if (window.matchMedia && window.matchMedia("(prefers-color-scheme: dark)").matches) return true;
+    return false;
+  }}
+
+  function applyTheme(dark) {{
     wrapper.classList.toggle("avs-dark", dark);
-    model.set("background", dark ? "#1a1a2e" : "#fafafa");
+    model.set("background", dark ? "#181c1a" : "#ffffff");
     model.save_changes();
+  }}
+
+  // Detect if we're inside a themed host (marimo, jupyter, etc.)
+  const hasHostTheme = !!el.getRootNode()?.host?.tagName?.startsWith("MARIMO-");
+  if (hasHostTheme) {{
+    wrapper.classList.add("avs-auto-theme");
+    model.set("dark_mode", detectHostDark());
+    applyTheme(detectHostDark());
+    // Watch host for live theme changes
+    const themeObserver = new MutationObserver(() => {{
+      const dark = detectHostDark();
+      if (model.get("dark_mode") !== dark) {{
+        model.set("dark_mode", dark);
+        applyTheme(dark);
+      }}
+    }});
+    themeObserver.observe(document.documentElement, {{ attributes: true, attributeFilter: ["class", "data-theme"] }});
+  }} else {{
+    applyTheme(model.get("dark_mode"));
+  }}
+
+  // Manual toggle from settings panel still works
+  model.on("change:dark_mode", () => {{
+    applyTheme(model.get("dark_mode"));
   }});
 
   let sidebar = null;
@@ -144,7 +183,7 @@ function render({{ model, el }}) {{
       onRunQuery: () => runQuery(),
       onFilterInput: (text) => {{
         if (canvas) {{
-          const result = canvas.applyFilter(text);
+          const result = canvas.applyFilter(text, false);
           if (toolbarUI) toolbarUI.setFilterCount(result.matched, result.total);
         }}
       }},
@@ -184,21 +223,22 @@ function render({{ model, el }}) {{
     main.appendChild(settingsPanel.element);
   }}
 
-  if (model.get("show_properties")) {{
-    propertiesPanel = createPropertiesPanel(model, {{
-      onClose: () => propertiesPanel?.close(),
-    }});
-    main.appendChild(propertiesPanel.element);
-  }}
+  propertiesPanel = createPropertiesPanel(model, {{
+    onClose: () => propertiesPanel?.close(),
+  }});
+  main.appendChild(propertiesPanel.element);
+  if (model.get("show_properties")) propertiesPanel.open();
 
-  const canvas = createCanvas(model, canvasContainer, {{}});
+  const canvas = createCanvas(model, canvasContainer, {{
+    onHover: (point) => propertiesPanel?.setHoveredPoint(point),
+  }});
 
   async function runQuery() {{
     const query = model.get("query_input") || "";
 
-    // Always apply client-side filter
+    // Apply hard client-side filter (completely hide non-matching)
     if (canvas) {{
-      const result = canvas.applyFilter(query);
+      const result = canvas.applyFilter(query, true);
       if (toolbarUI) toolbarUI.setFilterCount(result.matched, result.total);
     }}