# -*- coding: utf-8 -*-

import json
import requests

# goo
# Overview: Japanese Word Similarity API
app_id = #ID you get when registration

url_shortsum = "https://labs.goo.ne.jp/api/shortsum"
url_entity = "https://labs.goo.ne.jp/api/entity"
url_similarity = "https://labs.goo.ne.jp/api/similarity"


def shortsum(length, review_list, request_id="record001"):
    # shortsum API interface: https://labs.goo.ne.jp/api/2015/1150/
    payload = {"app_id": app_id, "request_id": request_id, "length": length, "review_list": review_list}
    headers = {'content-type': 'application/json'}
    r = requests.post(url_shortsum, data=json.dumps(payload), headers=headers)
    print r.text

def entity(sentence, class_filter, request_id="record002"):
    # entity API interface: https://labs.goo.ne.jp/api/2015/336/
    # “ORG”(organization name), “PSN”(person name), “LOC”(location name), “DAT”(date expression) and “TIM”(time expression)
    payload = {"app_id": app_id, "request_id": request_id, "sentence": sentence, "class_filter": class_filter}
    headers = {'content-type': 'application/json'}
    r = requests.post(url_entity, data=json.dumps(payload), headers=headers)
    print r.text

def similarity(word1, word2, request_id="record003"):
    # similarity API interface: https://labs.goo.ne.jp/api/2015/330/
    payload = {"app_id": app_id, "request_id": request_id, "query_pair": [word1, word2]}
    headers = {'content-type': 'application/json'}
    r = requests.post(url_similarity, data=json.dumps(payload), headers=headers)
    print r.text


if __name__ == '__main__':

    shortsum(120, [
        "以前にも購入しているので、こちらのスムージーはフルーティな味でとても美味しいです。今回はお安く購入できたのでお得でした。こちらのスムージーは飲みやすくて、マンゴー味が飽きがこなくて爽やかなのでまたリピートしたいです。",
        "待ちに待ったグリーンスムージーが届きました。牛乳も豆乳も嫌いなので、水180ccで割って飲みました、思っていたほど全然青臭くなくて、マンゴーの味もかすかで美味しくはありませんでした。もっと濃厚なものを想像していましたが、水割りのせいかかなり薄いです。常温で飲んだ方が良いらしいのですが、氷で割りたかったです。水割りのせいか腹持ち感はまるでなく、一食置き換えは無理みたいです。個人的には身体に酵素を取り入れたかったのですが…知人に購入したことを話したところ以下のような意見を頂きました。「代謝酵素と消化酵素は全く違います。スムージーは消化の助けにはなるかも？ですが、代謝酵素としては無理なのです。まず胃で分解されます、なので必要なところには行きません。」さて、効果のほどはどうなのでしょう？酵素を取り入れたい私です。"
        ]
    )
    entity("山下さんは明日の10時30分に新潟に行きます。", "ORG|PSN|LOC|DAT")
    similarity("のれん", "セールスマン")

(agile_env)nlp $ python test.py
{"request_id":"record001","length":60,"summary":"以前にも購入しているので、こちらのスムージーはフルーティな味でとても美味しいです。今回はお安く購入できたのでお得でした。"}
{"request_id":"record002","class_filter":"ORG|PSN|LOC|DAT","ne_list":[["山下","PSN"],["明日","DAT"],["新潟","LOC"]]}
{"request_id":"record003","score":0.4345982085070782}
(agile_env)nlp $ 
(agile_env)nlp $ 
(agile_env)nlp $ 
(agile_env)nlp $ python test.py
{"request_id":"record001","length":120,"summary":"今回はお安く購入できたのでお得でした。牛乳も豆乳も嫌いなので、水180ccで割って飲みました、思っていたほど全然青臭くなくて、マンゴーの味もかすかで美味しくはありませんでした。常温で飲んだ方が良いらしいのですが、氷で割りたかったです。"}
{"request_id":"record002","class_filter":"ORG|PSN|LOC|DAT","ne_list":[["山下","PSN"],["明日","DAT"],["新潟","LOC"]]}
{"request_id":"record003","score":0.4345982085070782}