from bs4 import BeautifulSoup
from flask import current_app as app

from . import main

@main.after_request
def prettify_html_response(response):
    if app.debug and response.content_type.startswith("text/html"):
        try:
            soup = BeautifulSoup(response.get_data(as_text=True), 'html5lib')
            pretty_html = soup.prettify()

            response.set_data(pretty_html.encode("utf-8")) # type: ignore
            response.headers['Content-Type'] = 'text/html; charset=utf-8'

        except Exception as e:
            print(f"⚠️ Prettifying failed: {e}")
    return response