AI_NovelGenerator/__pycache__/novel_generator.cpython-310.pyc

o
ýOœg@eã@s
dd
lZdd
l
Z
dd
lZddlmZmZmZ
zddlmZ
Wney-


ddl	mZ
Yn
wddl
mZ
ddlm
Z
mZmZ
ddl
mZ
ddlmZ
ddlmZ
dd
lZdd
lZdd	lmZ
dd
lmZ
ddlmZmZmZmZ
ddl m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)
e
j*e
j+d
d�
dZ,de-de-dee-defdd„Z.de-de-deefdd„Z/de-de-de-dd
fdd„Z0ddde-de-de-de1de-f
dd„Z2Gd d!„d!eƒZ3	"dede-de-d#e-d$e-d%e-d&e1d'e1d(e-d)e4dd
fd*d+„Z5id,d“
d-d“
d.d“
d/d0“
d1d“
d2d3“
d4d5“
d6d7“
d8d9“
d:d;“
d<d=“
d>d?“
d@dA“
dBdC“
dDdE“
dFdG“
Z6dHe-de1fdIdJ„Z7	0dfdKe-dLe1dMe1de-fdNdO„Z8	P	"dgdQe-dRe-de-de-dSe-dLe1d(e-d'e1dTe-dUe-d)e4de-fdVdW„Z9de-de-dXe-dd
fdYdZ„Z:	"	[dhd\e-d]e4d^e1dee-fd_d`„Z;didae-d^e1dee-fdbdc„Z<d
S)jéN)ÚDictÚListÚOptional)
Ú	TypedDict)
Ú
ChatOpenAI)Ú
StateGraphÚSTARTÚEND)
ÚOpenAIEmbeddings)
ÚChroma©
ÚDocument)
ÚSentenceTransformer)
Úcosine_similarity)Ú	read_fileÚappend_text_to_fileÚclear_file_contentÚsave_string_to_txt)	Ú
set_promptÚcharacter_promptÚdark_lines_promptÚfinalize_setting_promptÚnovel_directory_promptÚsummary_promptÚupdate_character_state_promptÚchapter_outline_promptÚchapter_write_promptz'%(asctime)s [%(levelname)s] %(message)s)ÚlevelÚformatÚvectorstoreÚapi_keyÚbase_urlÚtextsÚreturncCs6t||
d
�}dd„|Dƒ
}t
j||td�}| ¡
|S)uŸ
    �始化并返回一个Chroma��库，将传入的文本进行嵌入并�存到本地目录。
    如果�存在该目录，会自动创建。
    ©Zopenai_api_keyZopenai_api_basec
Sóg|]}
t|
d�
‘qS©
©
Zpage_contentr)Ú.0Ú
t©r*ú;c:\Users\xieli\Desktop\AI_NovelGenerator\novel_generator.pyÚ
<listcomp>3óz%init_vector_store.<locals>.<listcomp>)Z	embeddingÚpersist_directory)r
rZfrom_documentsÚVECTOR_STORE_DIRÚpersist)r r!r"Ú
embeddingsZ	documentsrr*r*r+Úinit_vector_store*s

þ



ý
r2cCs(tj
 t¡
sd
St||
d�}tt|d�S)u;读å�–已存在的å�‘é‡�库。若ä¸�存在则返回 None。Nr$)r.Zembedding_function)ÚosÚpathÚexistsr/r
r)r r!r1r*r*r+Úload_vector_store<s



þr6Únew_chaptercCsLt||
ƒ}|st
 d
¡

t||
|g
ƒ
dSt|d�
}| |g
¡

| ¡
dS)uf将最新章节文本æ�’入到å�‘é‡�库里，用于å�Žç»­æ£€ç´¢å�‚考。若库ä¸�存在则åˆ�始化。z6Vector store does not exist. Initializing a new one...Nr')r6ÚloggingÚinfor2r
Ú
add_documentsr0)r r!r7ÚstoreZnew_docr*r*r+Úupdate_vector_storeFs








r<éÚqueryÚ
kcCsBt||
ƒ}|st
 d
¡

dS|j||d�}d dd„|Dƒ
¡
}|S)uˆ
    从��库中检索与 query 最相关的 k �文本，拼接�返回。
    若��库�存在则返回空字符串。
    z0Vector store not found. Returning empty context.Ú©
r?Ú

c
Ssg|]}
|
j‘qSr*r')r(Ú
dr*r*r+r,\sz:get_relevant_context_from_vector_store.<locals>.<listcomp>)r6r8ÚwarningZsimilarity_searchÚjoin)r r!r>r?r;ÚdocsZcombinedr*r*r+Ú&get_relevant_context_from_vector_storeRs







