diff options
Diffstat (limited to 'Services/OcrService.cs')
| -rw-r--r-- | Services/OcrService.cs | 6 |
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(); |
