cognitive_vision 0.2.1 → 0.3.0

checksums.yaml

@@ -1,15 +1,15 @@
 ---
 !binary "U0hBMQ==":
   metadata.gz: !binary |-
-    YTIzZTE2NDNlNTRjOWZmYzU4OTEyY2E1ZWVkNjIzOGE2NGM1MTU3NA==
+    NDI3ZDdhODY3MDVmNTY0ZmU0Zjc3OGQ2N2Y0NGU2ZmViZDZkOTFjMw==
   data.tar.gz: !binary |-
-    ZWI2ODllZDljYzY5NmYzZmE3NGY2MThkNzczOTU1YzlmNjEzODdkNA==
+    OGUzMjVhODc0YzY1YjNhYjc4YmNmMmI1ZGQ0ZDZhOTUwYjk1MjAwMA==
 SHA512:
   metadata.gz: !binary |-
-    NzdiZjIwNGFjY2RiNjJjM2I3NWVkYTYwNDNlZmQyMTc5NDkzNDRjNGNhMjEx
-    M2Q1OGYyNWM2ODcwOTZhOTMwODUxZTQ4OWMyYWY4MDE1NzZkOTU1Y2U5MTRh
-    NjU1MGQyMDM4OTI5ZTNhZTEwN2I2YThmNjViMzRlODY3M2I4N2I=
+    OGFmMzFmMDVjMjkyYTE2MjcwZjdmMWY0OWI5NmYyODU4Y2JkYzEyOTA0YjU3
+    YjljOWQ2ODg5N2RmYjE1OWQ3OWJiZTU2YTkzYjY4OWMyZGI5MWI1ZTFlODdh
+    Mjc5NmQ3YTg3MmM0NDkyYWU0NTc1NjBjMzk4Mzk1MzVmZDQwOWE=
   data.tar.gz: !binary |-
-    ZDJjZjQwZjcwYWEyMTdiNjJiOTNmNjFmNmMwNWI2OWEwNDMzNzIwNmE1NDJm
-    YTYyY2M2MTkzZGJlNjczNTI0NjAzNGRlYmQzOTA5MWNhYzVmMTJhYzI3ODc2
-    N2VmZWExODZlY2VlOTlhY2MxOTEzMDc0MGNhNTg2NWZlMDM4MzA=
+    MzdmYzFkNGYzYjVlMzRiZGQ5ZmI2OTVjMTQwZmQzYTJlNTYxNGNkMjM4MjAx
+    OWU5ZDdjN2UwZDliN2MyZDk3ZDMwNjQ1MDE5MWUwMzI2YTgyZDZiZTU5ZDY1
+    NWM0YTgyZjE4ZjIxYzEwNTkzZjM2NWY0ZDdlNGQ2ZWY2ZmE5OTY=

data/lib/cognitive_vision.rb

@@ -1,6 +1,7 @@
 require 'cognitive_vision/version'
 require 'cognitive_vision/configuration'
 require 'cognitive_vision/connection'
+require 'cognitive_vision/adult'
 require 'cognitive_vision/face'
 require 'cognitive_vision/analyze_response'
 require 'cognitive_vision/analyze_image'

data/lib/cognitive_vision/adult.rb

@@ -0,0 +1,20 @@
+module CognitiveVision
+  class Adult
+    attr_reader :adult_score, :racy_score
+
+    def initialize(options = {})
+      @adult_content = options.fetch(:adult_content)
+      @adult_score   = options.fetch(:adult_score)
+      @racy_content  = options.fetch(:racy_content)
+      @racy_score    = options.fetch(:racy_score)
+    end
+
+    def adult_content?
+      @adult_content
+    end
+
+    def racy_content?
+      @racy_content
+    end
+  end
+end

data/lib/cognitive_vision/analyze_image.rb

@@ -7,9 +7,10 @@ module CognitiveVision
     class InvalidImageUrlError  < StandardError; end
     class UnknownError          < StandardError; end
 
-    def self.analyze_image(image_url)
+    def self.analyze_image(image_url, types = [:faces])
+      types    = [types] unless types.is_a?(Array)
       body     = { 'url' => image_url }
-      params   = { 'visualFeatures' => 'Faces' }
+      params   = { 'visualFeatures' => types.join(',') }
       response = Connection.new.post('/analyze', params, body)
 
       treat_errors(response) if response.code != 200

data/lib/cognitive_vision/analyze_response.rb

@@ -1,14 +1,21 @@
 module CognitiveVision
   class AnalyzeResponse
-    attr_reader :faces
+    attr_reader :adult, :faces
 
     def initialize(options = {})
+      @adult = options.fetch(:adult, {})
       @faces = options.fetch(:faces, [])
     end
 
     def self.parse(response_hash)
-      faces = response_hash['faces'].map { |face| Face.new(gender: face['gender'], age: face['age']) }
-      new(faces: faces)
+      faces = (response_hash['faces'] || []).map { |face| Face.new(gender: face['gender'], age: face['age']) }
+      adult = if response_hash['adult']
+                adult_response = response_hash['adult']
+                Adult.new(adult_content: adult_response['isAdultContent'],
+                          racy_content: adult_response['isRacyContent'], adult_score: adult_response['adultScore'],
+                          racy_score: adult_response['racyScore'])
+              end
+      new(faces: faces, adult: adult)
     end
   end
 end

data/lib/cognitive_vision/version.rb

@@ -1,3 +1,3 @@
 module CognitiveVision
-  VERSION = '0.2.1'.freeze
+  VERSION = '0.3.0'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: cognitive_vision
 version: !ruby/object:Gem::Version
-  version: 0.2.1
+  version: 0.3.0
 platform: ruby
 authors:
 - Carlos Ribeiro
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-09-05 00:00:00.000000000 Z
+date: 2016-09-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -170,6 +170,7 @@ files:
 - bin/setup
 - cognitive_vision.gemspec
 - lib/cognitive_vision.rb
+- lib/cognitive_vision/adult.rb
 - lib/cognitive_vision/analyze_image.rb
 - lib/cognitive_vision/analyze_response.rb
 - lib/cognitive_vision/configuration.rb