Merge pull request #489 from MaiM-with-u/main-fix

Main fix
This commit is contained in:
SengokuCola
2025-03-19 14:04:11 +08:00
committed by GitHub
6 changed files with 430 additions and 129 deletions

View File

@@ -430,7 +430,7 @@ if not exist config/bot_config.toml (
)
if not exist .env.prod (
copy /Y "template\.env.prod" ".env.prod"
copy /Y "template.env" ".env.prod"
)
start python webui.py

View File

@@ -161,8 +161,8 @@ switch_branch() {
sed -i "s/^BRANCH=.*/BRANCH=${new_branch}/" /etc/maimbot_install.conf
BRANCH="${new_branch}"
check_eula
systemctl restart ${SERVICE_NAME}
touch "${INSTALL_DIR}/repo/elua.confirmed"
whiptail --msgbox "✅ 已切换到分支 ${new_branch} 并重启服务!" 10 60
}
@@ -186,6 +186,42 @@ update_config() {
fi
}
check_eula() {
# 首先计算当前EULA的MD5值
current_md5=$(md5sum "${INSTALL_DIR}/repo/EULA.md" | awk '{print $1}')
# 首先计算当前隐私条款文件的哈希值
current_md5_privacy=$(md5sum "${INSTALL_DIR}/repo/PRIVACY.md" | awk '{print $1}')
# 检查eula.confirmed文件是否存在
if [[ -f ${INSTALL_DIR}/repo/eula.confirmed ]]; then
# 如果存在则检查其中包含的md5与current_md5是否一致
confirmed_md5=$(cat ${INSTALL_DIR}/repo/eula.confirmed)
else
confirmed_md5=""
fi
# 检查privacy.confirmed文件是否存在
if [[ -f ${INSTALL_DIR}/repo/privacy.confirmed ]]; then
# 如果存在则检查其中包含的md5与current_md5是否一致
confirmed_md5_privacy=$(cat ${INSTALL_DIR}/repo/privacy.confirmed)
else
confirmed_md5_privacy=""
fi
# 如果EULA或隐私条款有更新提示用户重新确认
if [[ $current_md5 != $confirmed_md5 || $current_md5_privacy != $confirmed_md5_privacy ]]; then
whiptail --title "📜 使用协议更新" --yesno "检测到麦麦Bot EULA或隐私条款已更新。\nhttps://github.com/SengokuCola/MaiMBot/blob/main/EULA.md\nhttps://github.com/SengokuCola/MaiMBot/blob/main/PRIVACY.md\n\n您是否同意上述协议 \n\n " 12 70
if [[ $? -eq 0 ]]; then
echo $current_md5 > ${INSTALL_DIR}/repo/eula.confirmed
echo $current_md5_privacy > ${INSTALL_DIR}/repo/privacy.confirmed
else
exit 1
fi
fi
}
# ----------- 主安装流程 -----------
run_installation() {
# 1/6: 检测是否安装 whiptail
@@ -195,7 +231,7 @@ run_installation() {
fi
# 协议确认
if ! (whiptail --title " [1/6] 使用协议" --yes-button "我同意" --no-button "我拒绝" --yesno "使用麦麦Bot及此脚本前请先阅读ELUA协议\nhttps://github.com/SengokuCola/MaiMBot/blob/main/EULA.md\n\n您是否同意协议?" 12 70); then
if ! (whiptail --title " [1/6] 使用协议" --yes-button "我同意" --no-button "我拒绝" --yesno "使用麦麦Bot及此脚本前请先阅读EULA协议及隐私协议\nhttps://github.com/SengokuCola/MaiMBot/blob/main/EULA.md\nhttps://github.com/SengokuCola/MaiMBot/blob/main/PRIVACY.md\n\n您是否同意上述协议?" 12 70); then
exit 1
fi
@@ -355,7 +391,15 @@ run_installation() {
pip install -r repo/requirements.txt
echo -e "${GREEN}同意协议...${RESET}"
touch repo/elua.confirmed
# 首先计算当前EULA的MD5值
current_md5=$(md5sum "repo/EULA.md" | awk '{print $1}')
# 首先计算当前隐私条款文件的哈希值
current_md5_privacy=$(md5sum "repo/PRIVACY.md" | awk '{print $1}')
echo $current_md5 > repo/eula.confirmed
echo $current_md5_privacy > repo/privacy.confirmed
echo -e "${GREEN}创建系统服务...${RESET}"
cat > /etc/systemd/system/${SERVICE_NAME}.service <<EOF
@@ -408,9 +452,10 @@ EOF
exit 1
}
# 如果已安装显示菜单
# 如果已安装显示菜单,并检查协议是否更新
if check_installed; then
load_install_info
check_eula
show_menu
else
run_installation

View File

@@ -1,6 +1,7 @@
import math
import random
import time
import re
from collections import Counter
from typing import Dict, List
@@ -253,7 +254,7 @@ def split_into_sentences_w_remove_punctuation(text: str) -> List[str]:
# 统一将英文逗号转换为中文逗号
text = text.replace(',', '')
text = text.replace('\n', ' ')
text, mapping = protect_kaomoji(text)
# print(f"处理前的文本: {text}")
text_no_1 = ''
@@ -292,6 +293,7 @@ def split_into_sentences_w_remove_punctuation(text: str) -> List[str]:
current_sentence += ' ' + part
new_sentences.append(current_sentence.strip())
sentences = [s for s in new_sentences if s] # 移除空字符串
sentences = recover_kaomoji(sentences, mapping)
# print(f"分割后的句子: {sentences}")
sentences_done = []
@@ -446,3 +448,55 @@ def truncate_message(message: str, max_length=20) -> str:
if len(message) > max_length:
return message[:max_length] + "..."
return message
def protect_kaomoji(sentence):
""""
识别并保护句子中的颜文字(含括号与无括号),将其替换为占位符,
并返回替换后的句子和占位符到颜文字的映射表。
Args:
sentence (str): 输入的原始句子
Returns:
tuple: (处理后的句子, {占位符: 颜文字})
"""
kaomoji_pattern = re.compile(
r'('
r'[\(\[(【]' # 左括号
r'[^()\[\]()【】]*?' # 非括号字符(惰性匹配)
r'[^\u4e00-\u9fa5a-zA-Z0-9\s]' # 非中文、非英文、非数字、非空格字符(必须包含至少一个)
r'[^()\[\]()【】]*?' # 非括号字符(惰性匹配)
r'[\)\])】]' # 右括号
r')'
r'|'
r'('
r'[▼▽・ᴥω・﹏^><≧≦ ̄`´∀ヮДд︿﹀へ。゚╥╯╰︶︹•⁄]{2,15}'
r')'
)
kaomoji_matches = kaomoji_pattern.findall(sentence)
placeholder_to_kaomoji = {}
for idx, match in enumerate(kaomoji_matches):
kaomoji = match[0] if match[0] else match[1]
placeholder = f'__KAOMOJI_{idx}__'
sentence = sentence.replace(kaomoji, placeholder, 1)
placeholder_to_kaomoji[placeholder] = kaomoji
return sentence, placeholder_to_kaomoji
def recover_kaomoji(sentences, placeholder_to_kaomoji):
"""
根据映射表恢复句子中的颜文字。
Args:
sentences (list): 含有占位符的句子列表
placeholder_to_kaomoji (dict): 占位符到颜文字的映射表
Returns:
list: 恢复颜文字后的句子列表
"""
recovered_sentences = []
for sentence in sentences:
for placeholder, kaomoji in placeholder_to_kaomoji.items():
sentence = sentence.replace(placeholder, kaomoji)
recovered_sentences.append(sentence)
return recovered_sentences

View File

@@ -61,7 +61,7 @@ class WillingManager:
reply_probability = 0
if chat_stream.group_info.group_id in config.talk_frequency_down_groups:
reply_probability = reply_probability / 3.5
reply_probability = reply_probability / config.down_frequency_rate
return reply_probability

View File

@@ -62,7 +62,7 @@ class WillingManager:
reply_probability = 0
if chat_stream.group_info.group_id in config.talk_frequency_down_groups:
reply_probability = reply_probability / 3.5
reply_probability = reply_probability / config.down_frequency_rate
if is_mentioned_bot and sender_id == "1026294844":
reply_probability = 1

444
webui.py
View File

@@ -1,25 +1,37 @@
import gradio as gr
import os
import sys
import toml
from src.common.logger import get_module_logger
import shutil
import ast
import json
from packaging import version
from decimal import Decimal, ROUND_DOWN
logger = get_module_logger("webui")
is_share = False
debug = True
# 检查配置文件是否存在
if not os.path.exists("config/bot_config.toml"):
logger.error("配置文件 bot_config.toml 不存在,请检查配置文件路径")
raise FileNotFoundError("配置文件 bot_config.toml 不存在,请检查配置文件路径")
if not os.path.exists(".env.prod"):
logger.error("环境配置文件 .env.prod 不存在,请检查配置文件路径")
raise FileNotFoundError("环境配置文件 .env.prod 不存在,请检查配置文件路径")
config_data = toml.load("config/bot_config.toml")
CONFIG_VERSION = config_data["inner"]["version"]
PARSED_CONFIG_VERSION = version.parse(CONFIG_VERSION)
HAVE_ONLINE_STATUS_VERSION = version.parse("0.0.9")
#==============================================
#env环境配置文件读取部分
#添加WebUI配置文件版本
WEBUI_VERSION = version.parse("0.0.8")
# ==============================================
# env环境配置文件读取部分
def parse_env_config(config_file):
"""
解析配置文件并将配置项存储到相应的变量中变量名以env_为前缀
@@ -53,7 +65,7 @@ def parse_env_config(config_file):
return env_variables
#env环境配置文件保存函数
# env环境配置文件保存函数
def save_to_env_file(env_variables, filename=".env.prod"):
"""
将修改后的变量保存到指定的.env文件中并在第一次保存前备份文件如果备份文件不存在
@@ -76,7 +88,7 @@ def save_to_env_file(env_variables, filename=".env.prod"):
logger.info(f"配置已保存到 {filename}")
#载入env文件并解析
# 载入env文件并解析
env_config_file = ".env.prod" # 配置文件路径
env_config_data = parse_env_config(env_config_file)
if "env_VOLCENGINE_BASE_URL" in env_config_data:
@@ -92,14 +104,90 @@ else:
logger.info("VOLCENGINE_KEY 不存在,已创建并使用默认值")
env_config_data["env_VOLCENGINE_KEY"] = "volc_key"
save_to_env_file(env_config_data, env_config_file)
MODEL_PROVIDER_LIST = [
"VOLCENGINE",
"CHAT_ANY_WHERE",
"SILICONFLOW",
"DEEP_SEEK"
]
#env读取保存结束
#==============================================
def parse_model_providers(env_vars):
"""
从环境变量中解析模型提供商列表
参数:
env_vars: 包含环境变量的字典
返回:
list: 模型提供商列表
"""
providers = []
for key in env_vars.keys():
if key.startswith("env_") and key.endswith("_BASE_URL"):
# 提取中间部分作为提供商名称
provider = key[4:-9] # 移除"env_"前缀和"_BASE_URL"后缀
providers.append(provider)
return providers
def add_new_provider(provider_name, current_providers):
"""
添加新的提供商到列表中
参数:
provider_name: 新的提供商名称
current_providers: 当前的提供商列表
返回:
tuple: (更新后的提供商列表, 更新后的下拉列表选项)
"""
if not provider_name or provider_name in current_providers:
return current_providers, gr.update(choices=current_providers)
# 添加新的提供商到环境变量中
env_config_data[f"env_{provider_name}_BASE_URL"] = ""
env_config_data[f"env_{provider_name}_KEY"] = ""
# 更新提供商列表
updated_providers = current_providers + [provider_name]
# 保存到环境文件
save_to_env_file(env_config_data)
return updated_providers, gr.update(choices=updated_providers)
# 从环境变量中解析并更新提供商列表
MODEL_PROVIDER_LIST = parse_model_providers(env_config_data)
# env读取保存结束
# ==============================================
#获取在线麦麦数量
import requests
def get_online_maimbot(url="http://hyybuth.xyz:10058/api/clients/details", timeout=10):
"""
获取在线客户端详细信息。
参数:
url (str): API 请求地址,默认值为 "http://hyybuth.xyz:10058/api/clients/details"
timeout (int): 请求超时时间,默认值为 10 秒。
返回:
dict: 解析后的 JSON 数据。
异常:
如果请求失败或数据格式不正确,将返回 None 并记录错误信息。
"""
try:
response = requests.get(url, timeout=timeout)
# 检查 HTTP 响应状态码是否为 200
if response.status_code == 200:
# 尝试解析 JSON 数据
return response.json()
else:
logger.error(f"请求失败,状态码: {response.status_code}")
return None
except requests.exceptions.Timeout:
logger.error("请求超时,请检查网络连接或增加超时时间。")
return None
except requests.exceptions.ConnectionError:
logger.error("连接错误请检查网络或API地址是否正确。")
return None
except ValueError: # 包括 json.JSONDecodeError
logger.error("无法解析返回的JSON数据请检查API返回内容。")
return None
online_maimbot_data = get_online_maimbot()
#==============================================
#env环境文件中插件修改更新函数
@@ -151,7 +239,7 @@ def delete_int_item(selected_item, current_list):
gr.update(choices=updated_list),
", ".join(map(str, updated_list))
]
#env文件中插件值处理函数
# env文件中插件值处理函数
def parse_list_str(input_str):
"""
将形如["src2.plugins.chat"]的字符串解析为Python列表
@@ -185,8 +273,8 @@ def format_list_to_str(lst):
return "[" + res + "]"
#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):
# env保存函数
def save_trigger(server_address, server_port, final_result_list, t_mongodb_host, t_mongodb_port, t_mongodb_database_name, t_console_log_level, t_file_log_level, t_default_console_log_level, t_default_file_log_level, t_api_provider, t_api_base_url, t_api_key):
final_result_lists = format_list_to_str(final_result_list)
env_config_data["env_HOST"] = server_address
env_config_data["env_PORT"] = server_port
@@ -194,23 +282,37 @@ def save_trigger(server_address, server_port, final_result_list,t_mongodb_host,t
env_config_data["env_MONGODB_HOST"] = t_mongodb_host
env_config_data["env_MONGODB_PORT"] = t_mongodb_port
env_config_data["env_DATABASE_NAME"] = t_mongodb_database_name
env_config_data["env_CHAT_ANY_WHERE_BASE_URL"] = t_chatanywhere_base_url
env_config_data["env_CHAT_ANY_WHERE_KEY"] = t_chatanywhere_key
env_config_data["env_SILICONFLOW_BASE_URL"] = t_siliconflow_base_url
env_config_data["env_SILICONFLOW_KEY"] = t_siliconflow_key
env_config_data["env_DEEP_SEEK_BASE_URL"] = t_deepseek_base_url
env_config_data["env_DEEP_SEEK_KEY"] = t_deepseek_key
env_config_data["env_VOLCENGINE_BASE_URL"] = t_volcengine_base_url
env_config_data["env_VOLCENGINE_KEY"] = t_volcengine_key
# 保存日志配置
env_config_data["env_CONSOLE_LOG_LEVEL"] = t_console_log_level
env_config_data["env_FILE_LOG_LEVEL"] = t_file_log_level
env_config_data["env_DEFAULT_CONSOLE_LOG_LEVEL"] = t_default_console_log_level
env_config_data["env_DEFAULT_FILE_LOG_LEVEL"] = t_default_file_log_level
# 保存选中的API提供商的配置
env_config_data[f"env_{t_api_provider}_BASE_URL"] = t_api_base_url
env_config_data[f"env_{t_api_provider}_KEY"] = t_api_key
save_to_env_file(env_config_data)
logger.success("配置已保存到 .env.prod 文件中")
return "配置已保存"
#==============================================
def update_api_inputs(provider):
"""
根据选择的提供商更新Base URL和API Key输入框的值
"""
base_url = env_config_data.get(f"env_{provider}_BASE_URL", "")
api_key = env_config_data.get(f"env_{provider}_KEY", "")
return base_url, api_key
# 绑定下拉列表的change事件
#==============================================
#主要配置文件保存函数
# ==============================================
# ==============================================
# 主要配置文件保存函数
def save_config_to_file(t_config_data):
filename = "config/bot_config.toml"
backup_filename = f"{filename}.bak"
@@ -235,49 +337,62 @@ def save_bot_config(t_qqbot_qq, t_nickname,t_nickname_final_result):
return "Bot配置已保存"
# 监听滑块的值变化,确保总和不超过 1并显示警告
def adjust_greater_probabilities(t_personality_1, t_personality_2, t_personality_3):
total = t_personality_1 + t_personality_2 + t_personality_3
if total > 1.0:
warning_message = f"警告: 人格1、人格2和人格3的概率总和为 {total:.2f},超过了 1.0!请调整滑块使总和等于 1.0。"
def adjust_personality_greater_probabilities(t_personality_1_probability, t_personality_2_probability, t_personality_3_probability):
total = Decimal(str(t_personality_1_probability)) + Decimal(str(t_personality_2_probability)) + Decimal(str(t_personality_3_probability))
if total > Decimal('1.0'):
warning_message = f"警告: 人格1、人格2和人格3的概率总和为 {float(total):.2f},超过了 1.0!请调整滑块使总和等于 1.0。"
return warning_message
else:
return "" # 没有警告时返回空字符串
return "" # 没有警告时返回空字符串
def adjust_less_probabilities(t_personality_1, t_personality_2, t_personality_3):
total = t_personality_1 + t_personality_2 + t_personality_3
if total < 1.0:
warning_message = f"警告: 人格1、人格2和人格3的概率总和为 {total:.2f},小于 1.0!请调整滑块使总和等于 1.0。"
def adjust_personality_less_probabilities(t_personality_1_probability, t_personality_2_probability, t_personality_3_probability):
total = Decimal(str(t_personality_1_probability)) + Decimal(str(t_personality_2_probability)) + Decimal(str(t_personality_3_probability))
if total < Decimal('1.0'):
warning_message = f"警告: 人格1、人格2和人格3的概率总和为 {float(total):.2f},小于 1.0!请调整滑块使总和等于 1.0。"
return warning_message
else:
return "" # 没有警告时返回空字符串
return "" # 没有警告时返回空字符串
def adjust_model_greater_probabilities(t_personality_1, t_personality_2, t_personality_3):
total = t_personality_1 + t_personality_2 + t_personality_3
if total > 1.0:
warning_message = f"警告: 选择模型1、模型2和模型3的概率总和为 {total:.2f},超过了 1.0!请调整滑块使总和等于 1.0。"
def adjust_model_greater_probabilities(t_model_1_probability, t_model_2_probability, t_model_3_probability):
total = Decimal(str(t_model_1_probability)) + Decimal(str(t_model_2_probability)) + Decimal(str(t_model_3_probability))
if total > Decimal('1.0'):
warning_message = f"警告: 选择模型1、模型2和模型3的概率总和为 {float(total):.2f},超过了 1.0!请调整滑块使总和等于 1.0。"
return warning_message
else:
return "" # 没有警告时返回空字符串
return "" # 没有警告时返回空字符串
def adjust_model_less_probabilities(t_personality_1, t_personality_2, t_personality_3):
total = t_personality_1 + t_personality_2 + t_personality_3
if total > 1.0:
warning_message = f"警告: 选择模型1、模型2和模型3的概率总和为 {total:.2f},小于了 1.0!请调整滑块使总和等于 1.0。"
def adjust_model_less_probabilities(t_model_1_probability, t_model_2_probability, t_model_3_probability):
total = Decimal(str(t_model_1_probability)) + Decimal(str(t_model_2_probability)) + Decimal(str(t_model_3_probability))
if total < Decimal('1.0'):
warning_message = f"警告: 选择模型1、模型2和模型3的概率总和为 {float(total):.2f},小于了 1.0!请调整滑块使总和等于 1.0。"
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
config_data["personality"]["personality_2_probability"] = t_personality_2
config_data["personality"]["personality_3_probability"] = t_personality_3
# ==============================================
# 人格保存函数
def save_personality_config(t_prompt_personality_1,
t_prompt_personality_2,
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"]["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)
logger.info("人格配置已保存到 bot_config.toml 文件中")
return "人格配置已保存"
def save_message_and_emoji_config(t_min_text_length,
t_max_context_size,
t_emoji_chance,
@@ -378,7 +493,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"]["tone_error_rate"] = t_tone_error_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
save_config_to_file(config_data)
logger.info("其他设置已保存到 bot_config.toml 文件中")
@@ -398,8 +513,15 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
gr.Markdown(
value="""
### 欢迎使用由墨梓柒MotricSeven编写的MaimBot配置文件编辑器\n
感谢ZureTz大佬提供的人格保存部分修复
"""
)
gr.Markdown(
value="## 全球在线MaiMBot数量: " + str((online_maimbot_data or {}).get('online_clients', 0))
)
gr.Markdown(
value="## 当前WebUI版本: " + str(WEBUI_VERSION)
)
gr.Markdown(
value="### 配置文件版本:" + config_data["inner"]["version"]
)
@@ -490,81 +612,99 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
)
with gr.Row():
gr.Markdown(
'''ChatAntWhere的baseURL和APIkey\n
'''日志设置\n
配置日志输出级别\n
改完了记得保存!!!
'''
)
with gr.Row():
chatanywhere_base_url = gr.Textbox(
label="ChatAntWhere的BaseURL",
value=env_config_data["env_CHAT_ANY_WHERE_BASE_URL"],
console_log_level = gr.Dropdown(
choices=["INFO", "DEBUG", "WARNING", "ERROR", "SUCCESS"],
label="控制台日志级别",
value=env_config_data.get("env_CONSOLE_LOG_LEVEL", "INFO"),
interactive=True
)
with gr.Row():
chatanywhere_key = gr.Textbox(
label="ChatAntWhere的key",
value=env_config_data["env_CHAT_ANY_WHERE_KEY"],
file_log_level = gr.Dropdown(
choices=["INFO", "DEBUG", "WARNING", "ERROR", "SUCCESS"],
label="文件日志级别",
value=env_config_data.get("env_FILE_LOG_LEVEL", "DEBUG"),
interactive=True
)
with gr.Row():
default_console_log_level = gr.Dropdown(
choices=["INFO", "DEBUG", "WARNING", "ERROR", "SUCCESS", "NONE"],
label="默认控制台日志级别",
value=env_config_data.get("env_DEFAULT_CONSOLE_LOG_LEVEL", "SUCCESS"),
interactive=True
)
with gr.Row():
default_file_log_level = gr.Dropdown(
choices=["INFO", "DEBUG", "WARNING", "ERROR", "SUCCESS", "NONE"],
label="默认文件日志级别",
value=env_config_data.get("env_DEFAULT_FILE_LOG_LEVEL", "DEBUG"),
interactive=True
)
with gr.Row():
gr.Markdown(
'''SiliconFlow的baseURL和APIkey\n
'''API设置\n
选择API提供商并配置相应的BaseURL和Key\n
改完了记得保存!!!
'''
)
with gr.Row():
siliconflow_base_url = gr.Textbox(
label="SiliconFlow的BaseURL",
value=env_config_data["env_SILICONFLOW_BASE_URL"],
with gr.Column(scale=3):
new_provider_input = gr.Textbox(
label="添加新提供商",
placeholder="输入新提供商名称"
)
add_provider_btn = gr.Button("添加提供商", scale=1)
with gr.Row():
api_provider = gr.Dropdown(
choices=MODEL_PROVIDER_LIST,
label="选择API提供商",
value=MODEL_PROVIDER_LIST[0] if MODEL_PROVIDER_LIST else None
)
with gr.Row():
api_base_url = gr.Textbox(
label="Base URL",
value=env_config_data.get(f"env_{MODEL_PROVIDER_LIST[0]}_BASE_URL", "") if MODEL_PROVIDER_LIST else "",
interactive=True
)
with gr.Row():
siliconflow_key = gr.Textbox(
label="SiliconFlow的key",
value=env_config_data["env_SILICONFLOW_KEY"],
api_key = gr.Textbox(
label="API Key",
value=env_config_data.get(f"env_{MODEL_PROVIDER_LIST[0]}_KEY", "") if MODEL_PROVIDER_LIST else "",
interactive=True
)
with gr.Row():
gr.Markdown(
'''DeepSeek的baseURL和APIkey\n
改完了记得保存!!!
'''
)
with gr.Row():
deepseek_base_url = gr.Textbox(
label="DeepSeek的BaseURL",
value=env_config_data["env_DEEP_SEEK_BASE_URL"],
interactive=True
)
with gr.Row():
deepseek_key = gr.Textbox(
label="DeepSeek的key",
value=env_config_data["env_DEEP_SEEK_KEY"],
interactive=True
)
with gr.Row():
volcengine_base_url = gr.Textbox(
label="VolcEngine的BaseURL",
value=env_config_data["env_VOLCENGINE_BASE_URL"],
interactive=True
)
with gr.Row():
volcengine_key = gr.Textbox(
label="VolcEngine的key",
value=env_config_data["env_VOLCENGINE_KEY"],
interactive=True
api_provider.change(
update_api_inputs,
inputs=[api_provider],
outputs=[api_base_url, api_key]
)
with gr.Row():
save_env_btn = gr.Button("保存环境配置",variant="primary")
with gr.Row():
save_env_btn.click(
save_trigger,
inputs=[server_address,server_port,final_result,mongodb_host,mongodb_port,mongodb_database_name,chatanywhere_base_url,chatanywhere_key,siliconflow_base_url,siliconflow_key,deepseek_base_url,deepseek_key,volcengine_base_url,volcengine_key],
inputs=[server_address, server_port, final_result, mongodb_host, mongodb_port, mongodb_database_name, console_log_level, file_log_level, default_console_log_level, default_file_log_level, api_provider, api_base_url, api_key],
outputs=[gr.Textbox(
label="保存结果",
interactive=False
)]
)
# 绑定添加提供商按钮的点击事件
add_provider_btn.click(
add_new_provider,
inputs=[new_provider_input, gr.State(value=MODEL_PROVIDER_LIST)],
outputs=[gr.State(value=MODEL_PROVIDER_LIST), api_provider]
).then(
lambda x: (env_config_data.get(f"env_{x}_BASE_URL", ""), env_config_data.get(f"env_{x}_KEY", "")),
inputs=[api_provider],
outputs=[api_base_url, api_key]
)
with gr.TabItem("1-Bot基础设置"):
with gr.Row():
with gr.Column(scale=3):
@@ -635,38 +775,92 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
with gr.Row():
prompt_personality_1 = gr.Textbox(
label="人格1提示词",
value=config_data['personality']['prompt_personality'][0],
interactive=True
value=config_data["personality"]["prompt_personality"][0],
interactive=True,
)
with gr.Row():
prompt_personality_2 = gr.Textbox(
label="人格2提示词",
value=config_data['personality']['prompt_personality'][1],
interactive=True
value=config_data["personality"]["prompt_personality"][1],
interactive=True,
)
with gr.Row():
prompt_personality_3 = gr.Textbox(
label="人格3提示词",
value=config_data['personality']['prompt_personality'][2],
interactive=True
value=config_data["personality"]["prompt_personality"][2],
interactive=True,
)
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_2 = gr.Slider(minimum=0, maximum=1, step=0.01, value=config_data["personality"]["personality_2_probability"], label="人格2概率")
personality_3 = gr.Slider(minimum=0, maximum=1, step=0.01, value=config_data["personality"]["personality_3_probability"], label="人格3概率")
# 创建三个滑块, 代表三个人格的概率
personality_1_probability = gr.Slider(
minimum=0,
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_less_text = gr.Markdown()
# 绑定滑块的值变化事件,确保总和必须等于 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])
personality_3.change(adjust_greater_probabilities, inputs=[personality_1, personality_2, personality_3], outputs=[warning_greater_text])
personality_1.change(adjust_less_probabilities, inputs=[personality_1, personality_2, personality_3], outputs=[warning_less_text])
personality_2.change(adjust_less_probabilities, inputs=[personality_1, personality_2, personality_3], outputs=[warning_less_text])
personality_3.change(adjust_less_probabilities, inputs=[personality_1, personality_2, personality_3], outputs=[warning_less_text])
# 输入的 3 个概率
personality_probability_change_inputs = [
personality_1_probability,
personality_2_probability,
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():
prompt_schedule = gr.Textbox(
label="日程生成提示词",
@@ -684,8 +878,16 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
personal_save_message = gr.Textbox(label="保存人格结果")
personal_save_btn.click(
save_personality_config,
inputs=[personality_1, personality_2, personality_3, prompt_schedule],
outputs=[personal_save_message]
inputs=[
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.Row():
@@ -728,7 +930,7 @@ with gr.Blocks(title="MaimBot配置文件编辑") as app:
choices=ban_words_list,
label="选择要删除的违禁词"
)
ban_words_delete_btn = gr.Button("删除", scale=1)
ban_words_delete_btn = gr.Button("删除", scale=1)
ban_words_final_result = gr.Text(label="修改后的违禁词")
ban_words_add_btn.click(