customize websocket client to take data input and forward it

Question

i found this example code here which i pretty much like:Issues listening incoming messages in websocket client on Python 3.6

However, I would like to customize it alittle