Make the repo more developer-friendly

- ~~add a licence~~
- ~~put links, screenshot in the README.md~~
- better documentation (how the prompt is constructed, how HTML images are generated etc)
- clean the code (remove console.log() etc)
- add unit tests
- add more storybook