cloud_vision 1.0.3 → 1.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f96c737ee30e5451e3c9a8ed5af6d6052de76f61
-  data.tar.gz: 1d193a4bfa49f637616383b4a2468cc1d386df3d
+  metadata.gz: b9f762023e7140a0e2065bb1a99eca0e2abfec0c
+  data.tar.gz: bd64be2c0f64957c6d003884c6850dbcb218bb18
 SHA512:
-  metadata.gz: f94ddf2341bc6ec97631a2f1e6ec8aa765d13b7e98965aa36c410241ae0cc1d9adb70a04cc1609ed439a3951cc05a1694f3d30e6ba92a6eb5264e5cc64575506
-  data.tar.gz: f5e8dfca2d49756af027377640487ae19556b48d02b3025e73a62aad8592f9e5211bbd0687f5a317a72464fab1308493e7bab2079a49e1b0911ba52cf9fe2345
+  metadata.gz: 08f5fba82c324a7856d031a398716e79af8f3ce07e3b80a5928feb9d4aeb120584031755640a81ad35c5c45fede20f112f18b87fe003cd2f7d9cf0866ae64610
+  data.tar.gz: e7c29cb28fc49b26af410bc20cc0ba5586bc76aedce0ac37990fbc8b822c557c1ee1623266cf2e2583c1a15754341854a58a6131aea41ff19ac536bc1d1dff2e

data/lib/cloud_vision.rb

@@ -0,0 +1,12 @@
+module CloudVision
+	SAFETY_TEST			= :safety
+	FACIAL_TEST			= :faces
+	LABEL_TEST				= :labels
+	TEXT_TEST				= :text
+	LOGO_TEST				= :logos
+	LANDMARK_TEST		= :landmarks
+	PROPERTIES_TEST	= :properties
+
+	require_relative 'cloud_vision/api'
+	require_relative 'cloud_vision/parser'
+end

data/lib/cloud_vision/api.rb

@@ -0,0 +1,94 @@
+require 'net/https'
+require 'base64'
+require 'json'
+
+module CloudVision
+	class Api
+		def initialize( api_key )
+			@uri = URI( "https://vision.googleapis.com/v1/images:annotate?key=#{api_key}" )
+
+			@cloud_vision = Net::HTTP.new( @uri.host, @uri.port )
+			@cloud_vision.use_ssl = true
+		end
+
+		def analyze( image_file, tests, options={} )
+			post_data = {
+				requests: [ {
+					image: {
+						content: Base64.strict_encode64( image_file.read )
+					},
+					features: build_feature_requests( tests )
+				} ]
+			}
+
+			response = send_request( post_data )
+
+			return_format = response.body
+
+			if return_format && !return_format.empty?
+				return_format = JSON.parse( return_format )
+				return_format = return_format[ 'responses' ].first
+			end
+
+			return_format
+		end
+
+		private
+
+		def send_request( body )
+			request = Net::HTTP::Post.new( @uri, {'Content-Type' => 'application/json'} )
+			request.body = JSON.generate( body )
+
+			@cloud_vision.request( request )
+		end
+
+		def build_feature_requests( tests )
+			features = []
+
+			if !tests.is_a?( Array )
+				tests = [ tests ]
+			end
+
+			tests.each do |detection_test|
+				filters = []
+
+				detection_test = detection_test.to_sym
+				if detection_test == FACIAL_TEST || detection_test == :all
+					filters << { type: 'FACE_DETECTION', max_results: 10 }
+				end
+
+				if detection_test == TEXT_TEST || detection_test == :all
+					filters << { type: 'TEXT_DETECTION', max_results: 10 }
+				end
+
+				if detection_test == LABEL_TEST || detection_test == :all
+					filters << { type: 'LABEL_DETECTION', max_results: 10 }
+				end
+
+				if detection_test == LOGO_TEST || detection_test == :all
+					filters << { type: 'LOGO_DETECTION', max_results: 10 }
+				end
+
+				if detection_test == LANDMARK_TEST || detection_test == :all
+					filters << { type: 'LANDMARK_DETECTION', max_results: 10 }
+				end
+
+				if detection_test == SAFETY_TEST || detection_test == :all
+					filters << { type: 'SAFE_SEARCH_DETECTION', max_results: 1 }
+				end
+
+				if detection_test == PROPERTIES_TEST || detection_test == :all
+					filters << { type: 'IMAGE_PROPERTIES', max_results: 1 }
+				end
+
+				if filters.empty?
+					raise( "Unrecognized test: #{detection_test.inspect}")
+				end
+
+				features += filters
+			end
+
+			features
+		end
+	end
+end

