import requests
import json
import sys
import re

DEVELOPER_API_BASE_URL = "https://api.ce-cotoha.com/api/dev/nlp/"
CLIENT_ID = "自分のID"
CLIENT_SECRET = "自分のシークレットコード"

def getAccessToken(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"]

# 言い淀み除去APIを実行
def remove_filler(text):
    url = DEVELOPER_API_BASE_URL + "beta/remove_filler"
    headers = {
        "Content-Type": "application/json",
        "Authorization": "Bearer "+ access_token
    }
    data = {
        "text": text,
        "do_segment": False
    }
    r_post = requests.post(url, headers=headers, json=data)
    r_json = r_post.json()
    return r_json

# 感情分析APIを実行
def sentiment(sentence):
    url = DEVELOPER_API_BASE_URL + "v1/sentiment"
    headers = {
        "Content-Type": "application/json",
        "Authorization": "Bearer "+ access_token
    }
    data = {
        "sentence": sentence
    }
    r_post = requests.post(url, headers=headers, json=data)
    r_json = r_post.json()
    return r_json

def parse(sentence):
    url = DEVELOPER_API_BASE_URL + "v1/parse"
    headers = {
        "Content-Type": "application/json",
        "Authorization": "Bearer "+ access_token
    }
    data = {
        "sentence": sentence       
    }
    r_post = requests.post(url, headers=headers, json=data)
    r_json = r_post.json()
    return r_json

if __name__ == "__main__":
    args = sys.argv
    if len(args) >= 2:
        sentence = str(args[1])

    access_token = getAccessToken(CLIENT_ID, CLIENT_SECRET)

    # 処理１：言い澱み除去
    rf_json = remove_filler(sentence)
    sentence2 = rf_json["result"][0]["fixed_sentence"]

    # 処理2：人称代名詞の変換
    parse_json = parse(sentence2)
    original_list = []
    result_list = []
    for chunk in parse_json["result"]:
        # 形態素ごとに分割
        for token in chunk["tokens"]:
            # 1人称修正
            if token["kana"] in ["ワタシ","オレ","ボク"]:
                # print("このDIO")
                result_list.append("このDIO")
            # 2人称修正
            elif token["kana"] in ["キミ","アナタ"]:
                result_list.append("貴様")
            # 3人称修正
            elif "固有" in token["features"]:
                result_list.append("承太郎")
            # 処理３：無駄ラッシュ処理
            elif token["kana"] == "ムダ":
                result_list.append("無駄無駄無駄無駄無駄")
            else:
                result_list.append(token["form"])
            original_list.append(token["form"])
    # 処理4：文末処理
    s=''.join(result_list)
    if re.search("([^一-龥ぁ-ん|ァ-ヶ|0-9|a-zA-Z]){1,}$",s):
        script = re.sub("([^一-龥ぁ-ん|ァ-ヶ|0-9|a-zA-Z]){1,}$","ッ！",s)
    else:
        script = s + "ッ！"

    # 処理５：感情分析
    sentiment_json_original = sentiment(sentence)
    if sentiment_json_original["result"]["sentiment"] == "Positive" and sentiment_json_original["result"]["score"] > 0.5:
        script = script + "最高に「ハイ！」ってやつだアアアアアアハハハハハハハハハハーッ "
    elif sentiment_json_original["result"]["sentiment"] == "Negative" and sentiment_json_original["result"]["score"] > 0.5:
        script = "頭痛がする は...吐き気もだ... くっ...ぐう な...なんてことだ... このDIOが...... 気分が悪いだと？？" + script

    print(script)

$ python TheWorld.py 今日も良い１日でした！君のお陰だよ！
今日も良い1日でした!貴様のお陰だよッ！
$ python TheWorld.py 明日は田中君と遊ぶ。
明日は承太郎君と遊ぶッ！
$ python TheWorld.py ﾎﾞｸは風呂に入るわ・・
このDIOは風呂に入るわッ！
$ python TheWorld.py 無駄無駄… 悲しいよ．．
無駄無駄無駄無駄無駄無駄無駄無駄無駄無駄... 悲しいよッ！
$ python TheWorld.py 僕は二郎ラーメンが好きです。
このDIOは承太郎ラーメンが好きですッ！最高に「ハイ！」ってやつだアアアアアアハハハハハハハハハハーッ