# Figure Generation

> Generate publication-quality scientific figures using matplotlib/seaborn with a three-phase pipeline (query expansion, code generation with execution, VLM visual feedback). Handles bar charts, line plots, heatmaps, training curves, ablation plots, and more. Use when the user needs figures, plots, or visualizations for a paper.

- **Type:** Skill
- **Install:** `agentstack add skill-aravindan20-claude-research-paper-os-figure-generation`
- **Verified:** Yes — security-reviewed for prompt injection and unsafe behavior
- **Seller:** [ARAVINDAN20](https://agentstack.voostack.com/s/aravindan20)
- **Installs:** 0
- **Category:** [Agent Skills](https://agentstack.voostack.com/c/agent-skills)
- **Latest version:** 0.1.0
- **License:** Apache-2.0
- **Upstream author:** [ARAVINDAN20](https://github.com/ARAVINDAN20)
- **Source:** https://github.com/ARAVINDAN20/Claude-Research-Paper-OS/tree/main/.claude/skills/agent-research-skills/skills/figure-generation

## Install

```sh
agentstack add skill-aravindan20-claude-research-paper-os-figure-generation
```

Requires the [AgentStack CLI](https://agentstack.voostack.com/docs/cli). Works with Claude Code, Cursor, and any MCP-compatible agent.

## About

# Scientific Figure Generation

Generate publication-quality figures for research papers.

## Input

- `$0` — Description of the desired figure
- `$1` — (Optional) Path to data file (CSV, JSON, NPY, PKL) or results directory

## Scripts

### Generate figure template
```bash
python ~/.claude/skills/figure-generation/scripts/figure_template.py --type bar --output figure_script.py --name comparison
python ~/.claude/skills/figure-generation/scripts/figure_template.py --list-types
```

Available types: `bar`, `training-curve`, `heatmap`, `ablation`, `line`, `scatter`, `radar`, `violin`, `tsne`, `attention`

## Three-Phase Pipeline (from MatPlotAgent)

### Phase 1: Query Expansion
Expand the user's figure description into step-by-step coding specifications using the prompts in `references/figure-prompts.md`. Determine: figure type, data mapping (x/y/color/hue), style requirements, paper conventions.

### Phase 2: Code Generation with Execution Loop (up to 4 retries)
1. Generate a self-contained Python script using the template from `scripts/figure_template.py` as a starting point
2. Write script to a temp file and execute: `python figure_script.py`
3. If error: capture traceback, feed back, regenerate (see ERROR_PROMPT in references)
4. If no `.png` produced: add explicit save instruction, retry
5. On success: report the generated figure path

### Phase 3: Visual Refinement
Read the generated PNG file and visually inspect using the VLM feedback prompts from `references/figure-prompts.md`:
- Does the figure type match the request?
- Are labels, titles, and legends correct?
- Is the color scheme appropriate and consistent?
- Are axis scales sensible? Is text readable at publication size?

If improvements needed: generate corrective instructions and re-execute.

## References

- All MatPlotAgent prompts: `~/.claude/skills/figure-generation/references/figure-prompts.md`
- Figure templates: `~/.claude/skills/figure-generation/scripts/figure_template.py`

## Output

Both PNG (preview, 300 DPI) and PDF (vector, for paper) formats. Plus the LaTeX include code:

```latex
\begin{figure}[t]
    \centering
    \includegraphics[width=\linewidth]{figures/figure_name.pdf}
    \caption{Description. Best viewed in color.}
    \label{fig:figure_name}
\end{figure}
```

## Quality Requirements
- DPI ≥ 300, or vector PDF
- Colorblind-friendly palette (no red-green only)
- All text ≥ 8pt at print size
- Consistent styling across all paper figures
- No matplotlib default title — use LaTeX caption

## Related Skills
- Upstream: [data-analysis](../data-analysis/), [experiment-code](../experiment-code/)
- Downstream: [paper-writing-section](../paper-writing-section/), [paper-compilation](../paper-compilation/), [slide-generation](../slide-generation/)
- See also: [table-generation](../table-generation/)

## Source & license

This open-source skill is cataloged on AgentStack and links to its original source — we do not rehost the code.

- **Author:** [ARAVINDAN20](https://github.com/ARAVINDAN20)
- **Source:** [ARAVINDAN20/Claude-Research-Paper-OS](https://github.com/ARAVINDAN20/Claude-Research-Paper-OS)
- **License:** Apache-2.0

Install and usage instructions live in the source repository linked above.

## Pricing

- **Free** — Free

## Security capabilities

Automated source analysis of v0.1.0 — what this tool can access:

- **Network access:** no
- **Filesystem access:** no
- **Shell / process execution:** no
- **Environment & secrets:** no
- **Dynamic code execution:** no

*"Yes" means the capability is present in the source — more access means more to trust, not that it is unsafe.*


## Versions

- **0.1.0** — security scan: passed — Imported from the upstream source.

## Links

- Listing page: https://agentstack.voostack.com/l/skill-aravindan20-claude-research-paper-os-figure-generation
- Seller: https://agentstack.voostack.com/s/aravindan20
- Browse the marketplace: https://agentstack.voostack.com/browse

---
Listed on AgentStack — the marketplace for AI agent skills and MCP servers. Every listing is security-reviewed. Creators keep 70%.