Fix modification panel not dismissing after accept

Author: intitni

Core/Sources/SuggestionWidget/FeatureReducers/PromptToCodeGroup.swift

@@ -19,6 +19,9 @@ public struct PromptToCodeGroup {
             }
             set {
                 selectedTabId = newValue?.id
+                if let id = selectedTabId {
+                    promptToCodes[id: id] = newValue
+                }
             }
         }
     }
@@ -88,7 +91,13 @@ public struct PromptToCodeGroup {
                 return .none
 
             case let .discardAcceptedPromptToCodeIfNotContinuous(id):
-                state.promptToCodes.removeAll { $0.id == id && $0.hasEnded }
+                for itemId in state.promptToCodes.ids {
+                    if itemId == id, state.promptToCodes[id: itemId]?.clickedButton == .accept {
+                        state.promptToCodes.remove(id: itemId)
+                    } else {
+                        state.promptToCodes[id: itemId]?.clickedButton = nil
+                    }
+                }
                 return .none
 
             case let .updateActivePromptToCode(documentURL):

Core/Sources/SuggestionWidget/FeatureReducers/PromptToCodePanel.swift

@@ -17,6 +17,11 @@ public struct PromptToCodePanel {
         public enum FocusField: Equatable {
             case textField
         }
+        
+        public enum ClickedButton: Equatable {
+            case accept
+            case acceptAndContinue
+        }
 
         @Shared public var promptToCodeState: ModificationState
         @ObservationStateIgnored
@@ -36,7 +41,7 @@ public struct PromptToCodePanel {
 
         public var generateDescriptionRequirement: Bool
 
-        public var hasEnded = false
+        public var clickedButton: ClickedButton?
 
         public var isActiveDocument: Bool = false
 
@@ -250,13 +255,14 @@ public struct PromptToCodePanel {
                 return .cancel(id: CancellationKey.modifyCode(state.id))
 
             case .acceptButtonTapped:
-                state.hasEnded = true
+                state.clickedButton = .accept
                 return .run { _ in
                     await commandHandler.acceptModification()
                     activatePreviousActiveXcode()
                 }
 
             case .acceptAndContinueButtonTapped:
+                state.clickedButton = .acceptAndContinue
                 return .run { _ in
                     await commandHandler.acceptModification()
                     activateThisApp()