ImagenetClassifier.php

<?php

/*
 * Copyright (c) 2021-2022 The Recognize contributors.
 * This file is licensed under the Affero General Public License version 3 or later. See the COPYING file.
 */
declare(strict_types=1);
namespace OCA\Recognize\Classifiers\Images;

use OCA\Recognize\Classifiers\Classifier;
use OCA\Recognize\Service\Logger;
use OCA\Recognize\Service\QueueService;
use OCA\Recognize\Service\TagManager;
use OCP\AppFramework\Services\IAppConfig;
use OCP\Files\IRootFolder;
use OCP\IPreview;
use OCP\ITempManager;
use Override;

final class ImagenetClassifier extends Classifier {
	public const IMAGE_TIMEOUT = 480; // seconds
	public const IMAGE_PUREJS_TIMEOUT = 600; // seconds
	public const MODEL_NAME = 'imagenet';

	private TagManager $tagManager;
	protected QueueService $queue;

	public function __construct(Logger $logger, IAppConfig $config, TagManager $tagManager, QueueService $queue, IRootFolder $rootFolder, ITempManager $tempManager, IPreview $previewProvider) {
		parent::__construct($logger, $config, $rootFolder, $queue, $tempManager, $previewProvider);
		$this->tagManager = $tagManager;
		$this->queue = $queue;
	}

	#[Override]
	public function classify(array $queueFiles): void {
		if ($this->config->getAppValueString('tensorflow.purejs', 'false', lazy: true) === 'true') {
			$timeout = self::IMAGE_PUREJS_TIMEOUT;
		} else {
			$timeout = self::IMAGE_TIMEOUT;
		}
		$classifierProcess = $this->classifyFiles(self::MODEL_NAME, $queueFiles, $timeout);

		/** @var \OCA\Recognize\Db\QueueFile $queueFile */
		/** @var list<string> $results */
		foreach ($classifierProcess as $queueFile => $results) {
			$this->tagManager->assignTags($queueFile->getFileId(), $results);
			$landmarkTags = array_filter($results, static function ($tagName) {
				return in_array($tagName, LandmarksClassifier::PRECONDITION_TAGS);
			});
			$this->config->setAppValueString(self::MODEL_NAME.'.status', 'true', lazy: true);
			$this->config->setAppValueString(self::MODEL_NAME.'.lastFile', (string)time(), lazy: true);

			if (count($landmarkTags) > 0) {
				try {
					$this->queue->insertIntoQueue(LandmarksClassifier::MODEL_NAME, $queueFile);
				} catch (\Throwable $e) {
					$this->logger->error('Cannot insert file into queue', ['exception' => $e]);
				}
			}
		}
	}
}