Tone Analyzer: Add support for JSON and HTML input to tone() method

Author: jeffpk62

examples/tone_analyzer_v3.py

@@ -1,13 +1,45 @@
 import json
+import os
+from os.path import join, dirname
 from watson_developer_cloud import ToneAnalyzerV3
 
-
 tone_analyzer = ToneAnalyzerV3(
     username='YOUR SERVICE USERNAME',
     password='YOUR SERVICE PASSWORD',
-    version='2016-02-11')
-
-print(json.dumps(tone_analyzer.tone(text='I am very happy'), indent=2))
+    version='2016-05-19')
 
+print("\ntone_chat() example 1:\n")
 utterances = [{'text': 'I am very happy', 'user': 'glenn'}]
 print(json.dumps(tone_analyzer.tone_chat(utterances), indent=2))
+
+print("\ntone() example 1:\n")
+print(json.dumps(tone_analyzer.tone(text='I am very happy. It is a good day.'),
+                 indent=2))
+
+print("\ntone() example 2:\n")
+with open(join(dirname(__file__),
+               '../resources/tone-example.json')) as tone_json:
+    tone = tone_analyzer.tone(json.load(tone_json)['text'],
+                              tones='emotion')
+print(json.dumps(tone, indent=2))
+
+print("\ntone() example 3:\n")
+with open(join(dirname(__file__),
+               '../resources/tone-example.json')) as tone_json:
+    tone = tone_analyzer.tone(json.load(tone_json)['text'],
+                              content_type='text/plain', tones='emotion')
+print(json.dumps(tone, indent=2))
+
+print("\ntone() example 4:\n")
+with open(join(dirname(__file__),
+               '../resources/tone-example.json')) as tone_json:
+    tone = tone_analyzer.tone(json.load(tone_json),
+                              content_type='application/json', tones='emotion')
+print(json.dumps(tone, indent=2))
+
+print("\ntone() example 5:\n")
+with open(join(dirname(__file__),
+               '../resources/tone-example-html.json')) as tone_json:
+    tone = tone_analyzer.tone(json.load(tone_json)['text'],
+                              content_type='text/html', tones='emotion')
+print(json.dumps(tone, indent=2))

resources/tone-example-html.json

@@ -0,0 +1,3 @@
+{
+  "text": "<p>Team, I know that times are tough!</p> <p>Product sales have been disappointing for the past three quarters.</p> <p>We have a competitive product, but <b>we need to do a better job of selling it!</b></p>"
+}

resources/tone-example.json

@@ -0,0 +1,3 @@
+{
+  "text": "Team, I know that times are tough! Product sales have been disappointing for the past three quarters. We have a competitive product, but we need to do a better job of selling it!"
+}

watson_developer_cloud/tone_analyzer_v3.py

@@ -28,7 +28,6 @@
 
 from .watson_developer_cloud_service import WatsonDeveloperCloudService
 
-
 class ToneAnalyzerV3(WatsonDeveloperCloudService):
     """Client for the ToneAnalyzer service."""
 
@@ -44,27 +43,43 @@ def __init__(self, version, url=default_url, **kwargs):
     # tone
     #########################
 
-    def tone(self, text, tones=None, sentences=None):
+    def tone(self, text, content_type='text/plain', tones=None, sentences=None):
         """
-        The tone API is the main API call: it analyzes the "tone" of a piece
-        of text. The message is analyzed from
-        several tones (social tone, emotional tone, writing tone), and for
-        each of them various traits are derived
-        (such as conscientiousness, agreeableness, openness).
-        :param text: Text to analyze
-        :param sentences: If "false", sentence-level analysis is omitted
-        :param tones: Can be one or more of 'social', 'language', 'emotion';
-        comma-separated.
+        The general purpose tone API analyzes the "tone" of input text.
+        The message is analyzed for several tones (social, emotional, and
+        writing), with various characteristics derived for each tone.
+        :param text: The input content to analyze.
+        :param content_type: The type of the input content: "text/plain"
+        (the default), "text/html", or "application/json".
+        :param sentences: If false, sentence-level analysis is omitted; by
+        default (or if true), each sentence is analyzed.
+        :param tones: A comma-separated list of one or more of the following
+        tones for which to analyze the input text, 'social', 'language', and
+        'emotion'; the default is all tones.
         """
         params = {'version': self.version}
         if tones is not None:
             params['tones'] = tones
         if sentences is not None:
-            params['sentences'] = str(
-                sentences).lower()  # Cast boolean to "false" / "true"
-        data = {'text': text}
-        return self.request(method='POST', url='/v3/tone', params=params,
-                            json=data, accept_json=True)
+            # Cast boolean to "false" / "true"
+            params['sentences'] = str(sentences).lower()
+        if content_type == 'text/plain':
+            text = {'text': text}
+            content_type = 'application/json'
+        headers = {'content-type': content_type}
+
+        if content_type == 'application/json':
+            return self.request(
+                method='POST', headers=headers, url='/v3/tone', params=params,
+                json=text, accept_json=True)
+        else:
+            return self.request(
+                method='POST', headers=headers, url='/v3/tone', params=params,
+                data=text, accept_json=True)
+
+    #########################
+    # tone_chat
+    #########################
 
     def tone_chat(self, utterances):
         """