Update app.py

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 from paddleocr import PaddleOCR
 from PIL import Image
 import numpy as np
+import cv2
 
 # Summarizer
 try:
@@ -131,23 +132,37 @@ def handle_summary(url: str, sent_n: int) -> str:
         return f"에러: {e}"
 
 # -----------------------------
-# PaddleOCR 초기화 (한국어+영어 지원)
+# 이미지 전처리 + PaddleOCR
 # -----------------------------
-ocr = PaddleOCR(use_angle_cls=True, lang='korean')  # 한국어 모델은 영어도 지원
+def preprocess_image(img):
+    if isinstance(img, np.ndarray):
+        arr = img
+    else:
+        arr = np.array(img)
+
+    gray = cv2.cvtColor(arr, cv2.COLOR_RGB2GRAY)
+    # 대비 강화 (이진화)
+    _, thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
+    # 크기 키우기
+    h, w = thresh.shape
+    if h < 600:
+        scale = 600 / h
+        thresh = cv2.resize(thresh, None, fx=scale, fy=scale, interpolation=cv2.INTER_LINEAR)
+    return thresh
+
+ocr = PaddleOCR(use_angle_cls=True, lang='korean')  # 한국어+영어 지원
 
 def handle_image(img) -> str:
     if img is None:
         return "❌ 이미지를 업로드하세요."
     try:
-        # numpy → PIL 변환 후 다시 numpy로
-        if isinstance(img, np.ndarray):
-            img = Image.fromarray(img)
-        results = ocr.ocr(np.array(img))
+        proc_img = preprocess_image(img)
+        results = ocr.ocr(proc_img)
         if not results or not results[0]:
             return "텍스트를 추출할 수 없습니다."
 
         lines = []
-        for res in results[0]:  # 첫 번째 이미지 결과
+        for res in results[0]:
             if len(res) == 2 and isinstance(res[1], tuple):
                 txt, conf = res[1]
                 lines.append(f"{txt} (conf:{conf:.2f})")
@@ -162,7 +177,7 @@ with gr.Blocks(css="""
   #container { max-width: 920px; margin: 0 auto; }
   .small { color:#666; font-size:14px; }
 """) as demo:
-    gr.Markdown("## URL → HTML/텍스트/요약 + 이미지 OCR (PaddleOCR)", elem_id="container")
+    gr.Markdown("## URL → HTML/텍스트/요약 + 이미지 OCR (PaddleOCR + 전처리)", elem_id="container")
 
     with gr.Row():
         url_input = gr.Textbox(label="URL", placeholder="https://example.com", scale=4)
@@ -175,7 +190,7 @@ with gr.Blocks(css="""
         sent_n = gr.Slider(1, 8, value=3, step=1, label="요약 문장 수")
         btn_sum = gr.Button("자동요약 보기", scale=1)
 
-    gr.Markdown("### 이미지 업로드 → OCR (PaddleOCR)")
+    gr.Markdown("### 이미지 업로드 → OCR (PaddleOCR + 전처리)")
     with gr.Row():
         img_input = gr.Image(type="numpy", label="이미지 업로드")
         btn_img = gr.Button("이미지 OCR 실행", scale=1)