taneemishere's picture
models added with directory path changed
4d0572e
raw
history blame
3.09 kB
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 dsl_code_generation(input_image):
trained_weights_path = "classes/model/bin"
trained_model_name = "pix2code2"
input_path = 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))
with open("{}/{}.gui".format(output_path, file_name), 'w') as out_f:
out_f.write(result.replace(START_TOKEN, "").replace(END_TOKEN, ""))
return file_name, output_path
def compile_gui(file_path, filename):
from os.path import basename
from compiler.Utils import Utils
from compiler.Compiler import Compiler
input_path = (file_path+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 = "compiler/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_method(input_image_from_interface):
file_name, file_output_path= dsl_code_generation(input_image_from_interface)
result = compile_gui(file_output_path, file_name)
return result