Dynamically added segments crash fix

Sources/SegmentedPicker/SegmentedPicker.swift

@@ -11,10 +11,10 @@ public struct SegmentedPicker<Element, Content, Selection>: View
     where
     Content: View,
     Selection: View {
-
+    
     public typealias Data = [Element]
-
-    @State private var frames: [CGRect]
+    
+    @State private var frames: [Data.Index: CGRect] = [:]
     @Binding private var selectedIndex: Data.Index?
 
     private let data: Data
@@ -32,19 +32,18 @@ public struct SegmentedPicker<Element, Content, Selection>: View
         self.content = content
         self.selection = selection
         self._selectedIndex = selectedIndex
-        self._frames = State(wrappedValue: Array(repeating: .zero,
-                                                 count: data.count))
         self.selectionAlignment = selectionAlignment
     }
+
 
     public var body: some View {
         ZStack(alignment: Alignment(horizontal: .horizontalCenterAlignment,
                                     vertical: selectionAlignment)) {
 
-            if let selectedIndex = selectedIndex {
+            if let index = selectedIndex {
                 selection()
-                    .frame(width: frames[selectedIndex].width,
-                           height: frames[selectedIndex].height)
+                    .frame(width: selectionFrame(at: index).width,
+                           height: selectionFrame(at: index).height)
                     .alignmentGuide(.horizontalCenterAlignment) { dimensions in
                         dimensions[HorizontalAlignment.center]
                     }
@@ -57,7 +56,9 @@ public struct SegmentedPicker<Element, Content, Selection>: View
                     )
                     .buttonStyle(PlainButtonStyle())
                     .background(GeometryReader { proxy in
-                        Color.clear.onAppear { frames[index] = proxy.frame(in: .global) }
+                        Color.clear.onAppear {
+                            frames[index] = proxy.frame(in: .global)
+                        }
                     })
                     .alignmentGuide(.horizontalCenterAlignment,
                                     isActive: selectedIndex == index) { dimensions in
@@ -67,4 +68,8 @@ public struct SegmentedPicker<Element, Content, Selection>: View
             }
         }
     }
+
+    private func selectionFrame(at index: Data.Index) -> CGRect {
+        frames[index] ?? .zero
+    }
 }