diff --git a/papis_extract/annotation.py b/papis_extract/annotation.py index 54b216d..2111f1b 100644 --- a/papis_extract/annotation.py +++ b/papis_extract/annotation.py @@ -6,8 +6,7 @@ import chevron import papis.config from papis.document import Document -TEXT_SIMILARITY_MINIMUM = 0.75 -COLOR_SIMILARITY_MINIMUM = 0.833 +COLOR_SIMILARITY_MINIMUM_FALLBACK = 0.833 COLORS: dict[str, tuple[float, float, float]] = { "blue": (0, 0, 1), @@ -38,9 +37,7 @@ class Annotation: page: int = 0, tag: str = "", type: str = "Highlight", - minimum_similarity_color: float = 1.0, ) -> None: - self.minimum_similarity_color = minimum_similarity_color self.file = file self._color = color self.content = content @@ -86,9 +83,9 @@ class Annotation: annot_colors = self.color or (0.0, 0.0, 0.0) nearest = None minimum_similarity = ( - papis.config.getfloat("minimum_similarity_color", "plugins.extract") or 1.0 + papis.config.getfloat("minimum_similarity_color", "plugins.extract") + or COLOR_SIMILARITY_MINIMUM_FALLBACK ) - minimum_similarity = self.minimum_similarity_color for name, values in COLORS.items(): similarity_ratio = self._color_similarity_ratio(values, annot_colors) if similarity_ratio >= minimum_similarity: