feat: AutoTextCorrector 적용

Lite-Weight · Nov 14, 2023 · b1bd753 · b1bd753
1 parent e47c439
commit b1bd753
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 4 deletions.
diff --git a/feat/capture/build.gradle.kts b/feat/capture/build.gradle.kts
@@ -46,6 +46,7 @@ android {
 
 dependencies {
     implementation(project(":common"))
+    implementation(project(":autotextcorrection"))
 
     // Android
     implementation(AndroidX.CORE_KTX)

diff --git a/feat/capture/src/main/java/com/konkuk/capture/ui/enroll/EnrollTextInputViewModel.kt b/feat/capture/src/main/java/com/konkuk/capture/ui/enroll/EnrollTextInputViewModel.kt
@@ -4,6 +4,7 @@ import android.graphics.Bitmap
 import android.net.Uri
 import androidx.lifecycle.SavedStateHandle
 import androidx.lifecycle.ViewModel
+import com.konkuk.autotextcorrection.AutoTextCorrector
 import com.konkuk.common.data.FoodInfo
 import dagger.hilt.android.lifecycle.HiltViewModel
 import kotlinx.coroutines.flow.MutableStateFlow
@@ -40,11 +41,14 @@ class EnrollTextInputViewModel @Inject constructor(
 
     init {
         savesStateHandle.get<String>(OCR_RESULT_KEY)?.let { text ->
-            val result = text.replace(" ", "")
-                .replace(",", ".")
-                .replace(")", "")
+            val correctedText =
+                AutoTextCorrector(nutritionNameList, 2).correctText(text.replace("%", "% "))
 
-            setFoodInfo(result)
+            setFoodInfo(
+                correctedText.replace(" ", "")
+                    .replace(",", ".")
+                    .replace(")", ""),
+            )
         }
         savesStateHandle.get<FoodInfo?>(API_RESULT_KEY)?.let {
             setFoodInfo(it)
@@ -123,6 +127,18 @@ class EnrollTextInputViewModel @Inject constructor(
         const val API_RESULT_KEY = "API_RESULT_KEY"
         const val BITMAP_PICTURE_KEY = "BITMAP_PICTURE_KEY"
         const val URI_PICTURE_KEY = "URI_PICTURE_KEY"
+
+        val nutritionNameList = listOf(
+            "칼로리",
+            "탄수화물",
+            "단백질",
+            "당류",
+            "지방",
+            "포화지빙",
+            "트랜스지방",
+            "콜레스트롤",
+            "나트륨",
+        )
     }
 }
 

diff --git a/settings.gradle b/settings.gradle
@@ -20,3 +20,4 @@ include ':feat:personal'
 include ':feat:history'
 include ':feat:capture'
 include ':common'
+include ':autotextcorrection'