data_pro.py

import json
import random
from generate import Generator

def generate_random_number():
    intervals = [(9, 12), (12, 15), (15, 18)]
    probabilities = [0.5, 0.3, 0.2]

    selected_interval = random.choices(intervals, weights=probabilities, k=1)[0]

    min_val, max_val = selected_interval
    random_number = random.randint(min_val, max_val)

    return random_number

with open('./generate_input_mp/5.json', 'r') as file:
    room_list = json.load(file)


generator = Generator()


for room in room_list:
    user_input = room["user_input"]

    try:
        generator.parse_input(user_input=user_input, max_number_of_objects=generate_random_number())
        generator.retrieve_local_assets()
        generator.create_scene_graph()
        generator.summary_language()
        generator.create_layout(debug=False)
        generator.summary_calculation()
    except:
        pass