it looks good and cool because it only uses one line, but you also need to pay attention to security, not just one line,
for example if the user inputs something that is not appropriate such as ("I'm a bad hexker", 2,10)
in the real world it will definitely be detrimental in the future,|

This is not judging but just giving advice