# coding: utf-8
import os
import sys
# Define root path
sys.path.append('../')
# Import necessary packages
from modules import json_parser
from modules import Client
from write_poem import start_model

# Initialise Client object
client = Client(api_key='33799e1c5d6fa05fdd7dec3aa7aad868445d1c737edcf9c37fa89cb3b39cb2d9',
                project_id='5bfd118f1afd942b66b36b30', user_ID='yangsaisai',
                project_type='app', source_file_path='main.ipynb',
                silent=True)

# Make run/train/predict command alias for further use
run = client.run
train = client.train
predict = client.predict

# Make controller alias for further use
controller = client.controller
writer = start_model()


def handle(conf):
    """
    该方法是部署之后，其他人调用你的服务时候的处理方法。
    请按规范填写参数结构，这样我们就能替你自动生成配置文件，方便其他人的调用。
    范例：
    params['key'] = value # value_type: str # description: some description
    参数请放到params字典中，我们会自动解析该变量。
    """

    style = conf['style']  # value_type: str # description: some description
    Chinese_word = conf['Chinese_word']  # value_type: str # description: some description
    # ticai = conf['体裁']
    # add your code
    # if ticai == '七言':
    #     char_len = 32
    # else:
    #     char_len = 24
    # add your code
    if style == '藏头诗':
        poetry_test = writer.cangtou(Chinese_word)
        # while len(poetry_test) != char_len:
        #     poetry_test = writer.cangtou(Chinese_word)
        poetry = poetry_test
    elif conf['style'] == '藏字诗':
        poetry_test = writer.hide_words(Chinese_word)
        # while len(poetry_test) != char_len:
        #     poetry_test = writer.cangtou(Chinese_word)
        poetry = poetry_test
    else:
        poetry_test = writer.rhyme_verse()
        # while len(poetry_test) != char_len:
        #     poetry_test = writer.cangtou(Chinese_word)
        poetry = poetry_test
    res1 = poetry.split('。')[:2]
    res2 = []
    for i in res1:
        res2.extend(i.split('，'))
    res3 = res2[0]+'，\n'+res2[1]+'。\n'+res2[2]+'，\n'+res2[3]+'。'
    return {'Poetry': res3}