liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

去掉AIlogo

Browse Source
This commit is contained in:
liuwei
2025-08-27 09:55:54 +08:00
parent b2f94f540d
commit 09c10c0db0
1 changed files with 18 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
plugins/ai_gen_image/main.py
View File

@@ -71,18 +71,20 @@ class AIGenImagePlugin(MessagePluginInterface):
self._commands = self._config.get("AIGenImage", {}).get("command", ["AI绘图", "绘图", "画图", "生成图片"])
self.command_format = self._config.get("AIGenImage", {}).get("command-format", "AI绘图 描述文字")
self.enable = self._config.get("AIGenImage", {}).get("enable", True)
# API配置
self.image_api_url = self._config.get("AIGenImage", {}).get("image_api_url", "https://image.pollinations.ai/prompt/{prompt}")
self.image_api_url = self._config.get("AIGenImage", {}).get("image_api_url",
"https://image.pollinations.ai/prompt/{prompt}")
self.default_width = self._config.get("AIGenImage", {}).get("default_width", 1024)
self.default_height = self._config.get("AIGenImage", {}).get("default_height", 1024)
self.default_model = self._config.get("AIGenImage", {}).get("default_model", "turbo")
self.default_timeout = self._config.get("AIGenImage", {}).get("default_timeout", 300)
# 确保临时目录存在
self.temp_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))), 'temp')
self.temp_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))),
'temp')
os.makedirs(self.temp_dir, exist_ok=True)
self.LOG.info(f"[{self.name}] 插件初始化完成，指令：{self._commands}")
return True
@@ -136,7 +138,7 @@ class AIGenImagePlugin(MessagePluginInterface):
try:
# 发送提示消息
await bot.send_text_message((roomid if roomid else sender), f"🎨正在生成图片，请稍候...", sender)
# 生成图片
image_path = self._generate_image(prompt)
if not image_path or not os.path.exists(image_path):
@@ -160,29 +162,30 @@ class AIGenImagePlugin(MessagePluginInterface):
"width": self.default_width,
"height": self.default_height,
"model": self.default_model,
"seed": int(time.time()) % 1000000 # 使用时间戳作为随机种子
"seed": int(time.time()) % 1000000, # 使用时间戳作为随机种子
"nologo": "true" # Optional, set to "true" for registered referrers/tokens
}
# 编码提示词
encoded_prompt = urllib.parse.quote(prompt)
url = self.image_api_url.format(prompt=encoded_prompt)
self.LOG.info(f"正在生成图片，提示词: {prompt[:30]}...")
# 发送请求
response = requests.get(url, params=params, timeout=self.default_timeout)
response.raise_for_status()
# 保存图片
image_filename = f"ai_image_{uuid.uuid4().hex[:8]}.jpg"
image_path = os.path.join(self.temp_dir, image_filename)
with open(image_path, 'wb') as f:
f.write(response.content)
self.LOG.info(f"图片生成成功，保存至: {image_path}")
return image_path
except Exception as e:
self.LOG.error(f"生成图片出错: {e}")
return ""
return ""