From dc74a9448b627bbdf002cdd0663b1e6e4aa8e87d Mon Sep 17 00:00:00 2001
From: Nils-Johan Gynther <nils-johan.gynther@neoleaf.se>
Date: Sun, 3 May 2026 15:03:17 +0200
Subject: [PATCH] refactor(edit-dialog): improve category selection handling in
 AI suggestion

Co-authored-by: Copilot <copilot@github.com>
---
 .../features/import/presentation/receipt_import_tab.dart | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/flutter/lib/features/import/presentation/receipt_import_tab.dart b/flutter/lib/features/import/presentation/receipt_import_tab.dart
index 705f270e..2b5ed286 100644
--- a/flutter/lib/features/import/presentation/receipt_import_tab.dart
+++ b/flutter/lib/features/import/presentation/receipt_import_tab.dart
@@ -349,6 +349,7 @@ class _EditDialogState extends State<_EditDialog> {
 
   void _applyAiSuggestionForExistingSelection() {
     final suggestedId = widget.item.suggestedProductId;
+    int? preselectedCategoryId = widget.item.categorySuggestionId;
     if (suggestedId != null) {
       setState(() {
         _productId = suggestedId;
@@ -360,11 +361,11 @@ class _EditDialogState extends State<_EditDialog> {
             _categoryPathForCategoryId(_productCategoryId) ?? widget.item.categorySuggestionPath;
         _productCategorySource = CategorySelectionSource.ai;
       });
-      return;
+      preselectedCategoryId = _productCategoryId;
     }
 
     _openExistingCategoryPicker(
-      preselectedCategoryId: widget.item.categorySuggestionId,
+      preselectedCategoryId: preselectedCategoryId,
     );
   }
 
@@ -634,7 +635,9 @@ class _EditDialogState extends State<_EditDialog> {
                       backgroundColor: Colors.green.shade50,
                       side: BorderSide(color: Colors.green.shade300),
                       visualDensity: VisualDensity.compact,
-                      onPressed: _applyAiSuggestionForExistingSelection,
+                      onPressed: () => _openExistingCategoryPicker(
+                        preselectedCategoryId: item.categorySuggestionId,
+                      ),
                     ),
                   ],
                 ],