i*ddZddlZddlmZmZmZmZddlmZddl m Z dZ dZ dd d Z d Zd ed efdZdeded eeeffdZdeeeefd eeeeffdZdeeeefd eeeeffdZdeeefd efdZdeeefdeeeefd eefdZdeeefded dfdZGddZy)u^웹북 패키지 생성 모듈 — 시나리오 에피소드를 웹북 에피소드로 분할.N)AnyDictListOptional)call_structured) load_prompti`idkoKorean)enEnglish)r r prototype_promptsfilenamereturncl|jdr|jddn|}tt|S)Nz.md)endswithreplacer_MODULE)rstems S/Users/manta/Documents/Projects/TheRoad-I1/backend/app/modules/webbook_generator.py _load_promptrs0*2*;*;E*B8  E2 &D w %%web_episode_countsections_per_episodec ddddiddiddidddddiddddddidddd ddidd dd d d gd d }ddd d|d ddiddidddddidd|||ddddddiddgdd }ddddiddiddid|||ddgdd S)NobjectFtypestringarray)rminItemsmaxItemsitemsinteger)rminimummaximum)still_id section_title image_captionopening_paragraphsclosing_paragraphs paragraphsimage_after_paragraph)radditionalProperties propertiesrequired)episode_numbertitlesubtitleopener_paragraphssectionscloser_paragraphs) series_titleadaptation_subtitleloglineepisodes)rrsection_schemaepisode_schemas r_make_webbook_package_schemarBsL %*$h/$h/ (+ #  (+ #  (+ "&- 8 ?(&NT %", h'* (+ " 00'   (+ ") 6 =&&NP %#X.$*H#5)--'     rentitiesc g}|D]X}|j|jdd|jdd|jdd|jdddZ|S)Nnamer entity_type description stable_traitsz{})rErFrGrHappendget)rCcompactitems r_compact_entitiesrNsaG,#xx r:#xx r:!%/4!@    Nrstillsc g}|D]i}|j|jdd|jdd|jdd|jdd|jdddk|S) N still_indexridrscreenplay_scene_heading beat_titlestill_frame_prompt)rQrRrSrTrUrI)rOrLss r_compact_stillsrWspG  uu]A6eeD"o,-EE2Lb,QeeL"5&'ee,@"&E    Nrepisodec d}dD]'}|td|j|gDz })|jdgD]}|t|jddz }|t|jddz }|td|jd gDz }|td |jd gDz }|td |jd gDz }|S)Nr)r8r:c32K|]}t|ywNlen.0texts r z+_episode_character_count..sA+@4SY+@r9r+rr,c32K|]}t|ywr[r\r^s rraz+_episode_character_count..sI+H4SY+Hrbr/c32K|]}t|ywr[r\r^s rraz+_episode_character_count.. "G$CI"Grbr-c32K|]}t|ywr[r\r^s rraz+_episode_character_count..rerbr.)sumrKr])rXtotalpartsections r_episode_character_countrks E: A7;;tR+@AAA;;;z2. W[["566 W[["566 I7;;|R+HIII  ")++.BB"G     ")++.BB"G   / Lrpayloadavailable_stillsc 0|Dchc]}|jdd}}g}|jdgD]Z}|jdgD]C}|jdd}|s||vs|jd|d|jdd E\|Scc}w) zEValidate still_ids in the package exist in the available stills list.rRrr>r9r*z still_id 'z ' in episode r5z not found in available stills)rKrJ)rlrmrV available_idswarningsrXrjsids r_validate_still_coveragerrs/??.>QUU4_.>M?H;;z2.{{:r2G++j"-Cs-/ ]7;;?O3P2QR453/ O@sBmin_episode_charsc|jdg}t|tr|s tdg}|D]8}t |}||ks|j |jdd|:|rtddj |y)Nr>z)Webbook package did not include episodes.r5:z!Webbook episodes were too short: z, )rK isinstancelist RuntimeErrorrkrJjoin)rlrsr> too_shortrX char_counts r_validate_webbook_packager|s{{:r*H h %XFGGI-g6 ) )    ,< =>a |L M/ )0D/E F  rceZdZdZddeeddfdZ ddedeeee fdeeee fd eee fd ed e d e deee ffd Z y)WebbookGeneratoru웹북 패키지 생성기.Nproject_llm_configrc||_yr[)_project_config)selfrs r__init__zWebbookGenerator.__init__s 1rfulltextrCrO world_guidelanguagerrc 6tj|d\}} tdj|| ||} tdjd||t j |dd|t j t |ddt j t|dd } t||} d } td d D]}td | | | |jdt}|dDcgc]}|dD]}|d }}}t|tt|k7r t|t d|||d|d<t#||}|r||dd<|cSt%d| cc}}w#t$$r }|} Yd }~nd }~wwxYw|dks| dz } )awGenerate webbook package for an episode. Takes screenplay fulltext, entities, scene stills, world guide, and language. Calls LLM via LiteLLM Router for structured JSON response. Returns webbook package dict with episodes, sections, text blocks including opening_paragraphs, image_caption, and closing_paragraphs per section. r zwebbook_package_system.md)source_language_codesource_language_namerrzwebbook_package_user.mdrXFr4) ensure_asciiindent) source_filerrworld_guide_jsonscreenplay_text entities_jsonscene_stills_jsonNr!r& webbook_genprototype_webbook_package)step system_prompt user_promptresponse_schemaproject_config schema_name max_tokensr>r9r*)rsv5)prompt_versionrrrgeneration_metadatastill_coverage_warningsr"z Regeneration correction: - The last output was too short or structurally weak. - Make each episode materially longer. - Keep the same strict schema. - Preserve chronology and still uniqueness. z/Failed to generate sufficient webbook package: )LANGUAGE_NAMESrKrformatjsondumpsrNrWrBrangerr!WEBBOOK_PACKAGE_MAX_OUTPUT_TOKENSr]setr|WEBBOOK_TARGET_EPISODE_CHARSrrrx)rrrCrOrrrr lang_code lang_namerrschema last_errorattemptrlrXrj still_idscoverage_warningsexcs rgeneratezWebbookGenerator.generates& .11(2T2"#$&c c tCH~&c T#s(^$ c #s(^ c  c c "c  c3hc rr~)rrtypingrrrrapp.modules.llm.llm_clientrapp.modules.prompt_loaderrrrrrrrrrBrNrWrkrrr|r~r?rrrsRd ,,61$)!#   &3&3& gg25g #s(^gT T#s(^ 4 d38n9M  Dc3h0 T$sCx.5I d38n" #s(^/3DcN/C #Y" #s(^ 03   "i i r