Add new probability value to photo view

LibrePhotos · Oct 12, 2024 · fe7c12c · fe7c12c
1 parent 8ca663f
commit fe7c12c
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/api/serializers/photos.py b/api/serializers/photos.py
@@ -300,6 +300,19 @@ def get_people(self, obj) -> list:
                         else ("classification" if f.classification_person else "")
                     )
                 ),
+                "probability": (
+                    1
+                    if f.person
+                    else (
+                        f.cluster_probability
+                        if f.cluster_person
+                        else (
+                            f.classification_probability
+                            if f.classification_person
+                            else 0
+                        )
+                    )
+                ),
                 "face_url": f.image.url,
                 "face_id": f.id,
             }