# AutoFigure: Generating and Refining Publication-Ready Scientific Illustrations [ICLR 2026]
[](https://openreview.net/forum?id=5N3z9JQJKq)
[](https://opensource.org/licenses/MIT)
[](https://www.python.org/)
[](https://huggingface.co/datasets/WestlakeNLP/FigureBench)
[](https://deepscientist.cc/)
From Text to Publication-Ready Diagrams
AutoFigure is an intelligent system that leverages Large Language Models (LLMs) with iterative refinement to generate high-quality scientific figures from text descriptions or research papers.
|
| **📊 Survey Case**
|
| **📝 Blog Case**
|
| **📘 Textbook Case**
|
---
## ⚡ Quick Start
### Option 1: Python SDK (Recommended)
You can install via cloning the repo:
```bash
git clone https://github.com/ResearAI/AutoFigure.git
cd AutoFigure
pip install -e .
playwright install chromium # Required for rendering
```
#### 1. Basic Usage (Text-to-Figure)
```python
from autofigure import AutoFigureAgent, Config
# 1. Configure
config = Config(
generation_api_key="your-api-key",
generation_provider="openrouter", # options: 'openrouter', 'gemini', 'bianxie'
generation_model="google/gemini-2.5-pro",
)
# 2. Generate
agent = AutoFigureAgent(config)
result = agent.generate(
description="A flowchart showing transformer training pipeline",
max_iterations=5,
output_format="svg",
topic="paper" # 'paper', 'survey', 'blog', 'textbook'
)
print(f"✅ Generated: {result.svg_path} (Score: {result.final_score}/10)")
```
#### 2. Generate from Paper (PDF/Markdown)
Extract methodology from a paper and generate a figure automatically.
```python
# Generate figure from paper (PDF or Markdown)
result = agent.generate_from_paper(
paper_path="./paper.pdf",
max_iterations=5,
output_format="svg",
enable_enhancement=True, # Enhance the result
)
if result.success:
print(f"Extracted methodology: {result.methodology_text[:200]}...")
print(f"Generated figure: {result.svg_path}")
```
#### 3. With Image Enhancement
Generate multiple enhanced aesthetic variants of the figure.
```python
result = agent.generate(
description="Neural network architecture diagram",
enable_enhancement=True,
enhancement_count=3, # Generate 3 variants
art_style="Modern scientific illustration with clean lines",
enhancement_input_type="code2prompt" # Best quality mode
)
if result.success:
print(f"Original Preview: {result.preview_path}")
print(f"Enhanced variants: {result.enhanced_paths}")
```
### Option 2: Web Interface
Ideally suited for visual interaction and editing.
```bash
./start.sh
# Then open http://localhost:6002 in your browser
```
---
## 📊 FigureBench Dataset
We introduce **FigureBench**, the first large-scale benchmark for generating scientific illustrations from long-form text.
