lunanting/lovart3
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 修复图片提取逻辑避免重复添加

Browse Source 
1. 只提取原始图片 URL，不提取缩略图
2. 有 markdown 格式时不再匹配普通 URL

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
Your Name 2026-02-01 12:47:55 +08:00
parent e0366c43fd
commit 3bac27265b
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
components/canvas/CanvasChatPanel.tsx
View File

@ -33,24 +33,31 @@ interface CanvasChatPanelProps {
// 建议卡片数据 - 已移除 // 建议卡片数据 - 已移除
const SUGGESTIONS: { title: string; desc: string; image: string }[] = []; const SUGGESTIONS: { title: string; desc: string; image: string }[] = [];
// 从消息中提取图片 URL // 从消息中提取图片 URL（只提取原始图片，避免重复）
function extractImageUrls(text: string): string[] { function extractImageUrls(text: string): string[] {
const urls: string[] = []; const urls: string[] = [];
// 匹配 markdown 图片格式: ![alt|||originalUrl|||size](thumbnailUrl) // 匹配 markdown 图片格式: ![thumb|||originalUrl|||size](thumbnailUrl)
const markdownRegex = /!\[(?:thumb)?\|\|\|([^|]+)\|\|\|[^\]]*\]\([^)]+\)/g; // 只提取原始图片 URL（第一个 ||| 和第二个 ||| 之间的 URL）
const markdownRegex = /!\[(?:thumb)?\|\|\|([^|]+)\|\|\|[^\]]*\]\(([^)]+)\)/g;
let match; let match;
let hasMarkdownImages = false;
while ((match = markdownRegex.exec(text)) !== null) { while ((match = markdownRegex.exec(text)) !== null) {
if (match[1]) { if (match[1]) {
urls.push(match[1]); urls.push(match[1]); // 原始图片 URL
hasMarkdownImages = true;
} }
} }
// 匹配普通图片 URL // 只有在没有 markdown 格式图片时，才匹配普通图片 URL
const urlRegex = /https?:\/\/[^\s"'<>]+\.(?:jpg|jpeg|png|gif|webp)/gi; // 这样可以避免重复匹配缩略图 URL
const urlMatches = text.match(urlRegex); if (!hasMarkdownImages) {
if (urlMatches) { const urlRegex = /https?:\/\/[^\s"'<>]+\.(?:jpg|jpeg|png|gif|webp)/gi;
urls.push(...urlMatches.filter(url => !urls.includes(url))); const urlMatches = text.match(urlRegex);
if (urlMatches) {
urls.push(...urlMatches.filter(url => !urls.includes(url)));
}
} }
return urls; return urls;