I think this is a limitation with the AggGrid plugin. One idea would be to use native Streamlit tables, but AFAIK, those don't allow one to search / filter by model ID which is quite handy
Updated. Let me know what you think. There are some functions in aggrid that allow dynamic resizing but they are a little limited.
Looks great, thanks!