figpack 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

figpack/__init__.py

@@ -2,6 +2,8 @@
 figpack - A Python package for creating shareable, interactive visualizations in the browser
 """
 
-__version__ = "0.1.0"
+__version__ = "0.1.4"
 __author__ = "Jeremy Magland"
 __email__ = "jmagland@flatironinstitute.org"
+
+from . import spike_sorting

figpack/cli.py

@@ -0,0 +1,312 @@
+"""
+Command-line interface for figpack
+"""
+
+import argparse
+import sys
+import pathlib
+import tempfile
+import tarfile
+import json
+import requests
+import threading
+import webbrowser
+import socket
+from concurrent.futures import ThreadPoolExecutor, as_completed
+from urllib.parse import urlparse, urljoin
+from typing import Dict, List, Tuple, Optional, Union
+from http.server import SimpleHTTPRequestHandler, ThreadingHTTPServer
+
+from . import __version__
+from .core._show_view import serve_files
+
+MAX_WORKERS_FOR_DOWNLOAD = 16
+
+
+def get_figure_base_url(figure_url: str) -> str:
+    """
+    Get the base URL from any figpack URL
+
+    Args:
+        figure_url: Any figpack URL (may or may not end with /index.html)
+
+    Returns:
+        str: The base URL for the figure directory
+    """
+    # Handle URLs that end with /index.html
+    if figure_url.endswith("/index.html"):
+        base_url = figure_url[:-11]  # Remove "/index.html"
+    elif figure_url.endswith("/"):
+        base_url = figure_url[:-1]  # Remove trailing slash
+    else:
+        # Assume it's already a directory URL
+        base_url = figure_url
+
+    # Ensure it ends with a slash for urljoin to work properly
+    if not base_url.endswith("/"):
+        base_url += "/"
+
+    return base_url
+
+
+def download_file(
+    base_url: str, file_info: Dict, temp_dir: pathlib.Path
+) -> Tuple[str, bool]:
+    """
+    Download a single file from the figure
+
+    Args:
+        base_url: The base URL for the figure
+        file_info: Dictionary with 'path' and 'size' keys
+        temp_dir: Temporary directory to download to
+
+    Returns:
+        Tuple of (file_path, success)
+    """
+    file_path = file_info["path"]
+    file_url = urljoin(base_url, file_path)
+
+    try:
+        response = requests.get(file_url, timeout=30)
+        response.raise_for_status()
+
+        # Create directory structure if needed
+        local_file_path = temp_dir / file_path
+        local_file_path.parent.mkdir(parents=True, exist_ok=True)
+
+        # Write file content
+        if file_path.endswith(
+            (
+                ".json",
+                ".html",
+                ".css",
+                ".js",
+                ".zattrs",
+                ".zgroup",
+                ".zarray",
+                ".zmetadata",
+            )
+        ):
+            # Text files
+            local_file_path.write_text(response.text, encoding="utf-8")
+        else:
+            # Binary files
+            local_file_path.write_bytes(response.content)
+
+        return file_path, True
+
+    except Exception as e:
+        print(f"Failed to download {file_path}: {e}")
+        return file_path, False
+
+
+def download_figure(figure_url: str, dest_path: str) -> None:
+    """
+    Download a figure from a figpack URL and save as tar.gz
+
+    Args:
+        figure_url: The figpack URL
+        dest_path: Destination path for the tar.gz file
+    """
+    print(f"Downloading figure from: {figure_url}")
+
+    # Get base URL
+    base_url = get_figure_base_url(figure_url)
+    print(f"Base URL: {base_url}")
+
+    # Check if manifest.json exists
+    manifest_url = urljoin(base_url, "manifest.json")
+    print("Checking for manifest.json...")
+
+    try:
+        response = requests.get(manifest_url, timeout=10)
+        response.raise_for_status()
+        manifest = response.json()
+        print(f"Found manifest with {len(manifest['files'])} files")
+    except requests.exceptions.RequestException as e:
+        print(f"Error: Could not retrieve manifest.json from {manifest_url}: {e}")
+        print("Make sure the URL points to a valid figpack figure with a manifest.")
+        sys.exit(1)
+    except json.JSONDecodeError as e:
+        print(f"Error: Invalid manifest.json format: {e}")
+        sys.exit(1)
+
+    # Create temporary directory for downloads
+    with tempfile.TemporaryDirectory() as temp_dir:
+        temp_path = pathlib.Path(temp_dir)
+
+        # Download all files in parallel
+        print(
+            f"Downloading {len(manifest['files'])} files with up to {MAX_WORKERS_FOR_DOWNLOAD} concurrent downloads..."
+        )
+
+        downloaded_count = 0
+        failed_files = []
+        count_lock = threading.Lock()
+
+        with ThreadPoolExecutor(max_workers=MAX_WORKERS_FOR_DOWNLOAD) as executor:
+            # Submit all download tasks
+            future_to_file = {
+                executor.submit(
+                    download_file, base_url, file_info, temp_path
+                ): file_info["path"]
+                for file_info in manifest["files"]
+            }
+
+            # Process completed downloads
+            for future in as_completed(future_to_file):
+                file_path = future_to_file[future]
+                try:
+                    downloaded_path, success = future.result()
+
+                    with count_lock:
+                        if success:
+                            downloaded_count += 1
+                            print(
+                                f"Downloaded {downloaded_count}/{len(manifest['files'])}: {downloaded_path}"
+                            )
+                        else:
+                            failed_files.append(downloaded_path)
+
+                except Exception as e:
+                    with count_lock:
+                        failed_files.append(file_path)
+                        print(f"Failed to download {file_path}: {e}")
+
+        if failed_files:
+            print(f"Warning: Failed to download {len(failed_files)} files:")
+            for failed_file in failed_files:
+                print(f"  - {failed_file}")
+
+            if len(failed_files) == len(manifest["files"]):
+                print("Error: Failed to download any files. Aborting.")
+                sys.exit(1)
+
+        # Save manifest.json to temp directory
+        manifest_path = temp_path / "manifest.json"
+        manifest_path.write_text(json.dumps(manifest, indent=2))
+        print("Added manifest.json to bundle")
+
+        # Create tar.gz file
+        print(f"Creating tar.gz archive: {dest_path}")
+        dest_pathlib = pathlib.Path(dest_path)
+        dest_pathlib.parent.mkdir(parents=True, exist_ok=True)
+
+        with tarfile.open(dest_path, "w:gz") as tar:
+            # Add all downloaded files (excluding figpack.json if it exists)
+            for file_path in temp_path.rglob("*"):
+                if file_path.is_file():
+                    arcname = file_path.relative_to(temp_path)
+                    # Skip figpack.json as requested
+                    if str(arcname) != "figpack.json":
+                        tar.add(file_path, arcname=arcname)
+
+        # Count files in archive (excluding directories)
+        archive_files = [
+            f for f in temp_path.rglob("*") if f.is_file() and f.name != "figpack.json"
+        ]
+        total_size = sum(f.stat().st_size for f in archive_files)
+
+        print(f"Archive created successfully!")
+        print(
+            f"Total files: {len(archive_files)} (including manifest.json, excluding figpack.json)"
+        )
+        print(f"Total size: {total_size / (1024 * 1024):.2f} MB")
+        print(f"Archive saved to: {dest_path}")
+
+
+def view_figure(archive_path: str, port: Union[int, None] = None) -> None:
+    """
+    Extract and serve a figure archive locally
+
+    Args:
+        archive_path: Path to the tar.gz archive
+        port: Optional port number to serve on
+    """
+    archive_pathlib = pathlib.Path(archive_path)
+
+    if not archive_pathlib.exists():
+        print(f"Error: Archive file not found: {archive_path}")
+        sys.exit(1)
+
+    if not archive_pathlib.suffix.lower() in [".gz", ".tgz"] or not str(
+        archive_pathlib
+    ).endswith(".tar.gz"):
+        print(f"Error: Expected a .tar.gz file, got: {archive_path}")
+        sys.exit(1)
+
+    print(f"Extracting figure archive: {archive_path}")
+
+    # Create temporary directory and extract files
+    with tempfile.TemporaryDirectory(prefix="figpack_view_") as temp_dir:
+        temp_path = pathlib.Path(temp_dir)
+
+        try:
+            with tarfile.open(archive_path, "r:gz") as tar:
+                tar.extractall(temp_path)
+
+            # Count extracted files
+            extracted_files = list(temp_path.rglob("*"))
+            file_count = len([f for f in extracted_files if f.is_file()])
+            print(f"Extracted {file_count} files")
+
+            # Check if index.html exists
+            index_html = temp_path / "index.html"
+            if not index_html.exists():
+                print("Warning: No index.html found in archive")
+                print("Available files:")
+                for f in sorted(extracted_files):
+                    if f.is_file():
+                        print(f"  {f.relative_to(temp_path)}")
+
+            # Serve the files
+            serve_files(
+                str(temp_path),
+                port=port,
+                open_in_browser=True,
+                allow_origin=None,
+            )
+
+        except tarfile.TarError as e:
+            print(f"Error: Failed to extract archive: {e}")
+            sys.exit(1)
+
+
+def main():
+    """Main CLI entry point"""
+    parser = argparse.ArgumentParser(
+        description="figpack - A Python package for creating shareable, interactive visualizations",
+        prog="figpack",
+    )
+    parser.add_argument("--version", action="version", version=f"figpack {__version__}")
+
+    subparsers = parser.add_subparsers(dest="command", help="Available commands")
+
+    # Download command
+    download_parser = subparsers.add_parser(
+        "download", help="Download a figure from a figpack URL"
+    )
+    download_parser.add_argument("figure_url", help="The figpack URL to download")
+    download_parser.add_argument("dest", help="Destination path for the tar.gz file")
+
+    # View command
+    view_parser = subparsers.add_parser(
+        "view", help="Extract and serve a figure archive locally"
+    )
+    view_parser.add_argument("archive", help="Path to the tar.gz archive file")
+    view_parser.add_argument(
+        "--port", type=int, help="Port number to serve on (default: auto-select)"
+    )
+
+    args = parser.parse_args()
+
+    if args.command == "download":
+        download_figure(args.figure_url, args.dest)
+    elif args.command == "view":
+        view_figure(args.archive, port=args.port)
+    else:
+        parser.print_help()
+
+
+if __name__ == "__main__":
+    main()

figpack/figpack-gui-dist/assets/{index-BA_v5Jep.css → index-BDa2iJW9.css}

@@ -1 +1 @@
-:root{font-family:system-ui,Avenir,Helvetica,Arial,sans-serif;line-height:1.5;font-weight:400;font-synthesis:none;text-rendering:optimizeLegibility;-webkit-font-smoothing:antialiased;-moz-osx-font-smoothing:grayscale}a{font-weight:500;color:#646cff;text-decoration:inherit}a:hover{color:#535bf2}body{margin:0;overflow:hidden}h1{font-size:3.2em;line-height:1.1}button{border-radius:8px;border:1px solid transparent;padding:.6em 1.2em;font-size:1em;font-weight:500;font-family:inherit;background-color:#1a1a1a;cursor:pointer;transition:border-color .25s}button:hover{border-color:#646cff}button:focus,button:focus-visible{outline:4px auto -webkit-focus-ring-color}.status-bar{position:fixed;bottom:0;left:0;right:0;height:30px;background-color:#2a2a2a;color:#fff;display:flex;align-items:center;padding:0 12px;font-size:12px;border-top:1px solid #444;z-index:1000}.status-bar.warning{background-color:#ff9800;color:#000}.status-bar.error{background-color:#f44336;color:#fff}.status-bar.expired{background-color:#d32f2f;color:#fff}.manage-button{background-color:#444;color:#fff;border:none;padding:4px 8px;border-radius:4px;cursor:pointer;font-size:11px;height:22px;margin-left:12px;transition:background-color .2s}.manage-button:hover{background-color:#666}.about-button{background-color:#444;color:#fff;border:none;padding:4px 8px;border-radius:4px;cursor:pointer;font-size:11px;height:22px;margin-left:12px;transition:background-color .2s}.about-button:hover{background-color:#666}.about-dialog-overlay{position:fixed;inset:0;background-color:#00000080;display:flex;justify-content:center;align-items:center;z-index:2000}.about-dialog{background-color:#2a2a2a;color:#fff;border-radius:8px;max-width:600px;max-height:80vh;width:90%;box-shadow:0 4px 20px #0000004d;overflow:hidden}.about-dialog-header{display:flex;justify-content:space-between;align-items:center;padding:16px 20px;border-bottom:1px solid #444}.about-dialog-header h2{margin:0;font-size:18px;font-weight:600}.about-dialog-close{background:none;border:none;color:#fff;font-size:24px;cursor:pointer;padding:0;width:30px;height:30px;display:flex;align-items:center;justify-content:center;border-radius:4px;transition:background-color .2s}.about-dialog-close:hover{background-color:#444}.about-dialog-content{padding:20px;overflow-y:auto;max-height:calc(80vh - 80px)}.about-dialog-description{line-height:1.6}.about-dialog-description h1,.about-dialog-description h2,.about-dialog-description h3{margin-top:0;margin-bottom:12px}.about-dialog-description p{margin-bottom:12px}.about-dialog-description code{background-color:#444;padding:2px 4px;border-radius:3px;font-family:Courier New,monospace;font-size:.9em}.about-dialog-description pre{background-color:#444;padding:12px;border-radius:4px;overflow-x:auto;margin-bottom:12px}.about-dialog-description ul,.about-dialog-description ol{margin-bottom:12px;padding-left:20px}.about-dialog-description blockquote{border-left:4px solid #666;padding-left:16px;margin:12px 0;font-style:italic}@media (prefers-color-scheme: light){:root{color:#213547;background-color:#fff}a:hover{color:#747bff}button{background-color:#f9f9f9}.status-bar{background-color:#f5f5f5;color:#333;border-top:1px solid #ddd}.status-bar.warning{background-color:#fff3cd;color:#856404;border-top:1px solid #ffeaa7}.status-bar.error,.status-bar.expired{background-color:#f8d7da;color:#721c24;border-top:1px solid #f5c6cb}.manage-button{background-color:#e0e0e0;color:#333;border:1px solid #ccc}.manage-button:hover{background-color:#d0d0d0}.about-button{background-color:#e0e0e0;color:#333;border:1px solid #ccc}.about-button:hover{background-color:#d0d0d0}.about-dialog{background-color:#fff;color:#333;box-shadow:0 4px 20px #00000026}.about-dialog-header{border-bottom:1px solid #ddd}.about-dialog-close{color:#333}.about-dialog-close:hover{background-color:#f0f0f0}.about-dialog-description code,.about-dialog-description pre{background-color:#f5f5f5;color:#333}.about-dialog-description blockquote{border-left:4px solid #ccc}}
+:root{font-family:system-ui,Avenir,Helvetica,Arial,sans-serif;line-height:1.5;font-weight:400;font-synthesis:none;text-rendering:optimizeLegibility;-webkit-font-smoothing:antialiased;-moz-osx-font-smoothing:grayscale}a{font-weight:500;color:#646cff;text-decoration:inherit}a:hover{color:#535bf2}body{margin:0;overflow:hidden}h1{font-size:3.2em;line-height:1.1}button{border-radius:8px;border:1px solid transparent;padding:.6em 1.2em;font-size:1em;font-weight:500;font-family:inherit;background-color:#1a1a1a;cursor:pointer;transition:border-color .25s}button:hover{border-color:#646cff}button:focus,button:focus-visible{outline:4px auto -webkit-focus-ring-color}.status-bar{position:fixed;bottom:0;left:0;right:0;height:30px;background-color:#2a2a2a;color:#fff;display:flex;align-items:center;padding:0 12px;font-size:12px;border-top:1px solid #444;z-index:1000}.status-bar.warning{background-color:#ff9800;color:#000}.status-bar.error{background-color:#f44336;color:#fff}.status-bar.expired{background-color:#d32f2f;color:#fff}.manage-button{background-color:#444;color:#fff;border:none;padding:4px 8px;border-radius:4px;cursor:pointer;font-size:11px;height:22px;margin-left:12px;transition:background-color .2s}.manage-button:hover{background-color:#666}.about-button{background-color:#444;color:#fff;border:none;padding:4px 8px;border-radius:4px;cursor:pointer;font-size:11px;height:22px;margin-left:12px;transition:background-color .2s}.about-button:hover{background-color:#666}.about-dialog-overlay{position:fixed;inset:0;background-color:#00000080;display:flex;justify-content:center;align-items:center;z-index:2000}.about-dialog{background-color:#2a2a2a;color:#fff;border-radius:8px;max-width:600px;max-height:80vh;width:90%;box-shadow:0 4px 20px #0000004d;overflow:hidden}.about-dialog-header{display:flex;justify-content:space-between;align-items:center;padding:16px 20px;border-bottom:1px solid #444}.about-dialog-header h2{margin:0;font-size:18px;font-weight:600}.about-dialog-close{background:none;border:none;color:#fff;font-size:24px;cursor:pointer;padding:0;width:30px;height:30px;display:flex;align-items:center;justify-content:center;border-radius:4px;transition:background-color .2s}.about-dialog-close:hover{background-color:#444}.about-dialog-content{padding:20px;overflow-y:auto;max-height:calc(80vh - 80px)}.about-dialog-description{line-height:1.6}.about-dialog-description h1,.about-dialog-description h2,.about-dialog-description h3{margin-top:0;margin-bottom:12px}.about-dialog-description p{margin-bottom:12px}.about-dialog-description code{background-color:#444;padding:2px 4px;border-radius:3px;font-family:Courier New,monospace;font-size:.9em}.about-dialog-description pre{background-color:#444;padding:12px;border-radius:4px;overflow-x:auto;margin-bottom:12px}.about-dialog-description ul,.about-dialog-description ol{margin-bottom:12px;padding-left:20px}.about-dialog-description blockquote{border-left:4px solid #666;padding-left:16px;margin:12px 0;font-style:italic}@media (prefers-color-scheme: light){:root{color:#213547;background-color:#fff}a:hover{color:#747bff}button{background-color:#f9f9f9}.status-bar{background-color:#f5f5f5;color:#333;border-top:1px solid #ddd}.status-bar.warning{background-color:#fff3cd;color:#856404;border-top:1px solid #ffeaa7}.status-bar.error,.status-bar.expired{background-color:#f8d7da;color:#721c24;border-top:1px solid #f5c6cb}.manage-button{background-color:#e0e0e0;color:#333;border:1px solid #ccc}.manage-button:hover{background-color:#d0d0d0}.about-button{background-color:#e0e0e0;color:#333;border:1px solid #ccc}.about-button:hover{background-color:#d0d0d0}.about-dialog{background-color:#fff;color:#333;box-shadow:0 4px 20px #00000026}.about-dialog-header{border-bottom:1px solid #ddd}.about-dialog-close{color:#333}.about-dialog-close:hover{background-color:#f0f0f0}.about-dialog-description code,.about-dialog-description pre{background-color:#f5f5f5;color:#333}.about-dialog-description blockquote{border-left:4px solid #ccc}}.VerticalToolbar{overflow:hidden;position:absolute}.HorizontalToolbar{display:flex;justify-content:center;align-items:center;overflow:hidden;margin:auto}.SortableTableWidget .MuiTableCell-root{padding:1px 0;font-size:12px;border:solid 1px #f0f0f0}.SortableTableWidget .MuiTableRow-root.selectedRow{background-color:#b5d1ff}.SortableTableWidget .MuiTableRow-root.currentRow{background-color:#95b1df}.SortCaretSpan{font-size:16px;color:gray;padding-left:3px;padding-right:5px;padding-top:2px}.SortableTableWidget .unitLabel{width:10px;height:10px;position:relative;display:inline-block}.SortableTableWidget .SortableTableCheckbox{padding:1px}.unitEntryBase{padding-right:13px;padding-top:10px}.unselectedUnitEntry{border:1px transparent solid;font-weight:400;color:#000;background-color:#fff}.selectedUnitEntry{border:1px solid blue;font-weight:700;color:#000;background-color:#b5d1ff}.unitIdsColumnLabelStyle{min-width:200px;font-weight:700;padding:7px 5px}.unitLabelsStyle{padding-right:3px;color:#333}.plotUnitLabel{text-align:left;margin-left:20px;margin-top:5px;font-weight:700}.plotWrapperButtonParent{display:inline-block}.plotWrapperStyleButton{padding-top:50px}.plotUnselectedStyle{border:3px solid rgb(202,224,231)}.plotSelectedStyle{border:3px solid #9999ff;background-color:#fff}.plotUnselectableStyle{border:3px solid transparent}.plotCurrentStyle{border:3px solid blue;background-color:#fff}.plotLabelStyle{font-weight:700;text-align:center;overflow:hidden;white-space:nowrap}