시각화 코드 추가

2025-10-30 09:38:52 +09:00
parent 328d5f3ce8
commit 9e51fd56e9
284 changed files with 1057 additions and 3348 deletions
--- a/workspace/visualizer.py
+++ b/workspace/visualizer.py
@@ -0,0 +1,110 @@
+import json
+import os
+
+import streamlit as st
+from PIL import Image
+
+st.set_page_config(layout="wide")
+
+st.title("OCR 결과 비교 시각화 도구")
+st.write("`flow.py` 실행 후 생성된 `ocr_compare_*.json` 파일을 업로드하세요.")
+
+uploaded_file = st.file_uploader("JSON 파일 선택", type="json")
+
+if uploaded_file is not None:
+    try:
+        data = json.load(uploaded_file)
+
+        if not data:
+            st.warning("JSON 파일에 처리된 문서가 없습니다.")
+        else:
+            # 문서 선택 드롭다운
+            source_images = [
+                item.get("source_image", f"문서 {i+1}") for i, item in enumerate(data)
+            ]
+            selected_doc_name = st.selectbox("분석할 문서를 선택하세요:", source_images)
+
+            selected_doc = next(
+                (
+                    item
+                    for item in data
+                    if item.get("source_image") == selected_doc_name
+                ),
+                None,
+            )
+
+            if selected_doc:
+                st.header(f"📄 원본 문서: `{selected_doc['source_image']}`")
+
+                cropped_regions = selected_doc.get("cropped_regions", [])
+
+                if not cropped_regions:
+                    st.warning("이 문서에는 추출된 영역이 없습니다.")
+                else:
+                    st.subheader("✂️ 영역별 OCR 결과 비교")
+
+                    for i, region in enumerate(cropped_regions):
+                        st.markdown("---")
+
+                        region_image_path_from_json = region.get("region_image_path")
+
+                        # JSON에 저장된 경로는 실행 환경에 따라 다를 수 있으므로, 파일 이름만 추출하여
+                        # 이 스크립트가 위치한 곳의 'crops' 디렉토리 기준으로 경로를 재구성합니다.
+                        image_filename = os.path.basename(region_image_path_from_json)
+                        current_script_dir = os.path.dirname(__file__)
+                        local_image_path = os.path.join(
+                            current_script_dir, "crops", image_filename
+                        )
+
+                        if os.path.exists(local_image_path):
+                            image = Image.open(local_image_path)
+
+                            col1, col2, col3 = st.columns([1, 2, 2])
+
+                            with col1:
+                                st.image(
+                                    image,
+                                    caption=f"영역 #{i+1} ({region.get('label')})",
+                                    use_column_width=True,
+                                )
+                                st.write(f"**BBox:** `{region.get('bbox')}`")
+                                st.write(
+                                    f"**Page:** `{region.get('page_index', 0) + 1}`"
+                                )
+
+                            with col2:
+                                st.markdown("##### **Paddle OCR 결과**")
+                                paddle_result = region.get("paddle_ocr_result", "N/A")
+                                st.text_area(
+                                    "Paddle",
+                                    paddle_result,
+                                    height=200,
+                                    key=f"paddle_{i}_{selected_doc_name}",
+                                )
+
+                            with col3:
+                                st.markdown("##### **Pytesseract OCR 결과**")
+                                pytesseract_result = region.get(
+                                    "pytesseract_ocr_result", "N/A"
+                                )
+                                st.text_area(
+                                    "Pytesseract",
+                                    pytesseract_result,
+                                    height=200,
+                                    key=f"pytesseract_{i}_{selected_doc_name}",
+                                )
+                        else:
+                            st.warning(
+                                f"이미지 파일을 찾을 수 없습니다: `{local_image_path}`"
+                            )
+                            st.info(
+                                f"(JSON 원본 경로: `{region_image_path_from_json}`)"
+                            )
+
+    except json.JSONDecodeError:
+        st.error("올바른 JSON 파일이 아닙니다. 파일이 손상되었을 수 있습니다.")
+    except Exception as e:
+        st.error(f"파일을 처리하는 중 오류가 발생했습니다: {e}")
+
+else:
+    st.info("시각화할 OCR 결과 파일을 업로드해주세요.")