youyaku.py

import requests
import json
import sys

BASE_URL = "https://api.ce-cotoha.com/api/dev/"
CLIENT_ID = "Enter the ID obtained by COTOHA API"
CLIENT_SECRET = "Enter the password obtained by COTOHA API"


def auth(client_id, client_secret):
    token_url = "https://api.ce-cotoha.com/v1/oauth/accesstokens"
    headers = {
        "Content-Type": "application/json",
        "charset": "UTF-8"
    }

    data = {
        "grantType": "client_credentials",
        "clientId": client_id,
        "clientSecret": client_secret
    }
    r = requests.post(token_url,
                      headers=headers,
                      data=json.dumps(data))
    return r.json()["access_token"]


def summary(document, access_token, sent_len):
    base_url = BASE_URL
    headers = {
        "Content-Type": "application/json",
        "charset": "UTF-8",
        "Authorization": "Bearer {}".format(access_token)
    }
    data = {
        "document": document,
        "sent_len": sent_len
    }
    r = requests.post(base_url + "nlp/beta/summary",
                      headers=headers,
                      data=json.dumps(data))
    return r.json()


if __name__ == "__main__":
    document = "The lyrics are listed here"
    args = sys.argv
    if len(args) >= 2:
        document = str(args[1])

    access_token = auth(CLIENT_ID, CLIENT_SECRET)
    summary_document = summary(document, access_token, 3)
    result_list = list()
    for chunks in summary_document['result']:
      result_list.append(chunks)

    print(''.join(result_list))

$ python3 youyaku.py < mitokomon.txt
If you don't like crying, walk now. There are tears and a smile in my life. Let's live in search of something.

$ python3 youyaku.py < doraemon.txt
Everyone, everyone, will come true. Ann An Ann. I really like Doraemon.

$ python3 youyaku.py < soregadaiji.txt
Don't lose, don't throw, don't run away, believe. When it seems to be ruined. That is the most important.

$ python3 youyaku.py < gachaman.txt
Gatchaman. fly. go.

$ python3 youyaku.py < makafushigi.txt
DRAGONBALL。try。fly。

$ python3 youyaku.py < yushaou.txt
Gagagatsu. Gaogaigar!.. Gagagaga.

Recommended Posts