Import CSV

script.js

@@ -513,10 +513,13 @@ function initDiagram() {
            .attr("viewBox", [0, 0, width, height])
            .attr("preserveAspectRatio", "xMinYMin");
         
-        // Create Sankey generator with increased padding for better distribution
+        // Create Sankey generator with dynamic padding based on node count
+        const nodeCount = sankeyData.nodes.length;
+        const dynamicPadding = Math.max(20, Math.min(120, Math.floor((height - textPadding) / (nodeCount * 1.5))));
+        
         const sankey = d3.sankey()
             .nodeWidth(parseInt(document.getElementById('node-width').value))
-            .nodePadding(120) // Increased padding for better vertical distribution
+            .nodePadding(dynamicPadding)
             .size([width - textPadding, height - textPadding]);    // Create a copy of the data
     const graph = {
         nodes: sankeyData.nodes.map(d => Object.assign({}, d)),
@@ -859,6 +862,9 @@ function importFromCSV(file) {
         const nodeMap = {};
         let nodeIndex = 0;
         
+        // First pass: collect all unique nodes and their relationships
+        const nodeRelations = new Map(); // Track incoming/outgoing links for each node
+        
         // Parse data rows
         for (let i = 1; i < lines.length; i++) {
             const values = lines[i].split(',').map(v => v.trim().replace(/"/g, ''));
@@ -870,15 +876,36 @@ function importFromCSV(file) {
             
             if (isNaN(value)) continue;
             
+            // Track relationships for column assignment
+            if (!nodeRelations.has(sourceName)) {
+                nodeRelations.set(sourceName, { in: 0, out: 0, totalValue: 0 });
+            }
+            if (!nodeRelations.has(targetName)) {
+                nodeRelations.set(targetName, { in: 0, out: 0, totalValue: 0 });
+            }
+            
+            nodeRelations.get(sourceName).out += 1;
+            nodeRelations.get(sourceName).totalValue += value;
+            nodeRelations.get(targetName).in += 1;
+            nodeRelations.get(targetName).totalValue += value;
+            
             // Add nodes if they don't exist
             if (!(sourceName in nodeMap)) {
                 nodeMap[sourceName] = nodeIndex++;
-                sankeyData.nodes.push({ name: sourceName, color: "#4f46e5" });
+                sankeyData.nodes.push({ 
+                    name: sourceName,
+                    color: "#4f46e5",
+                    value: 0 // Will be updated later
+                });
             }
             
             if (!(targetName in nodeMap)) {
                 nodeMap[targetName] = nodeIndex++;
-                sankeyData.nodes.push({ name: targetName, color: "#4f46e5" });
+                sankeyData.nodes.push({ 
+                    name: targetName,
+                    color: "#4f46e5",
+                    value: 0 // Will be updated later
+                });
             }
             
             // Add link
@@ -889,6 +916,16 @@ function importFromCSV(file) {
             });
         }
         
+        // Update node values based on maximum flow
+        sankeyData.nodes.forEach(node => {
+            const relation = nodeRelations.get(node.name);
+            node.value = Math.max(
+                relation.totalValue,
+                relation.in ? relation.totalValue / relation.in : 0,
+                relation.out ? relation.totalValue / relation.out : 0
+            );
+        });
+        
         updateLists();
         initDiagram();
         saveData();
@@ -896,6 +933,7 @@ function importFromCSV(file) {
     reader.readAsText(file);
 }
 
+
 // Center diagram
 function centerDiagram() {
     nodePositions = {};