scene.reset() doesn't work as defined

[AhmedEltayyebXL2]

Description

I'm trying to render a whole dataset, so I want to make it as efficient as possible. I try to create the scene once so that it takes less time overall. Then, I reset the scene every iteration because my functions create_entity_surface() and add_defects_to_scene() generate different entities every time.

Issue

My problem is that with the following code, I get the error: "Scene is already built" when I try to add the entity to the scene.
I also can't save the state of the scene before building it. I get the error: "genesis.GenesisException: Scene is not built yet."

Is there an efficient way to solve this?

Main()

def main():
args = parse_args()

# Initialize Genesis
gs.init(backend=gs.gpu)
print("Initializing genesis...")

# Create timestamp for unique filenames
timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S_%f")

# Create output directory if it doesn't exist
os.makedirs(args.output_dir, exist_ok=True)

# scene, cam = make_scene_ready()
scene = create_scene()

# Add camera
cam = add_camera(scene)
print("Camera added")

# Add lighting
add_lighting(scene)
print("Lighting added")

scene.build(compile_kernels=False)
print("Initial scene built")

empty_scene_state = scene.get_state()    

# Run the generation process
for i in range(args.count):
    print(f"Generating image {i+1}/{args.count}")

    scene.reset(empty_scene_state)
    print("Scene reset")
    
    # Create scene with camera and lighting
    # scene = create_scene()
    
    # Create textured metal surface
    entity_surface = create_entity_surface()

    # Add base sheet
    sheet_entity = scene.add_entity(
        gs.morphs.Plane(),
        surface=entity_surface 
    )
    print("Base sheet entity added")
    
    # Add defects if requested
    if args.defect_type != "none":
        current_defects = add_defects_to_scene(scene, args.defect_type)
        print(f"Added {len(current_defects)} defect entities")
    
    # Build and run the scene
    scene.build(compile_kernels=False)
    print("Scene built")
    
    # Render the scene
    results = cam.render(rgb=True, segmentation=True)
    print("Scene rendered")
    
    # Process and save the result
    img_id = f"{timestamp}_{i}"
    apply_post_processing(results, args.output_dir, img_id)
    print(f"Image {i+1} saved to {args.output_dir}")