Update app.py

app.py

@@ -150,43 +150,40 @@ def to_pdf(file_path):
             return tmp_file_path
 
 def to_markdown(file_path, end_pages, is_ocr, layout_mode, formula_enable, table_enable, language, progress=gr.Progress(track_tqdm=False)):
-    """
-    - PDF 변환 과정 단계별로 progress 표시
-    - 반환값: 최종 마크다운 텍스트
-    """
-    with progress:
-        progress(0, "파일을 PDF로 변환 중...")
-        file_path = to_pdf(file_path)
-        time.sleep(0.5)  # 예시로 딜레이
-
-        if end_pages > 20:
-            end_pages = 20
-        progress(30, "PDF 파싱 중...")
-        local_md_dir, file_name = parse_pdf(file_path, './output', end_pages - 1, is_ocr,
-                                            layout_mode, formula_enable, table_enable, language)
-        time.sleep(0.5)
-
-        progress(50, "압축(Zip) 생성 중...")
-        archive_zip_path = os.path.join("./output", compute_sha256(local_md_dir) + ".zip")
-        zip_archive_success = compress_directory_to_zip(local_md_dir, archive_zip_path)
-        if zip_archive_success == 0:
-            logger.info("압축 성공")
-        else:
-            logger.error("압축 실패")
-        time.sleep(0.5)
+    progress(0, "파일을 PDF로 변환 중...")
+    file_path = to_pdf(file_path)
+    time.sleep(0.5)
+
+    if end_pages > 20:
+        end_pages = 20
+
+    progress(30, "PDF 파싱 중...")
+    local_md_dir, file_name = parse_pdf(file_path, './output', end_pages - 1, is_ocr,
+                                        layout_mode, formula_enable, table_enable, language)
+    time.sleep(0.5)
+
+    progress(50, "압축(Zip) 생성 중...")
+    archive_zip_path = os.path.join("./output", compute_sha256(local_md_dir) + ".zip")
+    zip_archive_success = compress_directory_to_zip(local_md_dir, archive_zip_path)
+    if zip_archive_success == 0:
+        logger.info("압축 성공")
+    else:
+        logger.error("압축 실패")
+    time.sleep(0.5)
+
+    progress(70, "마크다운 읽는 중...")
+    md_path = os.path.join(local_md_dir, file_name + ".md")
+    with open(md_path, 'r', encoding='utf-8') as f:
+        txt_content = f.read()
+    time.sleep(0.5)
 
-        progress(70, "마크다운 읽는 중...")
-        md_path = os.path.join(local_md_dir, file_name + ".md")
-        with open(md_path, 'r', encoding='utf-8') as f:
-            txt_content = f.read()
-        time.sleep(0.5)
+    progress(90, "이미지 base64 변환 중...")
+    md_content = replace_image_with_base64(txt_content, local_md_dir)
+    time.sleep(0.5)
 
-        progress(90, "이미지 base64 변환 중...")
-        md_content = replace_image_with_base64(txt_content, local_md_dir)
-        time.sleep(0.5)
+    progress(100, "변환 완료!")
+    return md_content
 
-        progress(100, "변환 완료!")
-        return md_content
 
 def init_model():
     from magic_pdf.model.doc_analyze_by_custom_model import ModelSingleton