From a65aaac61ba3742ac27feb0adcb15443c9c40bc3 Mon Sep 17 00:00:00 2001
From: Benedikt Willi <benedikt.willi@unibas.ch>
Date: Tue, 13 Jan 2026 14:24:00 +0100
Subject: [PATCH] Enhance DOM graph visualization to support PNG images and
 improve legend display

---
 assets/pages/dom_graph.html | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/assets/pages/dom_graph.html b/assets/pages/dom_graph.html
index 59b1b8b..02234ed 100644
--- a/assets/pages/dom_graph.html
+++ b/assets/pages/dom_graph.html
@@ -179,7 +179,7 @@
         <div class="error">
             {{ error }}
         </div>
-        {% elif is_svg %}
+        {% elif is_svg or is_png %}
         <div class="legend">
             <h3>Color Legend</h3>
             <div class="legend-grid">
@@ -210,8 +210,12 @@
             </div>
         </div>
         
-        <div class="graph-wrapper">
+        <div class="graph-wrapper" {% if is_png %}style="display: block; width: 100%;"{% endif %}>
+            {% if is_png %}
+            <img src="data:image/png;base64,{{ graph_content }}" alt="DOM Graph" style="max-width: 100%; height: auto; display: block;">
+            {% else %}
             {{ graph_content|safe }}
+            {% endif %}
         </div>
         {% else %}
         <div class="install-note">