v0.11av0.11a

1 files changed, 5 insertions, 1 deletions

diff --git a/Services/OcrService.cs b/Services/OcrService.cs
index 40905aa..d43db2e 100644
--- a/Services/OcrService.cs
+++ b/Services/OcrService.cs
@@ -8,6 +8,8 @@ using Tesseract;
 namespace HyperBooru.Services;
 
 public class OcrService : IHostedService {
+    private readonly string[] InvalidMimeTypes = [ "image/heic", "image/webp" ];
+
     private readonly TimeSpan ProcessInterval = TimeSpan.FromMinutes(30);
     private readonly TimeSpan StartupDelay    = TimeSpan.FromSeconds(30);
 
@@ -65,10 +67,12 @@ public class OcrService : IHostedService {
 
         using var db = dbFactory.CreateDbContext();
         Guid[] guids = db.Media
+            .AsNoTracking()
             .Include(m => m.CurrentUploadedFile)
             .Include(m => m.OcrData)
             .Where(m => m.OcrData == null)
-            .Where(m => m.CurrentUploadedFile.MimeType.Contains("image/"))
+            .Where(m => m.CurrentUploadedFile!.MimeType.Contains("image/"))
+            .Where(m => !InvalidMimeTypes.Contains(m.CurrentUploadedFile!.MimeType))
             .Select(m => m.Guid)
             .ToArray();
         db.Dispose();