72 lines
10 KiB
Plaintext
72 lines
10 KiB
Plaintext
|
o
|
|||
|
|
Øšg= ã @ sz d dl Z d dlZd dlmZmZmZmZ d dlZd dlm Z m
|
|||
|
|
Z
|
|||
|
|
d dl m Z d dl
|
|||
|
|
mZmZ d dlmZ G dd„ dƒZdS ) é N)ÚttkÚ
|
|||
|
|
filedialogÚ scrolledtextÚ
|
|||
|
|
messagebox)Ú load_configÚ save_config)Ú read_file)Ú Novel_novel_directory_generateÚgenerate_chapter_with_state)Úcheck_consistencyc @ sr e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd d
|
|||
|
|
„ Zd d „ Zd
|
|||
|
|
e fdd„Z
|
|||
|
|
dd„ Z dd„ Z dd„ Z
|
|||
|
|
dd„ Zdd„ ZdS )ÚNovelGeneratorGUIc C s� || _ | j d¡ d| _t| jƒ| _tj|dd�}|jddd� t |¡| _ |j
|
|||
|
|
| j dd � tj|d
|
|||
|
|
d �| _ |j
|
|||
|
|
| j d d � | ¡ |
|
|||
|
|
¡ d S )
|
|||
|
|
Nz Novel Generator GUI - Advancedz config.jsonÚ
|
|||
|
|
horizontal)ZorientÚbothT)ZfillÚexpandé ©Zweightz 10 10 10 10)Zpaddingé )ÚmasterÚtitleÚ config_filer Ú
|
|||
|
|
loaded_configr Ú PanedWindowÚpackÚFrameÚ
|
|||
|
|
left_frameÚaddÚ right_frameÚbuild_left_layoutÚbuild_right_layout)Úselfr Z main_pane© r ú.c:\Users\xieli\Desktop\AI_NovelGenerator\ui.pyÚ__init__ s zNovelGeneratorGUI.__init__c C sþ | j jddd� | j jddd� | j jddd� tj| j dd�}|jdddddd� |jddd� |jddd� tj|d d
|
|||
|
|
d �| _| jjdddd � tj| j d
|
|||
|
|
d�}|jdddddd� |jddd� |jddd� tj|d d
|
|||
|
|
dd�| _ | j jdddd � d S )Nr r r u 输出日志©ÚtextÚnsewé )ÚrowÚcolumnÚstickyÚpadxÚpadyéP é
|
|||
|
|
©ÚwidthÚheight©r' r( r) u æœ¬ç« å†…å®¹Zblue)r/ r0 Z
|
|||
|
|
foreground)
|
|||
|
|
r Ú rowconfigureÚcolumnconfigurer Ú
|
|||
|
|
LabelFrameÚgridr Ú ScrolledTextÚlog_textÚchapter_result)r Z log_frameZ
|
|||
|
|
chapter_framer r r! r ( s z#NovelGeneratorGUI.build_left_layoutc C s† t dƒD ]
|
|||
|
|
}| jj|dd� q| jjddd� tj| jdd�jdddddd � tj| j
|
|||
|
|
d
|
|||
|
|
d ¡d �| _ tj | j| j d
|
|||
|
|
d�jdddddd � tj| jdd�jdddddd � tj| j
|
|||
|
|
dd¡d �| _
|
|||
|
|
tj | j| j
|
|||
|
|
d
|
|||
|
|
d�jdddddd � tj| jdd�jdddddd � tj| j
|
|||
|
|
dd¡d �| _tj | j| jd
|
|||
|
|
d�jdddddd � tj| jdd�jdddddd � tj| jd
|
|||
|
|
dd�| _| jjdddddd � | j
|
|||
|
|
d d ¡}|r×| j tj|¡ tj| jd d�jdddddd � tj| j
|
|||
|
|
d d ¡d �| _tj | j| jd
|
|||
|
|
d�jdddddd � tj| jd d�jdddddd � tj| j
|
|||
|
|
d!d"¡d �| _tj | j| jd#d�jdddddd � tj| jd$d�jd%ddddd � tj| j
|
|||
|
|
d&d'¡d �| _tj | j| jd#d�jd%ddddd � tj| jd(d�jd)ddddd � tj| j
|
|||
|
|
d*d ¡d �| _tj | j| jd
|
|||
|
|
d�jd)ddddd � tj| jd+| jd,�jd)ddddd � t | j¡}|jd#ddd-� tj|d.| j d,�jdddd/� tj|d0| j d,�jdddd/� d1}tj| jd2d�j|ddd-� tjdd �| _ tj | j| j d%d�j|ddddd � tj| jd3| j d,�| _ | j j|d ddddd4d5� tj| jd6| j!d,�| _"| j"j|d ddddd4d5� tj| jd7| j#d,�| _$| j$j|d ddddd4d5� d S )8Né r r r zAPI Key:r# r& Úe)r' r( r* r+ r) Úapi_keyÚ )Úvalueé )Ú textvariabler/ Úwz Base URL:Úbase_urlzhttps://api.agicto.cn/v1z Model Name:é Ú
|
|||
|
|
model_namez gpt-4o-miniu 主题(Topic):r Úneé r. Útopicu 类型(Genre):Úgenreu 玄幻u
|
|||
|
|
ç« èŠ‚æ•°:Ú num_chaptersr- é u
|
|||
|
|
æ¯�ç« å—æ•°:é Ú word_numberé¸ u
|
|||
|
|
ä¿�å˜è·¯å¾„:é Úfilepathu æµ�览...)r$ Úcommandr1 u ä¿�å˜é…�ç½®)r' r( r* u åŠ è½½é…�ç½®é u
|
|||
|
|
ç« èŠ‚å�·:u 1. 生æˆ�设定 & 目录Úew)r' r( Z
|
|||
|
|
columnspanr* r+ r) u 2. 生æˆ�å�•ç« (å�«è§’色状æ€�)u 3. ä¸€è‡´æ€§å®¡æ ¡)%Úranger r2 r3 r ÚLabelr5 ÚtkÚ StringVarr ÚgetÚ api_key_varÚEntryÚ base_url_varÚmodel_name_varr r6 Ú
|
|||
|
|
topic_textÚinsertÚENDÚ genre_varÚIntVarÚnum_chapters_varÚword_number_varÚ filepath_varÚButtonÚ
|
|||
|
|
browse_folderr Úsave_config_btnÚload_config_btnÚchapter_num_varÚgenerate_full_novelÚbtn_generate_fullÚgenerate_chapter_textÚbtn_generate_chapterÚdo_consistency_checkÚbtn_check_consistency)r ÚiZ
|
|||
|
|
topic_defaultZ config_frameZrow_baser r r! r ? sZ $ $ $ $ $ $ $$ $ z$NovelGeneratorGUI.build_right_layoutc C sà t | jƒ}|rh| j | dd¡¡ | j | dd¡¡ | j | dd¡¡ | j | dd¡¡ | j | dd¡¡ | j | dd ¡¡ | j
|
|||
|
|
| d
|
|||
|
|
d¡¡ | j d t
|
|||
|
|
j¡ | j t
|
|||
|
|
j| d d¡¡ | d
|
|||
|
|
¡ d S t dd¡ d S )Nr; r< rA rC rG rH r- rK rL rN ú1.0rF u å·²åŠ è½½é…�置。õ æ��示u' æœªæ‰¾åˆ°æˆ–æ— æ³•è¯»å�–é…�置文件。)r r rW ÚsetrV rY rZ r^ r` ra rb r[ ÚdeleterT r] r\ Úlogr Ú showwarning)r Zcfgr r r! rf ‰ s
|
|||
|
|
z!NovelGeneratorGUI.load_config_btnc C s† | j ¡ | j ¡ | j ¡ | j dtj¡ ¡ | j ¡ | j ¡ | j
|
|||
|
|
¡ | j ¡ dœ}t || j
|
|||
|
|
ƒr;t dd¡ | d¡ d S t dd¡ d S )Nro )r; rA rC rF rG rH rK rN rp u é…�置已ä¿�å˜è‡³ config.jsonu é…�置已ä¿�å˜ã€‚u 错误u ä¿�å˜é…�置失败。)rW rV rY rZ r[ rT r] Ústripr^ r` ra rb r r r Zshowinfors Z showerror)r Z config_datar r r! re œ s ø
|
|||
|
|
z!NovelGeneratorGUI.save_config_btnc C s t ¡ }|r| j |¡ d S d S )N)r Z askdirectoryrb rq )r Z selected_dirr r r! rd s ÿz NovelGeneratorGUI.browse_folderÚmessagec C s& | j tj|d ¡ | j tj¡ d S )NÚ
|
|||
|
|
)r7 r\ rT r] Úsee)r rv r r r! rs ³ s zNovelGeneratorGUI.logc C ó |j tjd� d S ©N)Ústate)ÚconfigrT ÚDISABLED©r Zbtnr r r! Údisable_button¸ ó z NovelGeneratorGUI.disable_buttonc C ry rz )r| rT ÚNORMALr~ r r r! Ú
|
|||
|
|
enable_button» r€ z NovelGeneratorGUI.enable_buttonc ó$ ‡ fdd„}t j|d�}| ¡ dS )u 生æˆ�å°�说设定 & 目录c
|
|||
|
|
s8 ˆ ˆ j¡ zŽzaˆ j ¡ ¡ } ˆ j ¡ ¡ }ˆ j ¡ ¡ }ˆ j dtj ¡ ¡ }ˆ j
|
|||
|
|
¡ ¡ }ˆ j ¡ }ˆ j ¡ }ˆ j
|
|||
|
|
¡ ¡ }|sQt dd¡ W W ˆ ˆ j¡ d S ˆ d¡ t| |||||||d� ˆ d¡ W n ty‚ } zˆ d|› �¡ W Y d }~nd }~ww W ˆ ˆ j¡ d S W ˆ ˆ j¡ d S ˆ ˆ j¡ w )Nro u è¦å‘Šu 请先选择ä¿�å˜æ–‡ä»¶è·¯å¾„u$ 开始生æˆ�å°�说设定和目录...)r; rA Z llm_modelrF rG Znumber_of_chaptersrK rN u[ ✅ å°�说设定和目录生æˆ�完æˆ�。查看 Novel_setting.txt å’Œ Novel_directory.txt。u* â�Œ 生æˆ�å°�说设定 & 目录时出错: )r ri rW rV ru rY rZ r[ rT r] r^ r` ra rb r rt r‚ rs r Ú Exception) r; rA rC rF rG rH rK rN r: ©r r r! ÚtaskÀ sD
|
|||
|
|
|
|||
|
|
|
|||
|
|
ñø
|
|||
|
|
€ÿÿþ z3NovelGeneratorGUI.generate_full_novel.<locals>.task©ÚtargetN©Ú threadingZThreadÚstart©r r† Úthreadr r… r! rh ¾ s ! z%NovelGeneratorGUI.generate_full_novelc rƒ )uO 多æ¥ç”Ÿæˆ�ç« èŠ‚ï¼šç»´æŠ¤å…¨å±€æ‘˜è¦�+角色状æ€�文档，å�‘é‡�检索辅助c sÐ ˆ ˆ j¡ zÚzˆ j ¡ ¡ } ˆ j ¡ ¡ }ˆ j ¡ ¡ }ˆ j ¡ }ˆ j ¡ ¡ }ˆ j ¡ }t
|