Metadata-Version: 2.4
Name: loom-pipeline
Version: 0.2.2
Summary: Visual pipeline editor and runner for task automation
Author-email: Relja Ljubobratovic <ljubobratovic.relja@gmail.com>
License-Expression: MIT
Project-URL: Homepage, https://github.com/ljubobratovicrelja/loom
Project-URL: Repository, https://github.com/ljubobratovicrelja/loom
Project-URL: Documentation, https://ljubobratovicrelja.github.io/loom/
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Requires-Python: >=3.11
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: PyYAML>=6.0
Requires-Dist: requests>=2.28.0
Provides-Extra: ui
Requires-Dist: fastapi>=0.100.0; extra == "ui"
Requires-Dist: uvicorn>=0.23.0; extra == "ui"
Requires-Dist: websockets>=12.0; extra == "ui"
Requires-Dist: ruamel.yaml>=0.18.0; extra == "ui"
Requires-Dist: send2trash>=1.8.2; extra == "ui"
Requires-Dist: opencv-python-headless>=4.8.0; extra == "ui"
Provides-Extra: examples
Requires-Dist: numpy>=1.24.0; extra == "examples"
Requires-Dist: scipy>=1.10.0; extra == "examples"
Requires-Dist: matplotlib>=3.7.0; extra == "examples"
Requires-Dist: opencv-python-headless>=4.8.0; extra == "examples"
Provides-Extra: dev
Requires-Dist: loom-pipeline[examples,ui]; extra == "dev"
Requires-Dist: pytest>=7.0.0; extra == "dev"
Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
Requires-Dist: httpx>=0.24.0; extra == "dev"
Requires-Dist: ruff>=0.1.0; extra == "dev"
Requires-Dist: mypy>=1.0.0; extra == "dev"
Requires-Dist: types-PyYAML>=6.0.0; extra == "dev"
Requires-Dist: types-requests>=2.28.0; extra == "dev"
Requires-Dist: mkdocs-material>=9.0.0; extra == "dev"
Dynamic: license-file

