You should be able to create 2 static headers on 2 columns of equal size before the chat elements – as follows:
import streamlit as st
col1, col2 = st.columns(2)
with col1:
st.markdown("## Header 1")
with col2:
st.markdown("## Header 2")
Would that work?
Best,
Charly