rGc@sVeZ
dZUeed
<eed<eed<eed<eed<eed<eed<eed<eed	<d
S)ÚOverallStateÚtopicÚgenreÚnumber_of_chaptersÚword_numberÚnovel_setting_baseÚcharacter_settingÚ
dark_linesÚfinal_novel_settingÚnovel_directoryN)Ú__name__Ú
__module__Ú__qualname__ÚstrÚ__annotations__Úintr*r*r*r+rHas









rHçffffffæ?Ú	llm_modelrIrJrKrLÚfilepathÚtemperaturec	stj
|d
d�
t|||
|d�‰
dtdtfdd„‰dtd	tttff‡‡
fd
d„}	dtd	tttff‡‡
fdd
„}
dtd	tttff‡‡
fdd„}dtd	tttff‡‡
fdd„}dtd	tttff‡‡
fdd„}
ttƒ
}| d|	¡
| d|
¡
| d|¡
| d|¡
| d|
¡
| t	d¡
| dd¡
| dd¡
| dd¡
| dd¡
| dt
¡
| ¡}||||dœ}| |¡
}|sÇt
 d¡

dS| dd¡}| dd¡}|r×|sÞt
 d¡

dStj |d ¡}tj |d!¡}d"td	tfd#d$„}||ƒ
}||ƒ
}t||ƒ
t||ƒ
t
 d%¡

dS)&uÿ

    使用多步�程，生� Novel_setting.txt 与 Novel_directory.txt 并�存到 filepath。

    :param api_key:       OpenAI API key
    :param base_url:      OpenAI API base url
    :param llm_model:     所使用的 LLM 模型�称
    :param topic:         �说主题
    :param genre:         �说类型
    :param number_of_chapters: 章节数
    :param word_number:   �章目标字数
    :param filepath:      存放生�文件的目录路径
    :param temperature:   生�温度
    T©
Úexist_ok©Úmodelr r!r[ÚpromptÚresponse_contentcSó(t 
d
|›d�¡

t 
d|
›d�¡

dS)u@在控制�打�或记录下�次Prompt与Response，[调试]z
[Prompt >>>] rBz[Response <<<] N©r8r9©r`rar*r*r+Ú	debug_logŽó
z1Novel_novel_directory_generate.<locals>.debug_logÚstater#c
s\tj
|d
|d|d|dd�}
ˆ
 |
¡
}|s!t d¡

ddi
Sˆ|
|jƒ
d|j ¡i
S)	NrIrJrKrL©rIrJrKrLz#generate_base_setting: No response.rMr@)rrÚinvoker8rDÚcontentÚstrip©rgr`Úresponse©rer_r*r+Úgenerate_base_setting“s




ü






z=Novel_novel_directory_generate.<locals>.generate_base_settingc
óJtj
|d
d�
}
ˆ
 |
¡
}|st d¡

ddi
Sˆ|
|jƒ
d|j ¡i
S)NrM)
Ú
novel_settingz(generate_character_setting: No response.rNr@)rrrir8rDrjrkrlrnr*r+Úgenerate_character_setting¡ó

ÿ






zBNovel_novel_directory_generate.<locals>.generate_character_settingc
rp)NrN)
Zcharacter_infoz!generate_dark_lines: No response.rOr@)rrrir8rDrjrkrlrnr*r+Úgenerate_dark_lines¬rsz;Novel_novel_directory_generate.<locals>.generate_dark_linesc
sVtj
|d
|d|dd�}
ˆ
 |
¡
}|st d¡

ddi
Sˆ|
|jƒ
d|j ¡i
S)NrMrNrO)rMrNrOz$finalize_novel_setting: No response.rPr@)rrrir8rDrjrkrlrnr*r+Úfinalize_novel_setting·s



ý






z>Novel_novel_directory_generate.<locals>.finalize_novel_settingc
sPtj
|d
|dd�}
ˆ
 |
¡
}|st d¡

ddi
Sˆ|
|jƒ
d|j ¡i
S)NrPrK)rPrKz&generate_novel_directory: No response.rQr@)rrrir8rDrjrkrlrnr*r+Úgenerate_novel_directoryÄs


þ






z@Novel_novel_directory_generate.<locals>.generate_novel_directoryrorrrtrurvrhuGNovel_novel_directory_generate: invoke() 结果为空，生æˆ�失败。NrPr@rQu@生æˆ�失败：缺少 final_novel_setting 或 novel_directory。zNovel_setting.txtzNovel_directory.txtÚtxtc

