# coding: utf-8
import os
import sys

# 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):
    # paste your code here
    if conf['style'] == '藏头诗':
        poetry = writer.cangtou(conf['Chinese_word'])
    elif conf['style'] == '藏字诗':
        poetry = writer.hide_words(conf['Chinese_word'])
    else:
        poetry = writer.rhyme_verse()
    return {"Poetry": poetry}
    # return {"Poetry": 'asadd'}

# if __name__ == '__main__':
#     conf = {"style": ['藏头诗'], "Chinese_word": '杨赛赛'}
#     aa = handle(conf)
#     print(aa)