Update app.py

app.py

@@ -8,11 +8,19 @@ import re
 import uuid
 import pymupdf
 
+# 이미지 전처리에 필요한 라이브러리
+import cv2
+import numpy as np
+
 ###############################
 # 환경 설정
 ###############################
 os.system('pip uninstall -y magic-pdf')
 os.system('pip install git+https://github.com/opendatalab/MinerU.git@dev')
+
+# ★ OpenCV 설치 (headless 버전)
+os.system('pip install opencv-python-headless')
+
 os.system('wget https://github.com/opendatalab/MinerU/raw/dev/scripts/download_models_hf.py -O download_models_hf.py')
 os.system('python download_models_hf.py')
 
@@ -145,15 +153,67 @@ def replace_image_with_base64(markdown_text, image_dir_path):
         return f"![{relative_path}](data:image/jpeg;base64,{base64_image})"
     return re.sub(pattern, replace, markdown_text)
 
+###############################
+# 이미지 전처리 함수 (Grayscale/Binarization + Deskew)
+###############################
+def preprocess_image(image_path):
+    """
+    1) Grayscale + Binarization(OTSU)
+    2) Deskew(기울임 보정)
+    전처리된 이미지를 임시 경로에 저장 후 해당 경로를 반환
+    """
+    img = cv2.imread(image_path, cv2.IMREAD_GRAYSCALE)
+    if img is None:
+        # 이미지 파일이 아닌 경우 혹은 로딩 실패 시 원본 경로 그대로 반환
+        return image_path
+
+    # (a) 이진화(binarization)
+    _, img_bin = cv2.threshold(img, 0, 255, cv2.THRESH_OTSU | cv2.THRESH_BINARY)
+
+    # (b) 기울임 보정(deskew)
+    coords = np.column_stack(np.where(img_bin > 0))
+    angle = cv2.minAreaRect(coords)[-1]
+    # OpenCV는 회전 각도를 [-90, 0)로 반환할 때가 많으므로 보정
+    if angle < -45:
+        angle = -(90 + angle)
+    else:
+        angle = -angle
+
+    (h, w) = img_bin.shape[:2]
+    center = (w // 2, h // 2)
+    M = cv2.getRotationMatrix2D(center, angle, 1.0)
+    img_rotated = cv2.warpAffine(
+        img_bin,
+        M,
+        (w, h),
+        flags=cv2.INTER_CUBIC,
+        borderMode=cv2.BORDER_CONSTANT,
+        borderValue=255
+    )
+
+    # 임시 파일로 저장
+    preprocessed_path = image_path + "_preprocessed.png"
+    cv2.imwrite(preprocessed_path, img_rotated)
+
+    return preprocessed_path
+
 def to_pdf(file_path):
     """
-    이미지(JPG/PNG 등)를 PDF로 컨버팅.
+    이미지(JPG/PNG 등)를 PDF로 컨버팅하되,
+    이미지일 경우 전처리(Grayscale/Binarization + Deskew)를 먼저 적용
     """
     with pymupdf.open(file_path) as f:
         if f.is_pdf:
             return file_path
         else:
-            pdf_bytes = f.convert_to_pdf()
+            # 이미지 파일인 경우, 전처리 수행 후 PDF 생성
+            f.close()
+            preprocessed_path = preprocess_image(file_path)
+
+            # 전처리된 이미지를 다시 PyMuPDF로 열어서 PDF 변환
+            with pymupdf.open(preprocessed_path) as img_doc:
+                pdf_bytes = img_doc.convert_to_pdf()
+
             unique_filename = f"{uuid.uuid4()}.pdf"
             tmp_file_path = os.path.join(os.path.dirname(file_path), unique_filename)
             with open(tmp_file_path, 'wb') as tmp_pdf_file:
@@ -384,7 +444,7 @@ latex_delimiters = [
 def to_markdown_ocr_flex(file_path, end_pages, is_ocr, layout_mode, formula_enable, table_enable, language):
     """
     스니펫에서 사용: 
-    업로드된 PDF/이미지를 변환 후
+    업로드된 PDF/이미지 -> PDF 변환 -> 마크다운 변환
     (마크다운 렌더링 / 마크다운 텍스트 / 압축파일 / PDF미리보기) 반환
     """
     file_path = to_pdf(file_path)