# coding = utf-8
# You can use other public modules via our Client object with module's identifier
# and parameters.
# For more detailes, please see our online document - https://momodel.github.io/docs/#

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.py', 
                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
    # path = os.getcwd()  # 获取当前工作目录
    # print(path)
    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}


# if __name__ == '__main__':
#     conf = {}
#     handle(conf)