from generate import Generator

generator = Generator(user_input = "A transitional kitchen area equipped with a brand new rectangular wooden kitchen cutting board, a blue and white carton of low-fat milk, and a metal kettle with handle.",
                   max_number_of_objects = 16,
                   room_dimensions = [5.0, 5.0, 3.0],
                   layout_elements = ['south_wall', 'north_wall', 'west_wall', 'east_wall', 'middle of the room', 'ceiling']
                   )

'''
Step 1: 解析用户输入的需求，并选择需要摆放的物体。
'''

generator.parse_input()
generator.retrieve_local_assets()

'''
Step 2: 生成Scene Graph
'''

generator.create_scene_graph()
generator.summary_language()

'''
Step 3: 生成Layout
'''


# generator.refine_design(verbose=False)
# generator.create_object_clusters(verbose=False)
generator.create_layout(debug=False)
generator.summary_calculation()