README.md

metadata

title: Diagram Structure Extractor
emoji: 📊
colorFrom: blue
colorTo: green
sdk: docker
app_port: 8000
pinned: false
license: mit

Diagram Structure Extractor

Upload an architecture diagram and get its structure back as schema-valid JSON: every text label, box, arrow, icon, and the relationships between components. Output is a typed Pydantic model, so the JSON parses on every call.

Demo: the Space opens on a Streamlit app. Upload a diagram or pick one of the samples to see the annotated overlay, the relationship graph and the JSON, with JSON and CSV downloads. The sidebar switches each detector stage.

API (same host)

• POST /extract — multipart image upload; returns the typed structure, an annotated PNG, and a flat edge CSV
• GET /health — liveness
• GET /docs — interactive API docs

Measured results (relationship F1; edges are directed, and a double-headed connector counts once each way)

• 14 synthetic reference architectures: 1.000
• A real draw.io export: 0.900
• 10 held-out draw.io-style diagrams: 0.789 (0.711 when first scored, before the code-review fixes)
• Schema-valid JSON on every input, at every pipeline stage
• Detectors: PaddleOCR text (F1 0.962), Canny boxes (1.000), directed-lines arrows (1.000 on the synthetic set), template-matched icons (1.000)

Handles draw.io conventions: thin dashed connectors, small and double-headed arrowheads, elbow routing, line labels, icon nodes with captions underneath.

Source: https://github.com/Mark007-R/Diagram-Structure-Extractor