Integrating element correlation with prompt-based spatial relation extraction