from __future__ import absolute_import
from __future__ import print_function

import os.path
from os.path import basename

from classes.Sampler import *
from classes.model.pix2code2 import *


def code_gen(path_to_input_image):
    # if len(sys.argv) < 1:
    #     print("Error: not enough argument supplied:")
    #     # print( "sample.py <trained weights path> <trained model name> <input image> <output path> <search method (
    #     # default: " "greedy)>")
    #     exit(0)
    # else:
    trained_weights_path = "../bin"
    trained_model_name = "pix2code2"
    # input_path = sys.argv[1]
    # input_path = "../data/test_gu.png"
    input_path = path_to_input_image
    output_path = "../data/output/"
    search_method = "greedy"

    meta_dataset = np.load("{}/meta_dataset.npy".format(trained_weights_path), allow_pickle=True)
    input_shape = meta_dataset[0]
    output_size = meta_dataset[1]

    model = pix2code2(input_shape, output_size, trained_weights_path)
    model.load(trained_model_name)

    sampler = Sampler(trained_weights_path, input_shape, output_size, CONTEXT_LENGTH)

    file_name = 'input_image_from_interface.png'
    file_name = basename(file_name)[:basename(file_name).find(".")]
    evaluation_img = Utils.get_preprocessed_img(input_path, IMAGE_SIZE)

    if search_method == "greedy":
        result, _ = sampler.predict_greedy(model, np.array([evaluation_img]))
        print("Result greedy: \n {}".format(result))
    # else:
    #     beam_width = int(search_method)
    #     print("Search with beam width: {}".format(beam_width))
    #     result, _ = sampler.predict_beam_search(model, np.array([evaluation_img]), beam_width=beam_width)
    #     print("Result beam: {}".format(result))

    with open("{}/{}.gui".format(output_path, file_name), 'w') as out_f:
        out_f.write(result.replace(START_TOKEN, "").replace(END_TOKEN, ""))

    return output_path, file_name


def compile_gui(outputpath, filename):
    from os.path import basename
    from compclasses.Utils import Utils
    from compclasses.Compiler import Compiler

    input_path = (outputpath+filename)

    # remove the path
    file_ = os.path.basename(input_path)
    # remove the extension
    file_ = os.path.splitext(file_)[0]
    # add the extension of gui
    file_ = "../data/output/" + file_ + ".gui"

    input_file = file_

    FILL_WITH_RANDOM_TEXT = True
    TEXT_PLACE_HOLDER = "[]"

    dsl_path = "assets/web-dsl-mapping.json"
    compiler = Compiler(dsl_path)

    def render_content_with_text(key, value):
        if FILL_WITH_RANDOM_TEXT:
            if key.find("btn") != -1:
                value = value.replace(TEXT_PLACE_HOLDER, Utils.get_random_text())
            elif key.find("title") != -1:
                value = value.replace(TEXT_PLACE_HOLDER, Utils.get_random_text(length_text=5, space_number=0))
            elif key.find("text") != -1:
                value = value.replace(TEXT_PLACE_HOLDER,
                                      Utils.get_random_text(length_text=56, space_number=7, with_upper_case=False))
        return value

    file_uid = basename(input_file)[:basename(input_file).find(".")]
    path = input_file[:input_file.find(file_uid)]

    input_file_path = "{}{}.gui".format(path, file_uid)
    output_file_path = "{}{}.html".format(path, file_uid)

    html_code = compiler.compile(input_file_path, output_file_path, rendering_function=render_content_with_text)
    print("Generated code is compiled..!!")
    return html_code


def main_program(path_to_file):
    output_path, file_name = code_gen(path_to_file)
    result = compile_gui(output_path, file_name)
    return result