![Loom Banner](https://raw.githubusercontent.com/ljubobratovicrelja/loom/main/media/banner.svg)

[![Live Demo](https://img.shields.io/badge/demo-live-brightgreen)](https://loom-examples.onrender.com/)
[![CI](https://github.com/ljubobratovicrelja/loom/actions/workflows/ci.yml/badge.svg)](https://github.com/ljubobratovicrelja/loom/actions/workflows/ci.yml)
[![Docs](https://img.shields.io/badge/docs-available-brightgreen)](https://ljubobratovicrelja.github.io/loom/)
[![PyPI](https://img.shields.io/pypi/v/loom-pipeline)](https://pypi.org/project/loom-pipeline/)

A lightweight visual pipeline runner for research.

Connect your Python scripts into a graph, tweak parameters, run experiments, see results — without setting up Airflow or learning a workflow framework.

**[Try the live demo](https://loom-examples.onrender.com/)** — no installation required. Browse and run the [example pipelines](examples/) in your browser. (First load may take ~30s to wake up.)

Loom gives you a CLI runner and visual editor for pipelines defined in YAML. Your scripts stay as regular Python with argparse — no framework to learn, no rewrites needed.

It's designed for research workflows. For production orchestration, tools like Airflow or Kubeflow are better suited.

## Installation

```bash
# Core runner only
pip install loom-pipeline

# With visual editor
pip install loom-pipeline[ui]
```

That's it. No configuration files to create, no external services to manage.

## Quick Start

Clone the repo and try an example:

```bash
git clone https://github.com/ljubobratovicrelja/loom.git
cd loom
pip install -e .[ui,examples]

# Run a pipeline from the command line
loom examples/image-processing/pipeline.yml
```

```
Pipeline: 3 step(s) to run [parallel]
----------------------------------------
[RUNNING] grayscale
[grayscale] Converted to grayscale: .loom-url-cache/35bb4a6_Lenna.png -> data/grayscale.png
[SUCCESS] grayscale
[RUNNING] blur
[blur] Gaussian blur (radius=15): data/grayscale.png -> data/blurred.png
[SUCCESS] blur
[RUNNING] edge_detect
[edge_detect] Edge detection: data/grayscale.png -> data/edges.png
[SUCCESS] edge_detect
----------------------------------------
Completed: 3/3 steps succeeded
```

Or open it in the visual editor:

```bash
# Edit a single pipeline
loom-ui examples/image-processing/pipeline.yml

# Browse all example pipelines
loom-ui examples/
```

The editor opens in your browser where you can see the pipeline graph, run steps, and view outputs.

## Building Your Own Pipeline

Add Loom to your project's environment:

```bash
pip install loom-pipeline[ui]  # or just loom-pipeline for CLI only
```

Now you can run pipelines from within your project. Here's how to set one up.

### 1. Point it at your scripts

Say you have some Python scripts that process data:

```
tasks/
  extract_features.py    # Takes video, outputs CSV
  train_model.py         # Takes CSV, outputs model
  evaluate.py            # Takes model + test data, outputs metrics
```

### 2. Describe the pipeline in YAML

```yaml
# experiment.yml
variables:
  video: data/raw/recording.mp4
  features: data/processed/features.csv
  model: models/classifier.pt
  metrics: results/metrics.json

parameters:
  learning_rate: 0.001
  epochs: 100

pipeline:
  - name: extract
    task: tasks/extract_features.py
    inputs:
      video: $video
    outputs:
      -o: $features

  - name: train
    task: tasks/train_model.py
    inputs:
      data: $features
    outputs:
      -o: $model
    args:
      --lr: $learning_rate
      --epochs: $epochs

  - name: evaluate
    task: tasks/evaluate.py
    inputs:
      model: $model
    outputs:
      -o: $metrics
```

### 3. Run it

```bash
# Run the full pipeline
loom experiment.yml

# Run just one step
loom experiment.yml --step train

# Run from a step onward
loom experiment.yml --from train

# Try different parameters
loom experiment.yml --set learning_rate=0.01 --set epochs=200

# Override file paths
loom experiment.yml --var video=other_recording.mp4

# Run steps in parallel
loom experiment.yml --parallel --max-workers 4

# Preview without executing
loom experiment.yml --dry-run

# Clean all data (move to trash) and re-run from scratch
loom experiment.yml --clean
loom experiment.yml

# Preview what would be cleaned
loom experiment.yml --clean-list
```

### 4. Or use the visual editor

```bash
loom-ui experiment.yml
```

This opens a browser-based editor where you can:
- See your pipeline as a visual graph
- Drag and drop to reorganize
- Run individual steps and see output in real-time
- Quickly see which outputs exist (green) vs missing (grey)

You can also point it at a directory to browse multiple pipelines:

```bash
loom-ui experiments/    # Browse all pipelines in a folder
```

Each pipeline should be in its own subdirectory with a `pipeline.yml` file inside. See [examples/](examples/) for the expected structure.

## How Scripts Work

Your scripts stay normal Python with argparse. Just add a YAML block in the docstring so Loom knows the interface:

```python
"""Extract features from video.

---
inputs:
  video:
    type: video
    description: Input video file
outputs:
  -o:
    type: csv
    description: Output features
args:
  --sample-rate:
    type: int
    default: 30
    description: Frames to sample per second
---
"""

import argparse

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("video")
    parser.add_argument("-o", "--output", required=True)
    parser.add_argument("--sample-rate", type=int, default=30)
    args = parser.parse_args()
    # ... your code ...

if __name__ == "__main__":
    main()
```

The YAML frontmatter is optional but enables the editor to show input/output types and provide better validation.

## Use Cases

**Parameter exploration**: Create parallel branches in your pipeline to test different configurations side by side.

**Reproducible experiments**: The YAML file captures your entire experiment setup. Commit it to git alongside your code.

**Iterative development**: Run just the steps you're working on. Loom tracks dependencies so upstream steps run only when needed.

**Result organization**: Variables point to file paths, so your outputs are organized by experiment configuration.

## Philosophy

Loom is intentionally minimal:

- **No database** — Everything is files: your scripts, YAML configs, and outputs
- **No external services** — The visual editor runs a local server that stops when you close it
- **No lock-in** — Your scripts work with or without Loom
- **No magic** — Loom just builds shell commands and runs them

This makes it easy to adopt incrementally. Start with one experiment, see if it helps, expand from there.

## License

MIT License — see [LICENSE](LICENSE) for details.

---

*Built for researchers who want to see their experiments, not manage infrastructure.*
