| """Visualization payload — formats the patient subgraph for react-force-graph. |
| |
| The raras-app /grafo page renders nodes/links via react-force-graph-3d. |
| This module produces the exact JSON shape that component expects, so the |
| front-end can drop in `<ForceGraph data={twin.viz_data} />`. |
| |
| Color map is consistent with raras-app conventions: |
| Patient #00d4ff |
| Phenotype #f59e0b |
| Disease #ef4444 |
| Gene #10b981 |
| Drug #8b5cf6 |
| Lab #6366f1 |
| """ |
| from __future__ import annotations |
| from typing import Optional |
|
|
| from .types import VizData, Subgraph |
|
|
| _COLOR = { |
| "Patient": "#00d4ff", |
| "Phenotype": "#f59e0b", |
| "Disease": "#ef4444", |
| "Gene": "#10b981", |
| "Drug": "#8b5cf6", |
| "Lab": "#6366f1", |
| "Snapshot": "#a78bfa", |
| } |
|
|
| _GROUP = { |
| "Patient": 0, "Phenotype": 1, "Disease": 2, "Gene": 3, |
| "Drug": 4, "Lab": 5, "Snapshot": 6, |
| } |
|
|
| _VAL = { |
| "Patient": 30, "Disease": 18, "Gene": 12, "Phenotype": 10, |
| "Drug": 14, "Lab": 8, "Snapshot": 9, |
| } |
|
|
|
|
| def from_subgraph(sg: Subgraph, *, center_id: Optional[str] = None) -> VizData: |
| """Convert a `Subgraph` into a force-graph-friendly payload.""" |
| nodes = [] |
| for n in sg.nodes: |
| nodes.append({ |
| "id": n.id, |
| "name": n.name, |
| "group": _GROUP.get(n.label, 9), |
| "label": n.label, |
| "val": _VAL.get(n.label, 8) * (0.5 + n.weight), |
| "color": _COLOR.get(n.label, "#94a3b8"), |
| "hpo": n.code if n.label == "Phenotype" else None, |
| "orpha": n.code if n.label == "Disease" else None, |
| "symbol": n.code if n.label == "Gene" else None, |
| "rxcui": n.code if n.label == "Drug" else None, |
| "weight": n.weight, |
| **(n.extra or {}), |
| }) |
| links = [] |
| for e in sg.edges: |
| links.append({ |
| "source": e.source, |
| "target": e.target, |
| "label": e.rel, |
| "value": max(0.5, e.weight), |
| "evidence": e.evidence, |
| }) |
|
|
| legend = {label: {"color": color, "group": _GROUP.get(label, 9)} for label, color in _COLOR.items()} |
|
|
| return VizData( |
| nodes=nodes, |
| links=links, |
| center_id=center_id, |
| legend=legend, |
| ) |
|
|
