修复气泡

js/app.js

@@ -338,14 +338,15 @@ class ProductCanvasApp {
   }
 
   // 创建流式消息容器
-  createStreamingMessageContainer(messageId) {
-    const messageDiv = document.createElement('div');
-    messageDiv.className = 'flex justify-start';
-    messageDiv.innerHTML = `
-      <div class="chat-bubble-ai relative group streaming-text" data-message-id="${messageId}">
-        <div class="typing-cursor"></div>
-      </div>
-    `;
+  createStreamingMessageContainer(messageId) {
+    const messageDiv = document.createElement('div');
+    messageDiv.className = 'flex justify-start';
+    messageDiv.dataset.messageId = messageId;
+    messageDiv.innerHTML = `
+      <div class="chat-bubble-ai relative group streaming-text" data-message-id="${messageId}">
+        <div class="typing-cursor"></div>
+      </div>
+    `;
     return messageDiv;
   }
 
@@ -415,8 +416,16 @@ class ProductCanvasApp {
 
   // 完成流式消息
   finalizeStreamingMessage(messageId, fullContent, svgId = null, beforeText = '') {
-    const container = document.querySelector(`[data-message-id="${messageId}"]`);
-    if (!container) return;
+    let container = document.querySelector(`.chat-bubble-ai[data-message-id="${messageId}"]`);
+    if (!container) {
+      const fallback = document.querySelector(`[data-message-id="${messageId}"]`);
+      if (fallback) {
+        container = fallback.classList.contains('chat-bubble-ai')
+          ? fallback
+          : fallback.querySelector('.chat-bubble-ai');
+      }
+    }
+    if (!container) return;
     
     const message = {
       id: messageId,
@@ -982,4 +991,4 @@ class ProductCanvasApp {
 // 页面加载完成后初始化应用
 document.addEventListener('DOMContentLoaded', () => {
   window.app = new ProductCanvasApp();
-});
+});