add some testing code for auto recognition

Author: cccraig

deep.py

@@ -0,0 +1,27 @@
+from deepface import DeepFace
+from PIL import Image
+import numpy as np
+import pandas as pd
+
+pd.set_option('display.max_colwidth', -1)
+
+db_path = "/home/collin/Web/piwigo/plugins/MugShot/training"
+
+target_img = "/home/collin/Downloads/IMG_8069.jpg"
+
+backends = ['opencv', 'ssd', 'dlib', 'mtcnn', 'retinaface']
+
+#face detection and alignment
+detected_face = DeepFace.detectFace(target_img, detector_backend = backends[4])
+
+#face recognition
+#df = DeepFace.find(target_img, db_path, detector_backend = backends[4])
+
+#Convert array to image
+images = Image.fromarray(detected_face)
+  
+# Display image
+for image in images:
+    image.show()
+
+print(detected_face)

include/capture.php

@@ -42,7 +42,7 @@ function book_mugshots($data, &$service) {
   }
 
   $imageId = pwg_db_real_escape_string($data['imageId']);
-  $plugin_config = unserialize(conf_get_param(MUGSHOT_ID));
+  $plugin_config = conf_get_param(MUGSHOT_ID);
 
   unset($data['imageId']);
   $imageIdTagIdInsertionString = '';             // 
@@ -70,9 +70,9 @@ function book_mugshots($data, &$service) {
 
     // If it's a brand new tag, we won't have sent a tag ID back with the data.
     $newTagId = ($existingTagId == -1) ? tag_id_from_tag_name($labeledTagName) : $existingTagId;
-
+var_dump($plugin_config);
     // Create or remove the training thumbnails, depending on webmaster settings.
-    if ($plugin_config['autotag']) {
+    if ($plugin_config['autotag'] == '1') {
       $sql = "SELECT * FROM `". IMAGES_TABLE . "` WHERE `id`=".$imageId.";";
 
       $imgData = pwg_db_fetch_assoc(pwg_query($sql));

include/helpers.php

@@ -25,6 +25,7 @@ function create_tag_group() {
     if ($result == 0)
     {
       $makeTaggerGroupQuery = "INSERT INTO " . GROUPS_TABLE . ' (name) VALUES ("Taggers")';
+
       pwg_query($makeTaggerGroupQuery);
 
       $result = fetch_sql($checkTaggerGroupQuery, 'id', false);

main.inc.php

@@ -29,6 +29,7 @@
 define('MUGSHOT_BASE_URL',   get_root_url() . 'admin.php?page=plugin-' . MUGSHOT_ID);
 define('MUGSHOT_VERSION', '2.0.2');
 define('MUGSHOT_TABLE', '`face_tag_positions`');
+define('MUGSHOT_QUEUE_TABLE', '`face_tag_queue`');
 
 
 /*
@@ -43,6 +44,7 @@
 add_event_handler('init', 'mugshot_lang_init');
 add_event_handler('loc_begin_page_header', 'mugshot_files', 40, 2);
 add_event_handler('loc_end_picture', 'mugshot_button');
+add_event_handler('loc_end_add_uploaded_file', 'mugshot_queue_for_processing');
 
 /*
  * Include custom helper functions
@@ -96,6 +98,30 @@
   define('MUGSHOT_USER_ADMIN', false);
 }
 
+/*
+ * Test to see if the upload fires as expected
+ */
+function create_facetag_tableb() {
+
+
+}
+
+function mugshot_queue_for_processing($image_info) {
+  $createTableQuery = 'CREATE TABLE IF NOT EXISTS `face_tag_queue` (
+    `image_id` mediumint(8) unsigned NOT NULL default "0",
+    `stuff` varchar(5000) NOT NULL default "0",
+    PRIMARY KEY (`image_id`)
+  )';
+
+  pwg_query($createTableQuery);
+
+  $item = serialize($image_info);
+  $id = $image_info['id'];
+
+  $insertImageQueueQuery = "INSERT INTO `face_tag_queue` (image_id, stuff) VALUES ('$id','$item')";
+
+  pwg_query($insertImageQueueQuery);
+}
 
 /*
  * Loads translations
@@ -244,7 +270,7 @@ function insert_tag_list() {
 
   global $template;
 
-  $plugin_config = unserialize(conf_get_param(MUGSHOT_ID));
+  $plugin_config = conf_get_param(MUGSHOT_ID);
 
   $max_tags = $plugin_config['max_tags'] ?? 500;