iY6UdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddl mZddlZdZdZd ed eZe d Ze d Zed z edz edz dZded<edz ZgddgdgdgdZded<dZdZej6ej8ddej:dZd+d Zd,d!Z d-d"Z! d.d#Z" d/d$Z# d0d%Z$ d1d&Z% d2d'Z&d3d(Z'd4d)Z(e)d*k(re(yy)5u8Gemini API를 사용한 v3 캠페인 이미지 생성 스크립트. gemini-2.0-flash-preview-image-generation 모델을 통해 v3_prompts.json에 정의된 프롬프트 기반으로 이미지를 생성합니다. 한글 깨짐 방지를 위한 다중 시도(attempt) 및 결과 기록 기능을 지원합니다. ) annotationsN)datetime)Path)Anyzgemini-3-pro-image-preview'AIzaSyB4X1239KxqIYIuhr9rBrawwxs2RddbYccz8https://generativelanguage.googleapis.com/v1beta/models/z:generateContent?key=z./home/jay/workspace/teams/dev1/v3_prompts.jsonz1/home/jay/workspace/output/campaign-top/v3-geminizmeta-carousel-azmeta-carousel-bz naver-gfa)meta_carousel_ameta_carousel_b naver_gfazdict[str, Path] OUTPUT_DIRSzgeneration_results.jsonrr r allabgfazdict[str, list[str]] GROUP_MAP,z2%(asctime)s [%(levelname)s] %(name)s - %(message)sz%Y-%m-%d %H:%M:%S)levelformatdatefmt v3_generatec.tjdy)uhAPI 키를 반환합니다. (하위호환용 — API 키가 URL에 포함되므로 빈 문자열 반환)u(API 키 인증 사용 (URL 파라미터))loggerinfo-/home/jay/workspace/teams/dev1/v3_generate.pyget_auth_tokenrEs KK:; rc~|js!tjd|td t |d5}t j |}dddtjd|S#1swY!xYw#t j$r'}tjd|td|d}~wwxYw)uv3_prompts.json 파일에서 프롬프트 설정을 읽어옵니다. Args: prompts_file: 프롬프트 JSON 파일 경로. Returns: 그룹명 → 슬라이드 목록 딕셔너리. Raises: SystemExit: 파일이 없거나 JSON 파싱 실패 시. u2프롬프트 파일을 찾을 수 없습니다: %sutf-8encodingNu%프롬프트 파일 로드 완료: %su*프롬프트 파일 JSON 파싱 실패: %s) existsrerror SystemExitopenjsonloadrJSONDecodeError) prompts_filefdataes r load_promptsr0Ns     I<Xm# , 1 AQ48IIaLD A ;\J  A A   # A1Em"#s/ BA6B6A?;BB<"B77B<cXddi}dd|igigddgddd d d }tjd ttjt||t }|j s>tjd|j|jdd|j|jS)uZGemini API에 이미지 생성 요청을 보냅니다. Args: token: Bearer 인증 토큰. prompt: 이미지 생성 프롬프트. Returns: API 응답 JSON 딕셔너리. Raises: requests.HTTPError: HTTP 4xx/5xx 오류 발생 시. requests.RequestException: 네트워크 오류 발생 시. z Content-Typezapplication/jsonpartstextIMAGETEXTr!gffffff?()responseModalities temperaturetopPtopK)contentsgenerationConfiguAPI 호출: %s)headersr)timeoutuHTTP 오류 %d: %sN) rdebugGEMINI_API_ENDPOINTrequestspostREQUEST_TIMEOUTokr& status_coder3raise_for_statusr))tokenpromptr=payloadresponses rcall_gemini_apirLks$ *G 0123#*F"3  G LL!#67}}  H ;;  MM$3  !!# ==?rc|jdg}|s$tdtj|dd|djdijdg}|D]>}d|vs|djd d }|dd }t j |}||fcS|Dcgc]}d |vs|jd d }}|rt jd|ddtd|ddcc}w)uGemini API 응답에서 이미지 데이터와 MIME 타입을 추출합니다. Args: response_data: Gemini API 응답 JSON. Returns: (이미지 바이트, mime_type) 튜플. Raises: ValueError: 이미지 데이터가 응답에 없는 경우. candidatesu.응답에 candidates가 없습니다. 응답: Nrrcontentr2 inlineDatamimeTypez image/jpegr.r3ru0이미지 데이터 없음. 텍스트 응답: %suA이미지 데이터가 응답에 없습니다. 텍스트 파트: )get ValueErrorr)dumpsbase64 b64decoderwarning) response_datarNr2part mime_type image_b64 image_bytesp text_partss rparse_image_from_responser`s(""<4J Z  KJWYXYNK[\ Cs % C1/C1cFd|vrdnd}|jj|k7r|j|}|jj dd|j |t jd|j|jjd||S)uh이미지 바이트를 MIME 타입에 맞는 확장자로 저장합니다. Args: image_bytes: 저장할 이미지 바이트 데이터. mime_type: 이미지 MIME 타입 (예: image/jpeg, image/png). output_path: 저장 경로 (확장자는 MIME 타입에 따라 보정됨). Returns: 실제 저장된 파일 경로. jpegz.jpg.pngTparentsexist_oku/이미지 저장 완료: %s (%s bytes, mime=%s),) suffixlower with_suffixparentmkdir write_bytesrrnamestatst_size)r]r[ output_pathexts r save_imagerssi'&VC!S(!--c2 TD9K( KK9     % %a (  rc |j}|jxsd}|j|d||}|t|ddddd}t j} t ||} t | \} } t| | |} t j|z } |j|t| | jjt| d| ddd|S#tj$r}t j|z } d |jjd |jj dd }t"j%d ||t| d|d <||d<Yd}~|Sd}~wt&$rV}t j|z } t"j%d||t| d|d <t||d<Yd}~|Sd}~wt($rg}t j|z } t+|j,d |}t"j%d||t| d|d <||d<Yd}~|Sd}~wwxYw)u단일 attempt에 대해 이미지를 생성하고 저장합니다. 파일명 규칙: `{stem}_attempt{attempt}{ext}` Args: token: Bearer 인증 토큰. prompt: 이미지 생성 프롬프트. base_output_path: 기본 출력 경로 (확장자 포함). attempt: 현재 시도 번호 (1부터 시작). Returns: 생성 결과 딕셔너리 (success, file_path, file_size, elapsed, error 등). rc_attemptNF)attempt file_path file_sizeelapsed_secondssuccessr&rRT)rvrwrxryr[rzr&zHTTP z: r?uAttempt %d HTTP 오류: %sryr&u&Attempt %d 이미지 파싱 오류: %su&Attempt %d 예기치 않은 오류: %s)stemrh with_namestrtimerLr`rsupdaterorproundrB HTTPErrorrKrFr3rr&rT Exceptiontype__name__)rHrIbase_output_pathrvr{rh attempt_pathresult start_timerYr]r[ saved_pathelapsedr/ error_msgs rgenerate_single_imagers4&  D  $ $ .F#--hwix.PQL& FJ#$'v6 !:=!I Y Y E ))+ * " _'__.66#(!#4&  > M)   $))+ *AJJ**+2ajjoods.C-D E   17IF$)'1$5 !#w M !))+ * =wJ$)'1$5 !a&w M $))+ *Aw''(1#.  =w R$)'1$5 !#w M$s4BC H>3A5E.. H>:A G  H>AH99H>c,|jdd}|jdd}|jdd}|jdd|d |d }t|} | |z } |d d } tjd |||| |||| t | gddd d d t j jd d } |rNPWXZ ''7 ) $ $^K%@ A  KK<!&&   NN<w'   [ LL=?P Q JJ( )C!*FIIK+-M%0L!",1-,CL()$+-  # #z'8'8 8#//0A0H0HIJ & 3 B    " " OO  #' Y*-j/ &'*4//*;*C*C &' $ W ?K:>V\*-b1':\s "' Y")+6[\f[g h W ;    !   rc tjjddtd|D}t ||z }t j jtt |||t|d|d}ttdd5}tj||d d d d d tjd ty #1swY$xYw) u생성 결과를 generation_results.json에 저장합니다. Args: results: 슬라이드별 결과 리스트. total_elapsed: 전체 소요 시간(초). Trdc3DK|]}|jdsdywrzr!N)rS).0rs r zsave_results..s?aaeeI.>?s  rR) run_timestampmodel total_slides success_count fail_countrresultswr"r#F) ensure_asciiindentNu결과 저장 완료: %s) RESULTS_JSONrkrlsumlenrrrMODEL_IDrr(r)dumprr)rrrr output_datar-s r save_resultsrsdT:?7??MW -J"113G & !&}a!8#K lC' 2@a +quQ?@ KK*L9@@s CCctjdtjd}|jdgddd|jd td d d |jddd|j S)u^CLI 인자를 파싱합니다. Returns: 파싱된 argparse.Namespace 객체. u5Gemini API v3 캠페인 이미지 생성 스크립트ub 예시: python v3_generate.py # 전체 생성 python v3_generate.py --group a # meta_carousel_a만 생성 python v3_generate.py --group gfa # naver_gfa만 생성 python v3_generate.py --max-retries 5 # 최대 5회 재시도 python v3_generate.py --dry-run # API 호출 없이 프롬프트 확인 ) descriptionformatter_classepilogz--groupr r u%생성할 그룹 선택 (기본: all))choicesdefaulthelpz --max-retriesNu0이미지당 최대 재시도 횟수 (기본: 3))rrmetavarrz --dry-run store_trueu3API 호출 없이 프롬프트 및 설정만 출력)actionr)argparseArgumentParserRawDescriptionHelpFormatter add_argumentint parse_args)parsers rrrs  $ $K << F ( 4     ?   B    rcjttdtdtdttdjtdjdtdj tdt tdtt}tj}tjd |d }j s t}g}tj}d }|D]}||vrtjd |||}tjd |t||D]}|d kDr?j s3tj!dt"tj$t"t'|||jj } |j)| |dz }tj|z } j s t+|| tdtdtdt-fd|D} j st|| z nd } tdt|j s3td| d| td| ddtdt.|D]} j r| j1drdnd}| j1dd }| j1dd }| j1d!d }t| j1d"g}| j1d#}| j1d$}|r|d%d&nd'}|r|ddnd'}| j1d(rd)| d(nd }td*|d+|d,|d-|d.|d/|d)||j rtd0td1y2y2)3u메인 실행 함수.zF======================================================================u$v3 캠페인 이미지 생성 시작u모델: u그룹: u최대 재시도: u회z Dry-run: u출력 디렉토리: u처리 대상 그룹: %srruD프롬프트 파일에 그룹 '%s'가 없습니다. 건너뜁니다.u/ [%s] 그룹 처리 시작 (%d개 슬라이드)u&rate limit 방지: %d초 대기 중...)rHrrrrr!zG ======================================================================u생성 결과 요약c3`K|]%}|jdsjr"d'ywr)rSr)rrargss rrzmain..Qs"Xa!%% 2B4<<Xs . ..u총 슬라이드: u성공: u / 실패: u총 소요 시간: z.1fu초u결과 파일: rzOKFAILrrrrnrrrrgz byteszN/Ar&z | z [z] z/slide_ru | 시도: u회 | uM [DRY-RUN 완료] 위 프롬프트로 이미지가 생성될 예정입니다.u?실제 생성하려면 --dry-run 옵션 없이 실행하세요.N)rrrrrr OUTPUT_BASEr0 PROMPTS_FILErrrrr~rXrr@rrrrrrrrS)prompts target_groupsrH all_resultsr slide_indexrslidesrrrrrrstatusrrrnrsizersize_strtime_str error_strrs @rmainrs  3v; # LQt|| EGXY ,-(#) ,, L   | , 1 K 4IIK+-M <<[-0 / ! (OX;XXM9=[!M11J s;/0 12 << {:,?@ #M##6c:;  ~./   << y)vgr"gs#uuVR quuZ,-uu&'%%/0(,d1XV$%*1gc]#&u*+%%.c!G*&b  &E7'%$8jxjH:i[ J  $ || ^_ OPr__main__)returnr})r,rrzdict[str, list[dict[str, Any]]])rHr}rIr}rdict[str, Any])rYrrztuple[bytes, str])r]bytesr[r}rqrrr) rHr}rIr}rrrvrrr) rHr}rr}rrrrrboolrr)rzlist[dict[str, Any]]rfloatrNone)rzargparse.Namespace)rr)*__doc__ __future__rrrVr)loggingrsysr~rpathlibrtypingrrBrGEMINI_API_KEYrArrr __annotations__rrrDr basicConfigINFO getLoggerrrr0rLr`rsrrrrrrrrrrs3#   (:>j%n%57 DE FG "%66"%66{*  _ 66  ?     = #   ,, ?    = ) #:. . ..h$!$$T BF F FF F  FXH HH!H H  H  H\: !:: :D$T]Q@ zFr