allow any input type

[akbor]

changed it such that it allows any input type to be set.

[blackary]

This is great overall! Can you please fix the linting issues? (You can do this with pre-commit install and then pre-commit run --all-files)

Also, is there a reason you changed some of the sizes in CSS? Could you change those back, so that it lines up better with existing streamlit widgets? My goal was for this to look as identical as possible to st.text_input