ADD: Endpoint for multi api requests

indicoio/utils/multi.py

@@ -0,0 +1,121 @@
+from indicoio.config import TEXT_APIS, IMAGE_APIS, API_NAMES
+from indicoio.utils import api_handler
+
+
+CLIENT_SERVER_MAP = dict((api, api.strip().replace("_", "").lower()) for api in API_NAMES)
+SERVER_CLIENT_MAP = dict((v, k) for k, v in CLIENT_SERVER_MAP.iteritems())
+
+def multi(data, type, apis, available, batch=False, **kwargs):
+    """
+    Helper to make multi requests of different types.
+
+    :param data: data to be sent in JSON.
+    :param type: String type of API request
+    :param apis: List of apis to use.
+    :param apis: List of apis available for use.
+    :type data: str or image
+    :type type: str or unicode
+    :type apis: list of str
+    :type available: list of str
+    :rtype: Dictionary of api responses
+    """
+    # Client side api name checking - strictly only accept func name api
+    invalid_apis = [api for api in apis if api not in available]
+    if invalid_apis:
+        raise ValueError("%s are not valid %s APIs. Please reference the available APIs below:\n%s"
+                            % (", ".join(invalid_apis), type, ", ".join(available))
+                        )
+    # Convert client api names to server names before sending request
+    apis = map(CLIENT_SERVER_MAP.get, apis)
+    result = api_handler(data, apis=apis, batch=batch, **kwargs)
+
+    if batch:
+        return [handle_response(each) for each in result]
+
+    return handle_response(result)
+
+def handle_response(result):
+    try:
+        # Parse out the results to a dicionary of api: result
+        return dict((SERVER_CLIENT_MAP[api], parsed_response(res))
+            for api, res in result.iteritems())
+    except KeyError:
+        for api in result:
+            if "error" in result[api]:
+                raise ValueError(result[api]["error"])
+        raise Exception("Sorry, %s API returned an unexpected response:\n%s" % (api, result[api]))
+
+
+def predict_text(input_text, apis, cloud=None, batch=False, api_key=None, **kwargs):
+    """
+    Given input text, returns the results of specified text apis. Possible apis
+    include: [ 'text_tags', 'political', 'sentiment', 'language' ]
+
+    Example usage:
+
+    .. code-block:: python
+
+       >>> import indicoio
+       >>> text = 'Monday: Delightful with mostly sunny skies. Highs in the low 70s.'
+       >>> results = indicoio.text(data = text, apis = ["language", "sentiment"])
+       >>> language_results = results["langauge"]
+       >>> sentiment_results = results["sentiment"]
+
+    :param text: The text to be analyzed.
+    :param apis: List of apis to use.
+    :type text: str or unicode
+    :type apis: list of str
+    :rtype: Dictionary of api responses
+    """
+
+    return multi(
+        api="apis",
+        data=input_text,
+        type="text",
+        available = TEXT_APIS,
+        cloud=cloud,
+        batch=batch,
+        api_key=api_key,
+        apis=apis,
+        **kwargs)
+
+
+def predict_image(image, apis, cloud=None, batch=False, api_key=None, **kwargs):
+    """
+    Given input image, returns the results of specified image apis. Possible apis
+    include: ['fer', 'facial_features', 'image_features']
+
+    Example usage:
+
+    .. code-block:: python
+
+       >>> import indicoio
+       >>> import numpy as np
+       >>> face = np.zeros((48,48)).tolist()
+       >>> results = indicoio.image(image = face, apis = ["fer", "facial_features"])
+       >>> fer = results["fer"]
+       >>> facial_features = results["facial_features"]
+
+    :param text: The text to be analyzed.
+    :param apis: List of apis to use.
+    :type text: str or unicode
+    :type apis: list of str
+    :rtype: Dictionary of api responses
+    """
+
+    return multi(
+        api="apis",
+        data=image,
+        type="image",
+        available=IMAGE_APIS,
+        cloud=cloud,
+        batch=batch,
+        api_key=api_key,
+        apis=apis,
+        **kwargs)
+
+def parsed_response(response):
+    result = response.get('results') or response.get('error', False)
+    if result:
+        return result
+    raise KeyError