Creating a container for chat input and responses

I’m trying to get my chat inputs and LLM responses to populate within a container for a cleaner look but just can’t seem to get it to work. Both the input and response are being written below the dedicated container.

Any help appreciated!