Ss| d
d¡ dd¡S)Nú
#r@Ú
*)
Úreplace)
rwr*r*r+Ú
clean_textùs
z2Novel_novel_directory_generate.<locals>.clean_textz4Novel settings and directory generated successfully.)r3ÚmakedirsrrUrHrrZadd_nodeZadd_edgerr	Úcompilerir8rDÚgetr4rErr9)r r!rYrIrJrKrLrZr[rorrrtrurvZgraphÚappZinput_paramsÚresultrPZfinal_novel_directoryZfilename_setZfilename_novel_directoryr{Zfinal_novel_setting_cleanedZfinal_novel_directory_cleanedr*rnr+ÚNovel_novel_directory_generatels`



ü     














ü












r�u零uâ—‹u〇u一é
u二u三éuå››éu五éuå…­éu七éuå…«éuä¹�é	uå��é
u百éduå�ƒièu万i'Úchinese_strc
Csnd
}
d}d
}t|ƒ
D]&}|t
vr/t
|}|dkr(||kr|}q
|
||7}
d
}q
|||}q
	q
|
|7}
|
S)u\
    �能处�到万(10000)以内的中文数字，正常�说章节应该够用了
    r
r‚rŠ)ÚreversedÚCHINESE_NUM_MAP)rŒÚtotalZcurrent_unitZtmp_valÚcharÚvalr*r*r+Úchinese_to_arabic

s







r’Únovel_directory_textÚnovel_numberÚ
range_sizecCs
| ¡}gd
¢
}i}|D]A}| 
¡}|sqd}|D]0}tj||tjd�}	|	rI|	 d¡
}
|	 d¡
 
¡}t d|
¡r=t|
ƒ
}nt|
ƒ
}|||<d}
n
q|rMqqg}
t|
||
|dƒD]%}||vr€||
krs|
 	d|›d	||›�¡

q[|
 	d
|›d	||›�¡

q[|
rˆd 
|
¡
SdS)
uë
    从�说目录文本中，��指定章节（以���几章）的目录信�。
    range_size=1，表示获�当�章节��一章和�一章的目录信�(若存在)。
    支�多�常�的章节格�。
    )u#^第\s*([\d]+)\s*章[:：]?\s*(.*)$uQ^第\s*([零○〇一二三四五六七八��百�万]+)\s*章[:：]?\s*(.*)$u"^Chapter\s+(\d+)\s*[:：]?\s*(.*)$u^Ch\s+(\d+)\s*[:：]?\s*(.*)$u#^第\s*([\d]+)\s*节[:：]?\s*(.*)$u#^第\s*([\d]+)\s*�[:：]?\s*(.*)$F)
Úflagsr‚r=u5^[零○〇一二三四五六七八ä¹�å��百å�ƒä¸‡]+$Tuã€�当å‰�】第u章：u第rBr@)Ú
splitlinesrkÚreÚmatchÚ
IGNORECASEÚgroupr’rWÚrangeÚappendrE)r“r”r•ÚlinesÚpatternsZ
directory_mapÚlineZmatchedÚpatr™Zchapter_num_strZ
chapter_titleZchapter_numZ
chapters_infoZcnumr*r*r+Ú"parse_chapter_title_from_directory+
sB	












ó
ÿ



€


r¢r@Únovel_settingsÚnovel_novel_directoryÚ
model_nameÚlastchapterÚ
user_guidancecs.tj
|d
d�
t||||
d�‰
dtdtfdd„‰tj |d¡}tj
|d
d�
tj |d	|›d
�¡}tj |d¡}
tj |d¡}tj |d
¡}t|ƒ
}t|ƒ
}dtdtdtf‡‡
fdd„}| ¡rf|||ƒ}n|}dtdtdtf‡‡
fdd„}| ¡r€|||ƒ}n|}t||ddd�}t	|
|dd�}dtdtdtdt
dtd td!tdtf‡‡
fd"d#„}||||||||	ƒ}dtdtdtd$td%t
dtd td!tdtf‡‡
fd&d'„}|||||||||	ƒ}|�
rt||ƒ
t|
ƒ

t||
ƒ
t|ƒ

t||ƒ
t|ƒ

t||ƒ
t
|||ƒ
t d(|›d)�¡

|St d(|›d*�¡

|S)+uN
    多步�程:
      1) 更新/创建全局摘�
      2) 更新/生�角色状�文档
      3) ��检索获�往期上下文
      4) 从Novel_directory.txt中获�当�(和��几章)的目录信�
      5) 大纲 -> 正文（�结�用户给出的�外指导）
      6) 写入 chapter_{novel_number}.txt, 更新 last_chapter.txt
      7) 更新��库

    :param novel_settings:        最终的作�设定（字符串）
    :param novel_novel_directory: �说目录信�
    :param api_key:              OpenAI API Key
    :param base_url:             OpenAI Base URL
    :param model_name:           LLM 模型�称
    :param novel_number:         当��生�的章节�
    :param filepath:             文件存放的目录
    :param word_number:          �章目标字数
    :param lastchapter:          上一章内容（若为空字符串，表示无上一章）
    :param user_guidance:        用户对当�章节的�外指导或想法
    :param temperature:          生�温度
    :return:                     本章生�的正文内容
    Tr\r^r`racSrb)uY在控制�打�或记录下�次的 Prompt 与 Response，便于观察生�过程。z
[Prompt >>>]
rBz[Response <<<]
Nrcrdr*r*r+re¥
rfz.generate_chapter_with_state.<locals>.debug_logZchaptersZchapter_z.txtzlast_chapter.txtzcharacter_state.txtzglobal_summary.txtÚchapter_textÚold_summaryr#có@tj
||
d
�}ˆ
 |¡
}|st d¡