data/lib/cloud_vision/parser.rb

@@ -0,0 +1,188 @@
+require 'json'
+
+module CloudVision
+	class Parser
+		RESPONSE_LABELS = {
+			SAFETY_TEST => 'safeSearchAnnotation',
+			FACIAL_TEST => 'faceAnnotations',
+			LABEL_TEST => 'labelAnnotations',
+			TEXT_TEST => 'textAnnotations',
+			LOGO_TEST => 'logoAnnotations',
+			LANDMARK_TEST => 'landmarkAnnotations',
+			PROPERTIES_TEST => 'imagePropertiesAnnotation'
+		}.freeze
+
+		RESPONSE_SCALE = {
+			'VERY_UNLIKELY' => -2,
+			'UNLIKELY' => -1,
+			'UNKNOWN' => 0,
+			'POSSIBLE' => 1,
+			'LIKELY' => 2,
+			'VERY_LIKELY' => 3
+		}.freeze
+
+		def self.parse_analysis( analysis )
+			parsed_data = {}
+
+			if analysis && analysis.is_a?( String )
+				analysis = JSON.parse( analysis )
+			elsif !analysis || !analysis.is_a?( Hash )
+				raise( 'Parsing Failed. Analysis must be valid JSON' )
+			end
+
+			if analysis[ RESPONSE_LABELS[SAFETY_TEST] ]
+				new_data = parse_safety( analysis )
+				parsed_data[ SAFETY_TEST ] = new_data
+			end
+
+			if analysis[ RESPONSE_LABELS[FACIAL_TEST] ]
+				new_data = parse_faces( analysis )
+				parsed_data[ FACIAL_TEST ] = new_data
+			end
+
+			if analysis[ RESPONSE_LABELS[LABEL_TEST] ]
+				new_data = parse_entity( analysis, LABEL_TEST )
+				parsed_data[ LABEL_TEST ] = new_data
+			end
+
+			if analysis[ RESPONSE_LABELS[TEXT_TEST] ]
+				new_data = parse_entity( analysis, TEXT_TEST )
+				parsed_data[ TEXT_TEST ] = new_data
+			end
+
+			if analysis[ RESPONSE_LABELS[LOGO_TEST] ]
+				new_data = parse_entity( analysis, LOGO_TEST )
+				parsed_data[ LOGO_TEST ] = new_data
+			end
+
+			if analysis[ RESPONSE_LABELS[LANDMARK_TEST] ]
+				new_data = parse_entity( analysis, LANDMARK_TEST )
+				parsed_data[ LANDMARK_TEST ] = new_data
+			end
+
+			if analysis[ RESPONSE_LABELS[PROPERTIES_TEST] ]
+				new_data = parse_properties( analysis )
+				parsed_data[ PROPERTIES_TEST ] = new_data
+			end
+
+			parsed_data
+		end
+
+		def self.parse_safety( analysis )
+			processed_data = {}
+
+			if analysis[ RESPONSE_LABELS[SAFETY_TEST] ]
+	      analysis[ RESPONSE_LABELS[SAFETY_TEST] ].each do |risk_test, value|
+	        processed_data[ risk_test.to_sym ] = RESPONSE_SCALE[ value ]
+	      end
+			end
+
+			processed_data
+		end
+
+		def self.parse_faces( analysis )
+			processed_data = []
+
+	    if analysis[ RESPONSE_LABELS[FACIAL_TEST] ]
+	      analysis[ RESPONSE_LABELS[FACIAL_TEST] ].each do |face_data|
+	        annotations = {}
+
+	        # Only include face data for users with 4 bounding points
+	        if face_data['boundingPoly']['vertices'].length == 4
+	          top_left = face_data['boundingPoly']['vertices'][ 0 ]
+	          bottom_right = face_data['boundingPoly']['vertices'][ 2 ]
+
+	          annotations[ :face ] = {
+	            x: top_left['x'],
+	            y: top_left['y'],
+	            width: bottom_right['x'].to_i - top_left['x'].to_i,
+	            height: bottom_right['y'].to_i - top_left['y'].to_i
+	          }
+	        end
+
+	        # Find the confidence in the image parse
+	        annotations[ :confidence ] = face_data[ 'detectionConfidence' ]
+
+	        # Store the image quality metrics
+	        annotations[ :quality ] = {
+	          under_exposed: RESPONSE_SCALE[ face_data['underExposedLikelihood'] ],
+	          blurred: RESPONSE_SCALE[ face_data['blurredLikelihood'] ],
+	          headwear: RESPONSE_SCALE[ face_data['headwearLikelihood'] ]
+	        }
+
+	        # Parse the sentiment analysis data
+	        annotations[ :sentiment ] = {
+	          joy: RESPONSE_SCALE[ face_data['joyLikelihood'] ],
+	          sorrow: RESPONSE_SCALE[ face_data['sorrowLikelihood'] ],
+	          anger: RESPONSE_SCALE[ face_data['angerLikelihood'] ],
+	          surprise: RESPONSE_SCALE[ face_data['surpriseLikelihood'] ]
+	        }
+
+	        processed_data << annotations
+	      end
+	    end
+
+			processed_data
+		end
+
+		def self.parse_entity( analysis, target )
+			processed_data = []
+
+			if analysis[ RESPONSE_LABELS[target] ]
+				analysis[ RESPONSE_LABELS[target] ].each do |label_data|
+					entity_data = {
+						description: label_data[ 'description' ],
+						locale: label_data[ 'locale' ],
+						score: label_data[ 'score' ].to_f * 100,
+						confidence: label_data[ 'confidence' ].to_f * 100,
+						relevance: label_data[ 'topicality' ].to_f * 100
+					}
+
+					# Some entity responses do not report confidence or relevance data
+					if entity_data[ :confidence ] == 0 && entity_data[ :relevance ] == 0
+						entity_data.delete( :relevance )
+						entity_data.delete( :confidence )
+					end
+
+					# Text filters include no score either.
+					if entity_data[ :score ] == 0
+						entity_data.delete( :score )
+					end
+
+					# Landmarks include no locale either.
+					if !entity_data[ :locale ]
+						entity_data.delete( :locale )
+					end
+
+					processed_data << entity_data
+				end
+			end
+
+			processed_data
+		end
+
+		def self.parse_properties( analysis )
+			processed_data = {}
+
+			properties_data = analysis[ RESPONSE_LABELS[PROPERTIES_TEST] ]
+			if properties_data && properties_data[ 'dominantColors' ]
+				properties_data = properties_data[ 'dominantColors' ][ 'colors' ]
+				processed_data[ :colors ] = []
+
+				properties_data.each do |color_data|
+					red = color_data[ 'color' ][ 'red' ].to_i.to_s( 16 )
+					green = color_data[ 'color' ][ 'green' ].to_i.to_s( 16 )
+					blue = color_data[ 'color' ][ 'blue' ].to_i.to_s( 16 )
+
+					processed_data[ :colors ] << {
+						hex: "##{red}#{green}#{blue}".upcase,
+						score: color_data[ 'score' ].to_f * 100,
+						percentage: color_data[ 'pixelFraction' ].to_f * 100
+					}
+				end
+			end
+
+			processed_data
+		end
+	end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cloud_vision
 version: !ruby/object:Gem::Version
-  version: 1.0.3
+  version: 1.0.4
 platform: ruby
 authors:
 - Ben Roux
@@ -15,7 +15,10 @@ email: liquid.ise@gmail.com
 executables: []
 extensions: []
 extra_rdoc_files: []
-files: []
+files:
+- lib/cloud_vision.rb
+- lib/cloud_vision/api.rb
+- lib/cloud_vision/parser.rb
 homepage: https://github.com/liquidise/cloud_vision_gem
 licenses:
 - MIT