a jY@sdZddlZddlZddlZddlZejejeZ ej e ddZ e dZ ddZd7d d Zd d Zd8ddZddZddZddddddddddddddddZd d!d"Zd9d&d'Zd:d)d*Zd;d+d,Zd-d.Zdu| AI 漫剧 · 剧本导入器 ===================== 把小说/剧本文本： 1) 按「第X章/回/节」拆分章节 2) 每章按目标字数拆成若干「集」 3) 用 LLM 提取人物/场景/道具等资产（无 key 时退化为规则提取） 4) 生成可导入画布的 JSON（{cards, view}） 用法: python3 aicomic-import.py 小说.txt # 默认输出 aicomic-import.json python3 aicomic-import.py 小说.txt -o out.json # 指定输出 python3 aicomic-import.py 小说.txt --target 1500 # 每集目标字数 python3 aicomic-import.py 小说.txt --no-llm # 只用规则，不调 AI NZmanjuzapi_config.jsonu^\s*(第\s*[0-9一二三四五六七八九十百千零两]+\s*[章回节卷]|楔子|序章|序言|引子|尾声|番外|终章|后记|大结局|(?:CHAPTER|Chapter)\s+[0-9IVXLC]+)[\s:：、.．]*([^\n]{0,30})?cCs|d}g}d}g}|D]}tdd|}t|}|r|dur`||d|d|d|drd|dnd}g}q||q|dur||d|d|sd |dg}d d |D}|S) uO按章节标题拆章。无标题时整篇算一章。返回 [{title, content}] Nz^#+\s*)titlecontent u正文cSsg|]}|dr|qS)rstrip.0craicomic_import.py 7z"split_chapters..) splitresubHEAD_REmatchr appendjoingroup)textlineschaptersZ cur_titlecurlnmrrrsplit_chapters#s& 2 r cCsddtd|D}|s,dd|D}ggd}}}|D]@}|rj||krj|d|gd}}|||t|7}q@|r|d||p|r|gSgS)uG按段落边界把一章拆成若干集，每集目标 target_chars 字cSsg|]}|r|qSrr r prrrr>rz"split_episodes..z\n\s*\ncSsg|]}|r|qSrr r"rrrr@rrr)rr splitlinesrrlenr )r target_charsZparasepisodesrZcurlenr#rrrsplit_episodes<s   r(c Cs.i}tjtr>ztttdd}Wnty<i}Yn0g}|drt| d|dd|d|ddd tj d r| d d tj d d d tj d}tj|r*z`t|ddD]N}| d r| dddtdtd}|r| d d |d d qWnty(Yn0|S)Nutf-8encodingZapi_keyZagnesZbase_urlzhttps://apihub.agnes-ai.com/v1Z llm_modelzagnes-2.5-flash)namebasekeymodelZDEEPSEEK_API_KEYZdeepseekzhttps://api.deepseek.com/v1z deepseek-chatz~/.dsh/.credentials.yaml:r"')ospathexists CONFIG_PATHjsonloadopen Exceptiongetrenviron expanduser startswithrr chr)Zcfg providersZcredrkrrr_load_llm_configNs>         (rB c Cs6ddl}g}|D] }t|dd|dd|dg|ddd}|jj|d d d |d d |ddd}zb|jj|dd>} t| d} | ddddWdWS1s0YWqt y} z.| d|dt | ddfWYd} ~ qd} ~ 00qtdd|dS)Nrr/system)roleruserg333333?)r/Zmessages max_tokensZ temperaturer)r-/z/chat/completionszapplication/jsonzBearer r.)z Content-TypeZ Authorization)dataZheadersx)Ztimeoutchoicesmessagerz%s: %sr,uLLM 全部不可用: z | )Zurllib.requestr7dumpsencodeZrequestZRequestrstripZurlopenloadsreaddecoder:rstr RuntimeErrorr) r@rDrFrGZurllibZerrsr#bodyZreqrderrr _llm_chatms2  <:rZcCs|}|tddrVtdtdddd|}tdtdddd|}z t|WStytYn0|d| d }}|d kr||krzt|||d WStyYn0d S) u,从 LLM 回复里宽容地抠出 JSON 对象`^z [a-zA-Z]* rr${}rrN) r r>r?rrr7rQr:findrfind)rsrYrrr _extract_jsons   rdc Csg}|D]8}tt|dd}||dd|dd|qd|dd}d}d |d }t|||}t|pzi} | d g| d g| d g| ddt| d t rg} | d D]}t|t rJ|drJ| t |d t |dd t |dd t |dd t |dd dqt|t r| r| | dddddq| | d <| S)uLLM 提取资产，返回 {'title','characters':[...],'scenes':[...],'props':[...]} 通读每章开头较多内容（合计最多约10万字），保证不遗漏。ri@rrNz iu你是资深网文编辑与漫剧制作人。请通读下面小说节选，完整提取制作漫剧所需的结构化信息。只输出一个 JSON 对象，不要任何其他文字。uI小说节选（较长，请通读全部内容，不要只看开头）： u? 请输出 JSON，字段如下： { "title": "小说/剧本名", "characters": [{"name": "角色名", "role": "主角/重要配角/反派/路人等", "description": "外貌与性格，20字内", "outfit": "标志性服装，如：灰色卫衣+牛仔裤", "feature": "标志特征，如：左眼角泪痣"}], "scenes": ["场景名，如：林府大门·清晨"], "props": ["关键道具名，如：木剑"] } 要求： 1. 完整提取：所有出场过且有名字/有戏份的角色都要列入，配角、反派、关键路人也不要遗漏； 2. 所有出现过的主要地点都要列入场景； 3. 反复出现或推动剧情的关键道具都要列入； 4. 上限：characters 最多25个，scenes 最多25个，props 最多20个，宁多勿漏； 5. 角色名必须从文中出现的人名提取，不要编造。 charactersscenesprops未命名剧本r,rEr descriptionoutfitfeature)r,rErirjrk) minr%rrrZrd setdefault isinstancer;listdictrTr ) rr@partsr ZtakecontextrDrFrawrXZfixedrrrextract_assets_llms6$         rtu日系赛璐璐（默认）zjapanese anime cel-shaded key visual, clean bold lineart, vibrant flat colors, cinematic rim lighting, soft shadows, expressive anime eyes, professional anime illustration)r,enu 国漫厚涂zchinese manhua thick-paint style, rich painterly brushwork, dramatic chiaroscuro, vivid saturated colors, expressive character art, professional manhua illustrationu 水墨仙侠zchinese ink-wash painting style, ethereal mist, flowing sumi-e brushstrokes, xianxia fantasy atmosphere, poetic negative space, high detailu写实电影感zcinematic photorealistic style, dramatic volumetric lighting, film grain, shallow depth of field, 8k, hyper-detailed, professional film stillu3D渲染z{high quality 3d render, soft global illumination, subsurface scattering, detailed character model, octane render, cinematic)celZguomanZinkZ realisticZ3dz#high quality, detailed, sharp focuszwmasterpiece, best quality, ultra-detailed, intricate details, 8k, sharp focus, professional illustration, award winning)ZstdhighrrvrwcCs|pg}|pddd}t|p,td}|d}t|td} g} |D]} | dd} g} | dr~| d | d| d r| d | d | d r| d | d | r| r| d| d| fqN| rN| d| qNd}t}|rd|pd| rdd| nd|r*d|ndf}zFt|d|dd}|pLddd}|rvt|dkrv|}Wnt yd}Yn0|s|p|rd|r|dddnddnd}|g} | d|| | | dd d!d"| DS)#u把中文分镜改写成英文画面描述（LLM），并由系统统一追加精修画风包+质量词， 保证多镜头画风稳定、人物细节出色。rN,rvrurwr, appearanceu 外貌：rju 服装：rku标志特征：u角色「%s」：%su，u角色「%s」u请把下面的漫剧分镜信息改写成一段用于 AI 生图的英文画面描述（2~4句，只写画面本身：人物外貌与动作、景别（近景/中景/全景）、光线、氛围、构图）。出场角色的外貌/服装/标志特征必须准确写进描述中，同一角色每次描述保持一致。不要加画风词、不要加质量词、不要加"vertical composition"等词，这些由系统统一追加。直接输出英文，不要任何解释。 场景：%s %s%su（未描述）ru 动作/光线细节：u!你是资深动漫分镜画师。irG"'u 角色「ru」zstyle: z0vertical 9:16 composition, no text, no watermark, cSsg|]}|r|r|qSrr r xrrrrrz)optimize_image_prompt..) r STYLE_PACKSr;QUALITY_TOKENSrrrBrZr%r:)sceneZdetailcharsmoveZstyleZ style_packZqualityZpackZstyle_enZ quality_enZ char_infor ZnmrqZvisualr@rFrsrrroptimize_image_promptsN   2  rc Cs|dg}dd|D}d}d|||ddd|p:d f} t||| d d } t| pZi} g} | d plgD]} t| ts~qnt| d d}|sqn| |t| ddt| ddt| ddpddd| dpgDdqn| S)u\把一集剧情用 LLM 拆成 3~5 个分镜镜头，返回 [{scene,prompt,lines,move,chars}]recSs"g|]}|dr|ddqS)r,rr;r rrrrrz&storyboard_episode..ua你是漫剧资深分镜师。把给定的小说/剧本片段拆成一组连贯的分镜镜头。uO章节：%s，第%d集 剧情片段： %s 出场角色池：%s 请只输出 JSON：{"shots":[{"scene":"画面描述（视觉化，含人物动作/景别/光线/情绪，可直接用于AI生图）","prompt":"画面补充（构图/氛围/细节，可为空字符串）","lines":"本镜头台词或旁白，格式：旁白：xxx 或 角色名：xxx；无台词写：无","move":"推镜/拉镜/摇镜/静止","chars":["出场角色名"]}]} 要求：拆成 3~5 个镜头；镜头顺序与剧情连贯；chars 只从角色池里选；lines 口语化带情绪；move 四选一；只输出 JSON。Nipr~u （无）i rzshotsrrpromptrr静止cSs g|]}t|rt|qSr)rTr rrrrr/rrrrrrr) r;rrZrdrnrprTr r)epZch_titleidxassetsr@Z target_shotsrZ char_namesrDrFrsrXrZshrrrrstoryboard_episodes.      rc Cs|durt}i}|s|St|D]r\}}tt||gD]T}|||}z$t||d|d|||||f<Wq>tyg|||f<Yq>0q>q"|S)uC给所有集生成分镜，返回 {(章序号, 集序号): [shots]}Nrr)rB enumerateranger%r;rr:) r episodes_maprr@boardichjrrrrstoryboard_all4s $ rc Cs@ddd|D}i}td|D]6}|d}t|dkr$|dvr$||dd||<q$d d |Dd d }g}td |D](}|d}|r||vr||qg}td|D](}|d} | r| |vr|| qtt j dkrt j t j ddddnddd |D|d d|d ddS)uU规则兜底：对话归属+词频提人物；地点后缀提场景；量词提道具rcss|]}|dVqdS)rNrr rrr Grz+extract_assets_heuristic..ui([\u4e00-\u9fa5]{1,4})(?:说|道|问|喊|答|心想|低语|叹|喝|惊|怒道|笑道|沉声道|淡淡道)rr)u他说u她说u众人u大家u其中rcSsg|]\}}|dkr|qS)r\r)r nr rrrrMrz,extract_assets_heuristic..N u(?:来到|走进|回到|离开|站在|位于|在|赶往|逃到)([^，。！？；]{0,10}?)(?:的)?(?:山|城|府|殿|宫|院|镇|村|洞|林|谷|塔|楼|阁|河|湖|门|庙|厅|房|屋|堡|营|寨|堂|场|地)u(?:一把|一柄|一枚|一颗|一包|一件|一瓶|一卷|一块|一顶|一身|一根|一册|一页|一个|一只|一支|一张|一条|一盏|一份)([^，。！？\s]{1,6}).rhcSsg|]}|dddqS)r)r,rErirr rrrrrZr )rrerfrg)rrfinditerrr%r;itemsr rsysargvr3r4basenamersplit) rZfullrrrnamesrfrcrgr#rrrextract_assets_heuristicEs,     0   rcspig}dgfddfdd}fdd}fdd }fd d } td d |D} d} tt|D]<tt|gD]"} | tt| fgd7} qqv||ddd|t|| | t|dgt|dgt|dgft|dgD]$\} }||dd| d|qi|D](}|ddkr>|d|dd<q>d}t|D]&\}|g}|||dd|dt|dfd d tt|Dftt|D]}|fp || dd!d"d#|| dd$d%gd&g}t|D]L\}}d'|d|d|df}|||d(|dd|d||q*q|d(tt|ddd)7}qtd*dd+d |dgDd,dd-d |dgD}|| |dd.||dddd/d0S)1Nics dd7<dtdS)Nrrr )rTr)uidrrcidfszbuild_canvas_json..cidcsd||dd|idS)NnoterridtyperyZstatusrIr)rrrrrrrjszbuild_canvas_json..notecsXd||d||ddfdd|dgD|dd|dd|d d d d S) Nrrrcsg|]}|vr|qSrrr)name_idrrrqrz4build_canvas_json..scene..rrrrr)rrZcharIdsrrrrr)rrrshot)rrrrrms     z build_canvas_json..scenec sBd||d|dd|dd|dd|dddd d d S) Ncharrr,u角色rirrjrku6由剧本导入提取，可再补一张参考图更稳)r,ryrjrkZrefImgZrefNoterr)rrr rrrrvs     zbuild_canvas_json..charcsd||d||ddS)Ntoolr)rrVrr)rrtbrrrrs zbuild_canvas_json..toolcss|]}t|VqdS)Nr%)r vrrrrrz$build_canvas_json..rr(u📖 剧本导入总览 《%s》 章节：%d 章 集数：%d 集 分镜镜头：约 %d 个 人物：%d 位 场景：%d 个 道具：%d 件 ■ 左侧：角色资产库（服装/特征已自动提取） ■ 中间：每章一列，每集一排分镜镜头卡（已填画面/台词/运镜） ■ 右上：场景/道具清单 ■ 直接点镜头卡「🎨 生图」出图，复制「台词」去配音rerfrgi@ihrrrrIr,i0u📚 %s 共拆为 %d 集： %srrc3s2|]*}d|dt|fgp$dfVqdS)u· 第%d集（%d个镜头）rN)r%r;)r r)rcirrrrru 旁白：Prru%s · 第%d集 · 镜头%drx<u 【场景】 css|]}d|VqdSu· Nr)r rcrrrrru 【道具】 css|]}d|VqdSrrr"rrrrru🧰 场景 / 道具资产)ZtxZtyzcardsview) sumvaluesrr%r;maxrrrr )rrrrrrrrrrZ total_epsZ total_shotsZejrr rrZepsrrrArZep_titlerVr)rrrrrrbuild_canvas_jsonasT   "  *    $0"@rTc s>t|}it|D]\}}t|d||<qd}|rht}|rhzt||}Wntyfd}Yn0|durxt|}i}t}|r|rt|||}t| dd|||} i} t|D]J\}}t t |gD],} t | || fg| d|d| df<qqd| ddfddt|D|| | d | d d S) Nrrrhu %s·第%d集rTcs&g|]\}}|dt|dqS)r)rr'r)r rr rrrrrzrun_import..rr)okrrrrrr) r rr(rBrtr:rrrr;rr%) rr&use_llmrrrrr@rZcanvasZ shots_summaryrrrr run_imports>  , rc Cstjdd}|s$tttd|d}d}d}d}d|vrR||dd}d|vrpt||dd}d|vr|d }tj |std |tdt |d d  }t |||d }t |dd d 0}t j|d|dd|d ddWdn1s0Ytdtd|dtdt|ddtdd|dDtdt|ddgdd d!d|ddgDdd"td#t|dd$gd%t|dd&gtd'tj|td(dS))Nrrzaicomic-import.jsonr!Tz-oz--targetz--no-llmFu文件不存在: %sr)r*)r&rwrrrr)Z ensure_asciiindentu✅ 导入完成u 标题:ru 章节:ru 章 / 集数:css|]}|dVqdS)r'Nrr rrrrrzmain..u 人物:rreu位 ->r~css|]}|dVqdS)r,Nrr rrrrrru 场景:rfu 个 / 道具:rgu 输出:u< 提示: 打开画布 → 点「⬆ 导入」选这个 json)rrprint__doc__exitindexintr3r4r5r9rRrr7dumpr%rr;rabspath)argsr4outtargetrrresfrrrmains8    @(B.r__main__)r!)rC)rrNrrrvrw)r)N)N)r!T)rr7r3rrr4dirnamer__file__ZBASE_DIRrr6compilerr r(rBrZrdrtrrrrrrrrr__name__rrrrsZ  . 7 "  F #