|
Sˆ||jƒ
|j ¡S)N)r¨Úglobal_summaryz#update_global_summary: No response.)rrrir8rDrjrk)r¨r©r`rmrnr*r+Úupdate_global_summary·
ó


þ







z:generate_chapter_with_state.<locals>.update_global_summaryÚ	old_statecrª)N)r¨r®z$update_character_state: No response.)rrrir8rDrjrk)r¨r®r`rmrnr*r+Úupdate_character_stateÉ
r­z;generate_chapter_with_state.<locals>.update_character_stateu回顾剧情r=rAr‚)
r•rqÚ
char_stater«Úchap_numÚ
extra_contextÚdirectory_hintÚ
user_guidec	s\td
|d|j
||
d|||d�}ˆ
 |¡
}|s#t d¡

dSˆ||jƒ
|j ¡S)uY
        将目录�示以�用户�外指导内容一起放入 Prompt 中。
        õ

�目录�考】
õ

�用户指导】
õ

�历�上下文】
)rqÚcharacter_stater«r”zoutline_chapter: No response.r@)rrrir8rDrjrk)	rqr°r«r±r²r³r´Zoutline_promptrmrnr*r+Úoutline_chapterã
s*
ÿ
ÿþþ




ø







z4generate_chapter_with_state.<locals>.outline_chapterÚoutlineÚwnumc
s^td
|d|j
||
d||||d�}ˆ
 |¡
}	|	s$t d¡

dSˆ||	jƒ
|	j ¡S)Nrµr¶r·)rqr¸r«Zchapter_outlinerLzwrite_chapter: No response.r@)rrrir8rDrjrk)
rqr°r«rºr»r²r³r´Zwriting_promptrmrnr*r+Ú
write_chapters,
ÿ
ÿþþ





÷







z2generate_chapter_with_state.<locals>.write_chapterzChapter z generated successfully.z generation failed.)r3r|rrUr4rErrkrGr¢rWrrr<r8r9rD)r£r¤r r!r¥r”rZrLr¦r§r[Zchapters_dirZchapter_fileZlastchapter_fileÚcharacter_state_fileÚglobal_summary_fileZold_char_stateZold_global_summaryr¬Znew_global_summaryr¯Znew_char_stateZrelevant_contextZthis_and_related_chaptersr¹Zchap_outliner¼Zchapter_contentr*rnr+Úgenerate_chapter_with_statev
s¶%



ü







ÿ
ÿþýüûúùø

þ
ÿþýüûúùø	÷







ø








þr¿Ú	file_pathcCsœtj
 |¡
st d
|›�¡

dSt|ƒ
}| ¡st d¡

dSt|ƒ
}t||
ƒ}|s7t 	d¡

t
||
|ƒ
dSdd„|Dƒ
}| |¡

| ¡
t 	d¡

dS)uŒ
    将用户选定的文本文件导入到��库，以便在写作时检索。
    �以在UI中�供按钮�调用此函数。
    u知识库文件ä¸�存在: Nu知识库文件内容为空。zKVector store does not exist. Initializing a new one for knowledge import...c
Sr%r&r)r(Ú
pr*r*r+r,dr-z)import_knowledge_file.<locals>.<listcomp>u-知识库文件已�功导入至��库。)
r3r4r5r8rDrrkÚadvanced_split_contentr6r9r2r:r0)r r!rÀrjZ
paragraphsr;rFr*r*r+Úimport_knowledge_fileHs"