허깅페이스 스페이스에 다음의 app.py로 앱을 만들었는데, 다운로드 받을 때 "FORMAT HTML"이라는 기능을 넣고 싶어요.(다운로드시 체크)

Quoda Code Editor에서 Tools에 "FORTMAT HTML"이라는 기능이 있는데, 생략했던 <html>, <head>, <body>도 넣어주며 짝이 안맞는 </div>도 고쳐주고 '자동줄바꿈+탭들여쓰기' 다 해주므로 몇천줄이나 되는 채팅 대화 웹페이지도 편집하기가 편해지더라구요.

아무튼 그런 기능을 넣고 싶네요. 밤이 늦었으므로 내일 해볼 것인데 대충은 미리 알고 싶어요. 

==========
==========


import gradio as gr
import re
import tempfile

def extract_html_from_mht(file, start_pattern="<!DOCTYPE html", end_pattern="</html>", mode="무효 처리", occurrence=None):
    if file is None:
        return "파일이 업로드되지 않았습니다.", None

    try:
        # 'binary' 타입으로 파일 내용을 직접 디코딩
        file_content = file.decode('utf-8', errors='ignore')
    except Exception as e:
        return f"파일을 읽는 중 오류가 발생했습니다: {e}", None

    # 정규 표현식 패턴 생성
    # 사용자가 입력한 패턴의 특수문자를 이스케이프 처리
    pattern = re.escape(start_pattern) + r'(.*?)' + re.escape(end_pattern)

    matches = re.finditer(pattern, file_content, re.DOTALL | re.IGNORECASE)

    matches_list = [match.group(0) for match in matches]

    if not matches_list:
        return "지정한 패턴에 맞는 내용이 파일에서 발견되지 않았습니다.", None

    if mode == "모두 추출":
        extracted_content = "\n\n".join(matches_list)
    elif mode == "특정 번째 추출":
        if not occurrence:
            return "추출할 Occurrence을 입력해주세요.", None
        try:
            index = int(occurrence) - 1
            if index < 0 or index >= len(matches_list):
                return f"입력한 occurrence({occurrence})가 범위를 벗어났습니다. 총 {len(matches_list)}개의 매치가 있습니다.", None
            extracted_content = matches_list[index]
        except ValueError:
            return "Occurrence은 숫자여야 합니다.", None
    elif mode == "무효 처리":
        if len(matches_list) > 1:
            return f"패턴이 여러 번 발견되었습니다 ({len(matches_list)}회). 무효로 처리됩니다.", None
        extracted_content = matches_list[0]
    else:
        return "유효하지 않은 처리 모드입니다.", None

    # 임시 파일 생성 및 내용 쓰기
    try:
        with tempfile.NamedTemporaryFile(delete=False, suffix=".html", mode='w', encoding='utf-8') as temp_file:
            temp_file.write(extracted_content)
            temp_file_path = temp_file.name
    except Exception as e:
        return f"임시 파일을 생성하는 중 오류가 발생했습니다: {e}", None

    return None, temp_file_path

def toggle_occurrence_visibility(mode):
    return gr.update(visible=(mode == "특정 번째 추출"))

with gr.Blocks() as demo:
    gr.Markdown("# 파일에서 특정 부분 추출 및 다운로드")
    gr.Markdown("""
    이 앱을 사용하면 업로드한 MHT 또는 HTML 파일에서 특정 패턴에 맞는 부분을 추출하여 다운로드할 수 있습니다.
    """)

    with gr.Row():
        with gr.Column():
            file_input = gr.File(
                label="파일 업로드",
                file_types=["text", "html", "mht"],
                type="binary"  # 'binary'로 수정
            )
            start_input = gr.Textbox(
                label="시작 패턴",
                value="<!DOCTYPE html>",
                placeholder="시작 부분을 지정하세요 (예: <!DOCTYPE html>)"
            )
            end_input = gr.Textbox(
                label="끝 패턴",
                value="</html>",
                placeholder="끝 부분을 지정하세요 (예: </html>)"
            )
            mode_dropdown = gr.Dropdown(
                label="매치 처리 방식",
                choices=["무효 처리", "모두 추출", "특정 번째 추출"],
                value="무효 처리"
            )
            occurrence_input = gr.Textbox(
                label="추출할 Occurrence 번호 (mode가 '특정 번째 추출'일 때)",
                placeholder="예: 1",
                visible=False
            )
            extract_button = gr.Button("부분 추출 및 다운로드")

        with gr.Column():
            output_message = gr.Markdown("")
            download_output = gr.File(label="추출된 파일 다운로드")

    mode_dropdown.change(
        fn=toggle_occurrence_visibility,
        inputs=mode_dropdown,
        outputs=occurrence_input
    )

    extract_button.click(
        fn=extract_html_from_mht,
        inputs=[file_input, start_input, end_input, mode_dropdown, occurrence_input],
        outputs=[output_message, download_output]
    )

    demo.launch()