Checking file type after uploading to st.file_uploader()

How do I check for what file type a file is after uploading to st.file_uploader()? For example, Iโ€™ve allowed st.file_uploader() to accept .pkl or .gzip files but have to perform a different operation for each.

Hello @vnguyendc,

If using two st.file_upload() is not an option, you could use a library to get the file type from your BytesIO.

1 Like

Thanks @okld Iโ€™ll check this out. I actually approached this using โ€˜try exceptโ€™ statements, and seems to be working alright. But this looks useful.