Update main.py

Planning for HTML books support.