user_shape = input() if user_shape == "square": print("***\n* *\n***") elif user_shape == "triangle": print(" *\n * *\n*****")
To embed this project on your website, copy the following code and paste it into your website's HTML: