idZddlmZddlZddlZddlZddlZddlZddlZddl m Z ddl m Z d dZ ddZddZdd Zdd Zdd  dd Zy)uOCR 기반 키프레임 검증 모듈. IDS §0.1 — 첫/중간/끝 프레임 한글 텍스트 존재 확인 - pytesseract 사용 가능 시: OCR로 텍스트 추출 - pytesseract 미설치 시: PIL 픽셀 분석으로 콘텐츠 존재 여부 확인 (fallback) ) annotationsN)Path)Optionalctjjdd}|r$t|j r t |Sd}t|j r|St jd}|r|Std)u/ffmpeg 실행 파일 경로를 반환합니다. FFMPEG_BINz/home/jay/.local/bin/ffmpegffmpegu"ffmpeg를 찾을 수 없습니다.) osenvirongetrexistsstrshutilwhichFileNotFoundError)env_bin local_binpath_bins 4/home/jay/workspace/skills/motion-cardnews-ko/ocr.py_get_ffmpeg_binrsojjnn\2.G4='')7|-I I||H%H @ AAclt|}t|}|jddt}tj|dt |dddgdd}t |j}tdd td |d z td |d z d }g}d D]}||}|d |d z } |dd|ddt |ddddt | g } tj| dd} | jdk7rtd|d|d| j|j| |d|d|dfS)uB동영상에서 첫/중간/끝 프레임을 PNG로 추출합니다. Args: video_path: 원본 MP4 파일 경로 output_dir: 추출된 PNG를 저장할 디렉토리 Returns: (first_frame, middle_frame, last_frame) Path 튜플 Raises: RuntimeError: ffmpeg 실행 실패 시 Tparentsexist_okz-iz-fnull-)capture_outputtextgg?g@firstmiddlelast keyframe_z.pngz-yz-ssz.3fz -frames:v1z-q:v2ru키프레임 추출 실패 (z, t=zs): ) rmkdirr subprocessrunr_parse_durationstderrmax returncode RuntimeErrorappend) video_path output_dir ffmpeg_binprobeduration timestamps frame_pathslabeltout_pathcmdresults rextract_keyframesr>$sq j!Jj!JTD1 "J NN T3z?D&#>  E u||,HS#c8c>*CC(J !K,% u )E7$ 77  aW #j/  C M  DtD    !.ugT!E&--Q  8$!%$ q>;q>;q> 99rctjd|}|sy|j\}}}t|dzt|dzzt |zS)uKffmpeg stderr에서 Duration 값을 파싱합니다. 실패 시 1.0 반환.z#Duration:\s*(\d+):(\d+):(\d+\.?\d*)g?i<)researchgroupsintfloat) ffmpeg_stderrmhmnss rr,r,_sN 8-HA xxzHAr1 q6D=3r7R< '%( 22rcr ddl}ddlm}|jt |}|j |ddj S#t$rtjdtd Yy t$r?}tjd t|jd |td Yd}~y d}~wwxYw) u+프레임 이미지에서 텍스트를 추출합니다. pytesseract가 없으면 빈 문자열을 반환하고 경고를 발생시킵니다. Args: frame_path: PNG 프레임 파일 경로 Returns: 추출된 텍스트 문자열 (pytesseract 없으면 빈 문자열) rNImagezkor+engz--psm 6)langconfigupytesseract가 설치되지 않았습니다. OCR을 건너뜁니다. 설치: pip install pytesseract && apt-get install tesseract-ocr tesseract-ocr-korr( stacklevelru OCR 오류 (z): ) pytesseractPILrMopenrimage_to_stringstrip ImportErrorwarningswarn UserWarning Exceptiontype__name__) frame_pathrRrMimges r ocr_framerahsjjZ)**3Yy*QWWYY   a     T!W%5%5$6c!={WXYsAA%B6/B675B11B6cJ ddlm}|jt|j d}t |j }|syt|t|z tfd|Dt|z }|dz}dkDxr|dkDS#t$rYywxYw)uPIL로 프레임이 비어있지 않은지(콘텐츠 존재 여부)를 확인합니다. mean > 5 AND stddev > 5 이면 콘텐츠가 있다고 판단합니다. rrLLFc3.K|] }|z dzyw)r(N).0pmeans r z&_is_frame_non_blank..s71DQ7sg?) rSrMrTrconvertlistgetdatasumlenr[)r^rMr_pixelsvariancestddevrhs @r_is_frame_non_blankrss jjZ)11#6ckkm$6{S[(777#f+ESax&FQJ& sA BAB B"!B")r3c$| ttjd}nt|}|jdd t ||\}}}i} d |fd|fd|ffD]J\} } t| d k(} rtfd|D} n t| } | t| | d | | <L| S#t $rW}t jd|tdd d d dd }t|d t|d t|d dcYd}~Sd}~wwxYw)uw동영상의 첫/중간/끝 프레임에서 한글 텍스트를 검증합니다. IDS §0.1 준수: pytesseract 없으면 PIL 픽셀 분석으로 폴백. Args: video_path: 검증할 MP4 파일 경로 expected_korean_chars: 존재 여부를 확인할 한글 문자/문자열 목록 output_dir: 키프레임 저장 디렉토리 (None이면 임시 디렉토리 사용) Returns: 3개 엔트리의 딕셔너리: { "first": {"frame": "first", "ocr_text": str, "has_expected": bool, "fallback": bool}, "middle": {...}, "last": {...}, } N motion_ocr_)prefixTru키프레임 추출 실패: r(rPrF)frameocr_text has_expectedfallbackr!)rwr"r#r c3&K|]}|v yw)Nre)rfchrxs rriz)validate_korean_frames..sN"rX~Ns)rtempfilemkdtempr)r>r[rXrYrZdictraanyrsbool)r2expected_korean_charsr3tmp_dirr!r"r#r`emptyresultsr9r^ is_fallbackryrxs @rvalidate_korean_framesrs=.x''}=>z" dT 2/ GDvt "G&.60BVTNS zZ("n N8MNNL/z:L  .#   " N1  4QC8+RST"eQUVe73tEQY?Zdhinv|d}~~ sB// D8A D D D)returnr)r2rr3rrztuple[Path, Path, Path])rFrrrE)r^rrr)r^rrr)r2rrz list[str]r3zOptional[Path]rr)__doc__ __future__rr rArr*r}rXpathlibrtypingrrr>r,rarsrrerrrs # B"8:8:8:8:v3:."& 77$7 7  7r