Splitting the main screen into parts (including st.chat_input)

I am working on a document chatbot powered by a LLM, and would like to "split the screen into two parts (one to display the document) and the other to display a chat interface.
I tried to do it with st.columns, but i saw that it’s not possible to put “st.chat_input” into a col.
Is there a way to do that ?
Thank you !

hi @Snow_Yoo ,

Once try st.text_input() in place of st.chat_input. Let me know if the problem continues.

using st.text_input, it works, but with st.chat_input it doesn’t work