From 47089e1e676b4be0c26f00985747d603cfe8e644 Mon Sep 17 00:00:00 2001
From: T8RIN <oolbp@bk.ru>
Date: Sun, 5 Jan 2025 09:06:14 +0300
Subject: [PATCH] UI improved

---
 .../core/ui/widget/image/ImagePreviewGrid.kt                 | 5 ++++-
 .../document_scanner/presentation/DocumentScannerContent.kt  | 5 ++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/core/ui/src/main/kotlin/ru/tech/imageresizershrinker/core/ui/widget/image/ImagePreviewGrid.kt b/core/ui/src/main/kotlin/ru/tech/imageresizershrinker/core/ui/widget/image/ImagePreviewGrid.kt
index 2bc1005af..3bdc8ee4f 100644
--- a/core/ui/src/main/kotlin/ru/tech/imageresizershrinker/core/ui/widget/image/ImagePreviewGrid.kt
+++ b/core/ui/src/main/kotlin/ru/tech/imageresizershrinker/core/ui/widget/image/ImagePreviewGrid.kt
@@ -174,7 +174,10 @@ fun ImagePreviewGrid(
             },
             onShare = onShareImage,
             onDismiss = { showImagePreviewDialog = false },
-            onNavigate = onNavigate
+            onNavigate = {
+                showImagePreviewDialog = false
+                onNavigate(it)
+            }
         )
     }
 
diff --git a/feature/document-scanner/src/main/java/ru/tech/imageresizershrinker/feature/document_scanner/presentation/DocumentScannerContent.kt b/feature/document-scanner/src/main/java/ru/tech/imageresizershrinker/feature/document_scanner/presentation/DocumentScannerContent.kt
index 4b800e237..ec22677a2 100644
--- a/feature/document-scanner/src/main/java/ru/tech/imageresizershrinker/feature/document_scanner/presentation/DocumentScannerContent.kt
+++ b/feature/document-scanner/src/main/java/ru/tech/imageresizershrinker/feature/document_scanner/presentation/DocumentScannerContent.kt
@@ -196,7 +196,10 @@ fun DocumentScannerContent(
                     visible = selectedUriForPreview != null,
                     selectedUri = selectedUriForPreview,
                     uris = component.uris,
-                    onNavigate = component.onNavigate,
+                    onNavigate = {
+                        selectedUriForPreview = null
+                        component.onNavigate(it)
+                    },
                     onUriSelected = { selectedUriForPreview = it },
                     onShare = component::shareUri,
                     onDismiss = { selectedUriForPreview = null }