File size: 889 Bytes
f20b100 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
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
|