full working hotdog analyzer

Author: codediodeio

functions/src/index.ts

@@ -9,33 +9,36 @@ admin.initializeApp(functions.config().firebase);
 import * as vision from '@google-cloud/vision';
 const visionClient =  new vision.ImageAnnotatorClient();
 
+// Dedicated bucket for cloud function invocation
 const bucketName = 'firestarter-96e46-vision';
 
-
-
 export const imageTagger = functions.storage
 
     .bucket(bucketName)
     .object()
     .onChange( async event => {
 
+            // File data
             const object = event.data;
             const filePath = object.name;   
 
+            // Location of saved file in bucket
             const imageUri = `gs://${bucketName}/${filePath}`;
 
+            // Firestore docID === file name
             const docId = filePath.split('.jpg')[0];
-            console.log(2, docId)
+
             const docRef  = admin.firestore().collection('photos').doc(docId);
-    
 
+            // Await the cloud vision response
             const results = await visionClient.labelDetection(imageUri);
-            const labels = results[0].labelAnnotations.map(obj => obj.description);
 
-            console.log(3, results)
+            // Map the data to desired format
+            const labels = results[0].labelAnnotations.map(obj => obj.description);
+            const hotdog = labels.includes('hot dog')
 
 
-            return docRef.set({ labels })
+            return docRef.set({ hotdog, labels })
 
 
 });