feat: ImagePHashDatabase

ui/implements/settings/settingsOcr.py

@@ -45,6 +45,17 @@ class SettingsOcr(SettingsBaseWidget):
             self.knnModelFileResetButton,
         )
 
+        if self.settings.b30KnnModelFile():
+            self.b30KnnModelFileValueWidget.selectFile(self.settings.b30KnnModelFile())
+        self.b30KnnModelFileValueWidget.filesSelected.connect(self.setB30KnnModelFile)
+        self.b30KnnModelFileResetButton.clicked.connect(self.resetB30KnnModelFile)
+        self.insertItem(
+            "b30KnnModelFile",
+            self.b30KnnModelFileLabel,
+            self.b30KnnModelFileValueWidget,
+            self.b30KnnModelFileResetButton,
+        )
+
         if self.settings.siftDatabaseFile():
             self.siftDatabaseFileValueWidget.selectFile(
                 self.settings.siftDatabaseFile()
@@ -58,6 +69,21 @@ class SettingsOcr(SettingsBaseWidget):
             self.siftDatabaseFileResetButton,
         )
 
+        if self.settings.phashDatabaseFile():
+            self.phashDatabaseFileValueWidget.selectFile(
+                self.settings.phashDatabaseFile()
+            )
+        self.phashDatabaseFileValueWidget.filesSelected.connect(
+            self.setPHashDatabaseFile
+        )
+        self.phashDatabaseFileResetButton.clicked.connect(self.resetPHashDatabaseFile)
+        self.insertItem(
+            "phashDatabaseFile",
+            self.phashDatabaseFileLabel,
+            self.phashDatabaseFileValueWidget,
+            self.phashDatabaseFileResetButton,
+        )
+
     def setDevicesJson(self):
         selectedFile = self.devicesJsonValueWidget.selectedFiles()
         if selectedFile and selectedFile[0]:
@@ -79,8 +105,7 @@ class SettingsOcr(SettingsBaseWidget):
         self.settings.resetDevicesJsonFile()
 
     def setDeviceUuid(self):
-        device = self.deviceUuidValueWidget.currentData()
-        if device:
+        if device := self.deviceUuidValueWidget.currentData():
             self.settings.setDeviceUuid(device.uuid)
 
     def resetDeviceUuid(self):
@@ -97,6 +122,16 @@ class SettingsOcr(SettingsBaseWidget):
         self.knnModelFileValueWidget.reset()
         self.settings.resetKnnModelFile()
 
+    def setB30KnnModelFile(self):
+        selectedFile = self.b30KnnModelFileValueWidget.selectedFiles()
+        if selectedFile and selectedFile[0]:
+            file = selectedFile[0]
+            self.settings.setB30KnnModelFile(file)
+
+    def resetB30KnnModelFile(self):
+        self.b30KnnModelFileValueWidget.reset()
+        self.settings.resetB30KnnModelFile()
+
     def setSiftDatabaseFile(self):
         selectedFile = self.siftDatabaseFileValueWidget.selectedFiles()
         if selectedFile and selectedFile[0]:
@@ -107,6 +142,16 @@ class SettingsOcr(SettingsBaseWidget):
         self.siftDatabaseFileValueWidget.reset()
         self.settings.resetSiftDatabaseFile()
 
+    def setPHashDatabaseFile(self):
+        selectedFile = self.phashDatabaseFileValueWidget.selectedFiles()
+        if selectedFile and selectedFile[0]:
+            file = selectedFile[0]
+            self.settings.setPHashDatabaseFile(file)
+
+    def resetPHashDatabaseFile(self):
+        self.phashDatabaseFileValueWidget.reset()
+        self.settings.resetPHashDatabaseFile()
+
     def setupUi(self, *args):
         self.devicesJsonLabel = QLabel(self)
         self.devicesJsonValueWidget = FileSelector(self)
@@ -120,10 +165,18 @@ class SettingsOcr(SettingsBaseWidget):
         self.knnModelFileValueWidget = FileSelector(self)
         self.knnModelFileResetButton = QPushButton(self)
 
+        self.b30KnnModelFileLabel = QLabel(self)
+        self.b30KnnModelFileValueWidget = FileSelector(self)
+        self.b30KnnModelFileResetButton = QPushButton(self)
+
         self.siftDatabaseFileLabel = QLabel(self)
         self.siftDatabaseFileValueWidget = FileSelector(self)
         self.siftDatabaseFileResetButton = QPushButton(self)
 
+        self.phashDatabaseFileLabel = QLabel(self)
+        self.phashDatabaseFileValueWidget = FileSelector(self)
+        self.phashDatabaseFileResetButton = QPushButton(self)
+
         super().setupUi(self)
         self.retranslateUi()
 
@@ -142,6 +195,12 @@ class SettingsOcr(SettingsBaseWidget):
         self.knnModelFileLabel.setText(QCoreApplication.translate("Settings", "ocr.knnModelFile.label"))
         self.knnModelFileResetButton.setText(QCoreApplication.translate("Settings", "resetButton"))
 
+        self.b30KnnModelFileLabel.setText(QCoreApplication.translate("Settings", "ocr.b30KnnModelFile.label"))
+        self.b30KnnModelFileResetButton.setText(QCoreApplication.translate("Settings", "resetButton"))
+
         self.siftDatabaseFileLabel.setText(QCoreApplication.translate("Settings", "ocr.siftDatabaseFile.label"))
         self.siftDatabaseFileResetButton.setText(QCoreApplication.translate("Settings", "resetButton"))
+
+        self.phashDatabaseFileLabel.setText(QCoreApplication.translate("Settings", "ocr.phashDatabaseFile.label"))
+        self.phashDatabaseFileResetButton.setText(QCoreApplication.translate("Settings", "resetButton"))
         # fmt: on