gardel/Mofox-Core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修复了人格设置保存不了的bug(感谢大佬ZureTz提供的修复！)，修复其他项配置保存不了的bug，修复部分文字表述错误

Browse Source
This commit is contained in:
DrSmoothl
2025-03-18 20:37:34 +08:00
parent c1dbdcff18
commit cabd632bfd
1 changed files with 133 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

189
webui.py
View File

@@ -2,12 +2,13 @@ import gradio as gr
import os import os
import sys import sys
import toml import toml
from loguru import logger from src.common.logger import get_module_logger
import shutil import shutil
import ast import ast
import json import json
from packaging import version from packaging import version
logger = get_module_logger("webui")
is_share = False is_share = False
debug = True debug = True
@@ -17,8 +18,8 @@ CONFIG_VERSION = config_data["inner"]["version"]
PARSED_CONFIG_VERSION = version.parse(CONFIG_VERSION) PARSED_CONFIG_VERSION = version.parse(CONFIG_VERSION)
HAVE_ONLINE_STATUS_VERSION = version.parse("0.0.9") HAVE_ONLINE_STATUS_VERSION = version.parse("0.0.9")
#============================================== # ==============================================
#env环境配置文件读取部分 # env环境配置文件读取部分
def parse_env_config(config_file): def parse_env_config(config_file):
""" """
解析配置文件并将配置项存储到相应的变量中（变量名以env_为前缀）。 解析配置文件并将配置项存储到相应的变量中（变量名以env_为前缀）。
@@ -52,7 +53,7 @@ def parse_env_config(config_file):
return env_variables return env_variables
#env环境配置文件保存函数 # env环境配置文件保存函数
def save_to_env_file(env_variables, filename=".env.prod"): def save_to_env_file(env_variables, filename=".env.prod"):
""" """
将修改后的变量保存到指定的.env文件中，并在第一次保存前备份文件（如果备份文件不存在）。 将修改后的变量保存到指定的.env文件中，并在第一次保存前备份文件（如果备份文件不存在）。
@@ -75,7 +76,7 @@ def save_to_env_file(env_variables, filename=".env.prod"):
logger.info(f"配置已保存到 {filename}") logger.info(f"配置已保存到 {filename}")
#载入env文件并解析 # 载入env文件并解析
env_config_file = ".env.prod" # 配置文件路径 env_config_file = ".env.prod" # 配置文件路径
env_config_data = parse_env_config(env_config_file) env_config_data = parse_env_config(env_config_file)
if "env_VOLCENGINE_BASE_URL" in env_config_data: if "env_VOLCENGINE_BASE_URL" in env_config_data:
@@ -97,8 +98,8 @@ MODEL_PROVIDER_LIST = [
"SILICONFLOW", "SILICONFLOW",
"DEEP_SEEK" "DEEP_SEEK"
] ]
#env读取保存结束 # env读取保存结束
#============================================== # ==============================================
#获取在线麦麦数量 #获取在线麦麦数量
import requests import requests
@@ -188,7 +189,7 @@ def delete_int_item(selected_item, current_list):
gr.update(choices=updated_list), gr.update(choices=updated_list),
", ".join(map(str, updated_list)) ", ".join(map(str, updated_list))
] ]
#env文件中插件值处理函数 # env文件中插件值处理函数
def parse_list_str(input_str): def parse_list_str(input_str):
""" """
将形如["src2.plugins.chat"]的字符串解析为Python列表 将形如["src2.plugins.chat"]的字符串解析为Python列表
@@ -222,7 +223,7 @@ def format_list_to_str(lst):
return "[" + res + "]" return "[" + res + "]"
#env保存函数 # env保存函数
def save_trigger(server_address, server_port, final_result_list,t_mongodb_host,t_mongodb_port,t_mongodb_database_name,t_chatanywhere_base_url,t_chatanywhere_key,t_siliconflow_base_url,t_siliconflow_key,t_deepseek_base_url,t_deepseek_key,t_volcengine_base_url,t_volcengine_key): def save_trigger(server_address, server_port, final_result_list,t_mongodb_host,t_mongodb_port,t_mongodb_database_name,t_chatanywhere_base_url,t_chatanywhere_key,t_siliconflow_base_url,t_siliconflow_key,t_deepseek_base_url,t_deepseek_key,t_volcengine_base_url,t_volcengine_key):
final_result_lists = format_list_to_str(final_result_list) final_result_lists = format_list_to_str(final_result_list)
env_config_data["env_HOST"] = server_address env_config_data["env_HOST"] = server_address
@@ -243,11 +244,11 @@ def save_trigger(server_address, server_port, final_result_list,t_mongodb_host,t
logger.success("配置已保存到 .env.prod 文件中") logger.success("配置已保存到 .env.prod 文件中")
return "配置已保存" return "配置已保存"
#============================================== # ==============================================
#============================================== # ==============================================
#主要配置文件保存函数 # 主要配置文件保存函数
def save_config_to_file(t_config_data): def save_config_to_file(t_config_data):
filename = "config/bot_config.toml" filename = "config/bot_config.toml"
backup_filename = f"{filename}.bak" backup_filename = f"{filename}.bak"
@@ -272,49 +273,62 @@ def save_bot_config(t_qqbot_qq, t_nickname,t_nickname_final_result):
return "Bot配置已保存" return "Bot配置已保存"
# 监听滑块的值变化，确保总和不超过 1，并显示警告 # 监听滑块的值变化，确保总和不超过 1，并显示警告
def adjust_greater_probabilities(t_personality_1, t_personality_2, t_personality_3): def adjust_personality_greater_probabilities(t_personality_1_probability, t_personality_2_probability, t_personality_3_probability):
total = t_personality_1 + t_personality_2 + t_personality_3 total = t_personality_1_probability + t_personality_2_probability + t_personality_3_probability
if total > 1.0: if total > 1.0:
warning_message = f"警告: 人格1、人格2和人格3的概率总和为 {total:.2f}，超过了 1.0！请调整滑块使总和等于 1.0。" warning_message = f"警告: 人格1、人格2和人格3的概率总和为 {total:.2f}，超过了 1.0！请调整滑块使总和等于 1.0。"
return warning_message return warning_message
else: return "" # 没有警告时返回空字符串
return "" # 没有警告时返回空字符串
def adjust_less_probabilities(t_personality_1, t_personality_2, t_personality_3): def adjust_personality_less_probabilities(t_personality_1_probability, t_personality_2_probability, t_personality_3_probability):
total = t_personality_1 + t_personality_2 + t_personality_3 total = t_personality_1_probability + t_personality_2_probability + t_personality_3_probability
if total < 1.0: if total < 1.0:
warning_message = f"警告: 人格1、人格2和人格3的概率总和为 {total:.2f}，小于 1.0！请调整滑块使总和等于 1.0。" warning_message = f"警告: 人格1、人格2和人格3的概率总和为 {total:.2f}，小于 1.0！请调整滑块使总和等于 1.0。"
return warning_message return warning_message
else: return "" # 没有警告时返回空字符串
return "" # 没有警告时返回空字符串
def adjust_model_greater_probabilities(t_personality_1, t_personality_2, t_personality_3): def adjust_model_greater_probabilities(t_model_1_probability, t_model_2_probability, t_model_3_probability):
total = t_personality_1 + t_personality_2 + t_personality_3 total = t_model_1_probability + t_model_2_probability + t_model_3_probability
if total > 1.0: if total > 1.0:
warning_message = f"警告: 选择模型1、模型2和模型3的概率总和为 {total:.2f}，超过了 1.0！请调整滑块使总和等于 1.0。" warning_message = f"警告: 选择模型1、模型2和模型3的概率总和为 {total:.2f}，超过了 1.0！请调整滑块使总和等于 1.0。"
return warning_message return warning_message
else: return "" # 没有警告时返回空字符串
return "" # 没有警告时返回空字符串
def adjust_model_less_probabilities(t_personality_1, t_personality_2, t_personality_3): def adjust_model_less_probabilities(t_model_1_probability, t_model_2_probability, t_model_3_probability):
total = t_personality_1 + t_personality_2 + t_personality_3 total = t_model_1_probability + t_model_2_probability + t_model_3_probability
if total > 1.0: if total > 1.0:
warning_message = f"警告: 选择模型1、模型2和模型3的概率总和为 {total:.2f}，小于了 1.0！请调整滑块使总和等于 1.0。" warning_message = f"警告: 选择模型1、模型2和模型3的概率总和为 {total:.2f}，小于了 1.0！请调整滑块使总和等于 1.0。"
return warning_message return warning_message
else: return "" # 没有警告时返回空字符串
return "" # 没有警告时返回空字符串
#==============================================
#人格保存函数 # ==============================================
def save_personality_config(t_personality_1, t_personality_2, t_personality_3, t_prompt_schedule): # 人格保存函数
config_data["personality"]["personality_1_probability"] = t_personality_1 def save_personality_config(t_prompt_personality_1,
config_data["personality"]["personality_2_probability"] = t_personality_2 t_prompt_personality_2,
config_data["personality"]["personality_3_probability"] = t_personality_3 t_prompt_personality_3,
t_prompt_schedule,
t_personality_1_probability,
t_personality_2_probability,
t_personality_3_probability):
# 保存人格提示词
config_data["personality"]["prompt_personality"][0] = t_prompt_personality_1
config_data["personality"]["prompt_personality"][1] = t_prompt_personality_2
config_data["personality"]["prompt_personality"][2] = t_prompt_personality_3
# 保存日程生成提示词
config_data["personality"]["prompt_schedule"] = t_prompt_schedule config_data["personality"]["prompt_schedule"] = t_prompt_schedule
# 保存三个人格的概率
config_data["personality"]["personality_1_probability"] = t_personality_1_probability
config_data["personality"]["personality_2_probability"] = t_personality_2_probability
config_data["personality"]["personality_3_probability"] = t_personality_3_probability
save_config_to_file(config_data) save_config_to_file(config_data)
logger.info("人格配置已保存到 bot_config.toml 文件中") logger.info("人格配置已保存到 bot_config.toml 文件中")
return "人格配置已保存" return "人格配置已保存"
def save_message_and_emoji_config(t_min_text_length, def save_message_and_emoji_config(t_min_text_length,
t_max_context_size, t_max_context_size,
t_emoji_chance, t_emoji_chance,
@@ -415,7 +429,7 @@ def save_other_config(t_keywords_reaction_enabled,t_enable_advance_output, t_ena
config_data["chinese_typo"]["min_freq"] = t_min_freq config_data["chinese_typo"]["min_freq"] = t_min_freq
config_data["chinese_typo"]["tone_error_rate"] = t_tone_error_rate config_data["chinese_typo"]["tone_error_rate"] = t_tone_error_rate
config_data["chinese_typo"]["word_replace_rate"] = t_word_replace_rate config_data["chinese_typo"]["word_replace_rate"] = t_word_replace_rate
if PARSED_CONFIG_VERSION > 0.8: if PARSED_CONFIG_VERSION > HAVE_ONLINE_STATUS_VERSION:
config_data["remote"]["enable"] = t_remote_status config_data["remote"]["enable"] = t_remote_status
save_config_to_file(config_data) save_config_to_file(config_data)
logger.info("其他设置已保存到 bot_config.toml 文件中") logger.info("其他设置已保存到 bot_config.toml 文件中")
@@ -435,6 +449,7 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
gr.Markdown( gr.Markdown(
value=""" value="""
### 欢迎使用由墨梓柒MotricSeven编写的MaimBot配置文件编辑器\n ### 欢迎使用由墨梓柒MotricSeven编写的MaimBot配置文件编辑器\n
感谢ZureTz大佬提供的人格保存部分修复！
""" """
) )
gr.Markdown( gr.Markdown(
@@ -531,19 +546,19 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
) )
with gr.Row(): with gr.Row():
gr.Markdown( gr.Markdown(
'''ChatAntWhere的baseURL和APIkey\n '''ChatAnyWhere的baseURL和APIkey\n
改完了记得保存！！！ 改完了记得保存！！！
''' '''
) )
with gr.Row(): with gr.Row():
chatanywhere_base_url = gr.Textbox( chatanywhere_base_url = gr.Textbox(
label="ChatAntWhere的BaseURL", label="ChatAnyWhere的BaseURL",
value=env_config_data["env_CHAT_ANY_WHERE_BASE_URL"], value=env_config_data["env_CHAT_ANY_WHERE_BASE_URL"],
interactive=True interactive=True
) )
with gr.Row(): with gr.Row():
chatanywhere_key = gr.Textbox( chatanywhere_key = gr.Textbox(
label="ChatAntWhere的key", label="ChatAnyWhere的key",
value=env_config_data["env_CHAT_ANY_WHERE_KEY"], value=env_config_data["env_CHAT_ANY_WHERE_KEY"],
interactive=True interactive=True
) )
@@ -676,38 +691,92 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
with gr.Row(): with gr.Row():
prompt_personality_1 = gr.Textbox( prompt_personality_1 = gr.Textbox(
label="人格1提示词", label="人格1提示词",
value=config_data['personality']['prompt_personality'][0], value=config_data["personality"]["prompt_personality"][0],
interactive=True interactive=True,
) )
with gr.Row(): with gr.Row():
prompt_personality_2 = gr.Textbox( prompt_personality_2 = gr.Textbox(
label="人格2提示词", label="人格2提示词",
value=config_data['personality']['prompt_personality'][1], value=config_data["personality"]["prompt_personality"][1],
interactive=True interactive=True,
) )
with gr.Row(): with gr.Row():
prompt_personality_3 = gr.Textbox( prompt_personality_3 = gr.Textbox(
label="人格3提示词", label="人格3提示词",
value=config_data['personality']['prompt_personality'][2], value=config_data["personality"]["prompt_personality"][2],
interactive=True interactive=True,
) )
with gr.Column(scale=3): with gr.Column(scale=3):
# 创建三个滑块 # 创建三个滑块, 代表三个人格的概率
personality_1 = gr.Slider(minimum=0, maximum=1, step=0.01, value=config_data["personality"]["personality_1_probability"], label="人格1概率") personality_1_probability = gr.Slider(
personality_2 = gr.Slider(minimum=0, maximum=1, step=0.01, value=config_data["personality"]["personality_2_probability"], label="人格2概率") minimum=0,
personality_3 = gr.Slider(minimum=0, maximum=1, step=0.01, value=config_data["personality"]["personality_3_probability"], label="人格3概率") maximum=1,
step=0.01,
value=config_data["personality"]["personality_1_probability"],
label="人格1概率",
)
personality_2_probability = gr.Slider(
minimum=0,
maximum=1,
step=0.01,
value=config_data["personality"]["personality_2_probability"],
label="人格2概率",
)
personality_3_probability = gr.Slider(
minimum=0,
maximum=1,
step=0.01,
value=config_data["personality"]["personality_3_probability"],
label="人格3概率",
)
# 用于显示警告消息 # 用于显示警告消息
warning_greater_text = gr.Markdown() warning_greater_text = gr.Markdown()
warning_less_text = gr.Markdown() warning_less_text = gr.Markdown()
# 绑定滑块的值变化事件，确保总和必须等于 1.0 # 绑定滑块的值变化事件，确保总和必须等于 1.0
personality_1.change(adjust_greater_probabilities, inputs=[personality_1, personality_2, personality_3], outputs=[warning_greater_text])
personality_2.change(adjust_greater_probabilities, inputs=[personality_1, personality_2, personality_3], outputs=[warning_greater_text]) # 输入的 3 个概率
personality_3.change(adjust_greater_probabilities, inputs=[personality_1, personality_2, personality_3], outputs=[warning_greater_text]) personality_probability_change_inputs = [
personality_1.change(adjust_less_probabilities, inputs=[personality_1, personality_2, personality_3], outputs=[warning_less_text]) personality_1_probability,
personality_2.change(adjust_less_probabilities, inputs=[personality_1, personality_2, personality_3], outputs=[warning_less_text]) personality_2_probability,
personality_3.change(adjust_less_probabilities, inputs=[personality_1, personality_2, personality_3], outputs=[warning_less_text]) personality_3_probability,
]
# 绑定滑块的值变化事件，确保总和不大于 1.0
personality_1_probability.change(
adjust_personality_greater_probabilities,
inputs=personality_probability_change_inputs,
outputs=[warning_greater_text],
)
personality_2_probability.change(
adjust_personality_greater_probabilities,
inputs=personality_probability_change_inputs,
outputs=[warning_greater_text],
)
personality_3_probability.change(
adjust_personality_greater_probabilities,
inputs=personality_probability_change_inputs,
outputs=[warning_greater_text],
)
# 绑定滑块的值变化事件，确保总和不小于 1.0
personality_1_probability.change(
adjust_personality_less_probabilities,
inputs=personality_probability_change_inputs,
outputs=[warning_less_text],
)
personality_2_probability.change(
adjust_personality_less_probabilities,
inputs=personality_probability_change_inputs,
outputs=[warning_less_text],
)
personality_3_probability.change(
adjust_personality_less_probabilities,
inputs=personality_probability_change_inputs,
outputs=[warning_less_text],
)
with gr.Row(): with gr.Row():
prompt_schedule = gr.Textbox( prompt_schedule = gr.Textbox(
label="日程生成提示词", label="日程生成提示词",
@@ -725,8 +794,16 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
personal_save_message = gr.Textbox(label="保存人格结果") personal_save_message = gr.Textbox(label="保存人格结果")
personal_save_btn.click( personal_save_btn.click(
save_personality_config, save_personality_config,
inputs=[personality_1, personality_2, personality_3, prompt_schedule], inputs=[
outputs=[personal_save_message] prompt_personality_1,
prompt_personality_2,
prompt_personality_3,
prompt_schedule,
personality_1_probability,
personality_2_probability,
personality_3_probability,
],
outputs=[personal_save_message],
) )
with gr.TabItem("3-消息&表情包设置"): with gr.TabItem("3-消息&表情包设置"):
with gr.Row(): with gr.Row():