From dda7be99ebe65970d2143b1b87db24516c538b61 Mon Sep 17 00:00:00 2001
From: leccelecce <24962424+leccelecce@users.noreply.github.com>
Date: Wed, 19 Mar 2025 12:44:07 +0000
Subject: [PATCH] Face rec: only consider webp files in /faces and
 handle_request (#17244)

* Face rec: only consider webp files

* Allow png/jpg/jpeg as well as webp
---
 frigate/api/classification.py             | 5 ++++-
 frigate/data_processing/real_time/face.py | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/frigate/api/classification.py b/frigate/api/classification.py
index df804f34a..b9d714ca3 100644
--- a/frigate/api/classification.py
+++ b/frigate/api/classification.py
@@ -39,7 +39,10 @@ def get_faces():
         face_dict[name] = []
 
         for file in sorted(
-            os.listdir(face_dir),
+            filter(
+                lambda f: (f.lower().endswith((".webp", ".png", ".jpg", ".jpeg"))),
+                os.listdir(face_dir),
+            ),
             key=lambda f: os.path.getctime(os.path.join(face_dir, f)),
             reverse=True,
         ):
diff --git a/frigate/data_processing/real_time/face.py b/frigate/data_processing/real_time/face.py
index acb891449..102913442 100644
--- a/frigate/data_processing/real_time/face.py
+++ b/frigate/data_processing/real_time/face.py
@@ -484,7 +484,7 @@ class FaceRealTimeProcessor(RealTimeProcessorApi):
                 shutil.move(current_file, new_file)
 
                 files = sorted(
-                    os.listdir(folder),
+                    filter(lambda f: (f.endswith(".webp")), os.listdir(folder)),
                     key=lambda f: os.path.getctime(os.path.join(folder, f)),
                     reverse=True,
                 )