ai_dandi_plot

AI-assisted plotting tool for DANDI datasets.

Installation

pip install -e .

Usage

Command-line usage:

ai-dandi-plot create-plot [OPTIONS]

Options:
  --dandiset-id TEXT            DANDI dataset ID
  --special-instructions TEXT   Special instructions for plot creation
  --completion-model TEXT       LLM model for completions (default: google/gemini-2.0-flash-001)
  --vision-model TEXT          LLM model for vision analysis (default: openai/gpt-4o)
  --output-dir TEXT           Output directory for the figure
  --help                       Show this message and exit

Python usage:

from ai_dandi_plot import create_plot

plot_path = create_plot(
    dandiset_id="123456",
    special_instructions="Plot the spike raster",
    completion_model="google/gemini-2.0-flash-001",
    vision_model="openai/gpt-4o",
    output_dir="output"
)