From c85692360c4d9f4e946965542f9f94026541e8bb Mon Sep 17 00:00:00 2001 From: Neeraj Gupta <254676+ua741@users.noreply.github.com> Date: Tue, 2 Apr 2024 13:35:32 +0530 Subject: [PATCH] [mob] Fix face cluster suggestion --- mobile/lib/face/db.dart | 9 +++++---- .../lib/ui/settings/debug/face_debug_section_widget.dart | 1 + 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/mobile/lib/face/db.dart b/mobile/lib/face/db.dart index e3b2d56c5c..502f48dee4 100644 --- a/mobile/lib/face/db.dart +++ b/mobile/lib/face/db.dart @@ -545,13 +545,14 @@ class FaceMLDataDB { final db = instance.database; return db.then((db) async { final List> maps = await db.rawQuery( - 'SELECT $cluserIDColumn, $fileIDColumn FROM $facesTable ' - 'WHERE $cluserIDColumn IN (${clusterIDs.join(",")})', + 'SELECT $fcClusterID, $fcFaceId FROM $faceClustersTable ' + 'WHERE $fcClusterID IN (${clusterIDs.join(",")})', ); final Map> result = {}; for (final map in maps) { - final clusterID = map[cluserIDColumn] as int; - final fileID = map[fileIDColumn] as int; + final clusterID = map[fcClusterID] as int; + final faceId = map[fcFaceId] as String; + final fileID = int.parse(faceId.split("_").first); result[fileID] = (result[fileID] ?? {})..add(clusterID); } return result; diff --git a/mobile/lib/ui/settings/debug/face_debug_section_widget.dart b/mobile/lib/ui/settings/debug/face_debug_section_widget.dart index 5a76320c32..937201be90 100644 --- a/mobile/lib/ui/settings/debug/face_debug_section_widget.dart +++ b/mobile/lib/ui/settings/debug/face_debug_section_widget.dart @@ -186,6 +186,7 @@ class _FaceDebugSectionWidgetState extends State { pressedColor: getEnteColorScheme(context).fillFaint, trailingIcon: Icons.chevron_right_outlined, trailingIconIsMuted: true, + alwaysShowSuccessState: true, onTap: () async { await showChoiceDialog( context,