Update app.py

app.py

@@ -719,10 +719,22 @@ function draw() {
         let transfers = TRANSFERS.filter(x=>x[0]===nodeName || x[1]===nodeName).map(x=>x[0]===nodeName?x[1]:x[0]);
         let loops = LOOPS.filter(x=>x[0]===nodeName || x[2]===nodeName).map(x=>x[0]===nodeName?x[2]:x[0]);
     
-        // <-- NEW: include FRESH edges so teal arcs behave like real connections
+        // include FRESH edges
         let fresh = FRESH.filter(x=>x[0]===nodeName || x[1]===nodeName).map(x=>x[0]===nodeName?x[1]:x[0]);
+
+        // include COMPLETE_EXIT connections:
+        // - if nodeName is an AMC, include companies in COMPLETE_EXIT[nodeName]
+        // - if nodeName is a company, include AMCs that list that company in their COMPLETE_EXIT
+        let exits = [];
+        if (COMPLETE_EXIT[nodeName]) {
+            // nodeName is an AMC
+            exits = exits.concat(COMPLETE_EXIT[nodeName]);
+        } else {
+            // nodeName may be a company; find AMCs that have it
+            exits = exits.concat(Object.keys(COMPLETE_EXIT).filter(amc => (COMPLETE_EXIT[amc] || []).includes(nodeName)));
+        }
     
-        return new Set([].concat(buys, sells, transfers, loops, fresh));
+        return new Set([].concat(buys, sells, transfers, loops, fresh, exits));
     }
 
 
@@ -786,18 +798,19 @@ function draw() {
         const loops = LOOPS.filter(x => x[0]===nodeName || x[2]===nodeName)
                            .map(x => x[0]===nodeName ? x[2] : x[0]);
     
-        // NEW — detect Fresh Buys
+        // detect Fresh Buys
         const fresh = FRESH.filter(x => x[0] === nodeName || x[1] === nodeName)
                            .map(x => x[0] === nodeName ? x[1] : x[0]);
 
-        // NEW — detect Complete Exits
-        const exits = [];
-        COMPLETE_EXIT = __COMPLETE_EXIT__;   // inject the map (see below)
+        // detect Complete Exits:
+        // if nodeName is AMC, list companies they completely exited
+        // if nodeName is company, list AMCs who completely exited that company
+        let exits = [];
         if (COMPLETE_EXIT[nodeName]) {
-            exits.push(...COMPLETE_EXIT[nodeName]);
+            exits = exits.concat(COMPLETE_EXIT[nodeName]); // nodeName is AMC
+        } else {
+            exits = exits.concat(Object.keys(COMPLETE_EXIT).filter(amc => (COMPLETE_EXIT[amc] || []).includes(nodeName)));
         }
-
-
     
         const box = document.getElementById("info-box");
         box.innerHTML = `
@@ -807,7 +820,7 @@ function draw() {
             <b>Sells:</b> ${sells.length ? sells.join(", ") : "<span style='color:#777'>None</span>"}<br>
             <b>Transfers:</b> ${transfers.length ? transfers.join(", ") : "<span style='color:#777'>None</span>"}<br>
             <b>Loops:</b> ${loops.length ? loops.join(", ") : "<span style='color:#777'>None</span>"}<br>
-            <b>Fresh Buys:</b> ${fresh.length ? fresh.join(", ") : "<span style='color:#777'>None</span>"}
+            <b>Fresh Buys:</b> ${fresh.length ? fresh.join(", ") : "<span style='color:#777'>None</span>"}<br>
             <b>Complete Exits:</b> ${exits.length ? exits.join(", ") : "<span style='color:#777'>None</span>"}<br>
           </div>
         `;