Preserve dataflow layout across SQL edits

webui/src/features/dataflow/DataflowPage.tsx

@@ -397,7 +397,12 @@ export default function DataflowPage({ settings, auth }: { settings: ApiSettings
         sqlText: response.sql_text ?? draft.sqlText,
         editorMode: switchToGraph ? "graph" : "sql"
       });
-      setSelectedNodeId(response.graph.nodes[0]?.id ?? null);
+      setSelectedNodeId(
+        selectedNodeId
+          && response.graph.nodes.some((node) => node.id === selectedNodeId)
+          ? selectedNodeId
+          : response.graph.nodes[0]?.id ?? null
+      );
       setSuccess("SQL compiled into the pipeline graph.");
       return true;
     } catch (compileError) {

webui/src/styles/dataflow.css

@@ -66,8 +66,7 @@
 
 .dataflow-toolbar-actions,
 .dataflow-command-bar,
-.dataflow-identity-fields,
-.dataflow-results-toolbar .segmented-control {
+.dataflow-identity-fields {
   display: flex;
   align-items: center;
   gap: 7px;