优化聊天服务检测逻辑，增加基于主机名和URL的判断；更新内容脚本以输出当前主机名；修复数据提取器中的聊天标题获取逻辑；调整提示按钮的类名以符合新样式。

Author: whyiyhw

src/content/core/chatServices.js

@@ -7,7 +7,25 @@
  * @returns {String} 聊天服务名称或空字符串
  */
 export function detectChatService() {
-  // 尝试通过页面内容检测AI聊天服务
+  // 优先使用URL进行判断
+  const currentHost = window.location.hostname;
+  const currentUrl = window.location.href;
+
+  // 基于URL的精确检测
+  if (currentHost.includes('deepseek.com') || currentUrl.includes('deepseek.com')) {
+    return 'deepseek';
+  } else if (currentHost.includes('claude.ai') || currentUrl.includes('claude.ai')) {
+    return 'claude';
+  } else if (currentHost.includes('chat.openai.com') || currentUrl.includes('chat.openai.com')) {
+    return 'chatgpt';
+  } else if (currentHost.includes('gemini.google.com') || currentUrl.includes('gemini.google.com') ||
+             currentHost.includes('bard.google.com') || currentUrl.includes('bard.google.com')) {
+    return 'gemini';
+  } else if (currentHost.includes('yuanbao.tencent.com') || currentUrl.includes('yuanbao.tencent.com')) {
+    return 'yuanbao';
+  }
+
+  // 如果URL判断不成功，尝试通过页面内容检测
   if (document.title.includes('ChatGPT') || document.body.innerHTML.includes('ChatGPT')) {
     return 'chatgpt';
   } else if (document.title.includes('Claude') || document.body.innerHTML.includes('Claude')) {

src/content/features/data-extractors.js

@@ -465,7 +465,7 @@ async function extractAllDeepSeekData() {
                     const allChats = records.map(record => {
                         let chatData = {
                             id: record.key || (record.chat_session && record.chat_session.id) || '',
-                            title: record.data.title || '未命名聊天',
+                            title: record.chat_session.title || '未命名聊天',
                             timestamp: record.data.timestamp || new Date().toISOString(),
                             messages: []
                         };

src/content/features/prompt/promptButton.js

@@ -738,13 +738,13 @@ function createYuanbaoPromptButton(searchButton, buttonArea, isChinese) {
     });
 
     // 特别设置按钮的类名
-    promptButton.className = 'style__switch-model--btn___SVFm2 t-button t-button--theme-default t-button--variant-outline';
+    promptButton.className = 'yb-switch-model-btn yb-button-normal--border style__switch-model--btn___SVFm2 t-button t-button--theme-default t-button--variant-outline';
 
     // 设置内联样式
     promptButton.style.fontWeight = '500';
   } else {
     // 如果没有找到搜索按钮，使用默认样式
-    promptButton.className = 'style__switch-model--btn___SVFm2 t-button t-button--theme-default t-button--variant-outline';
+    promptButton.className = 'yb-switch-model-btn yb-button-normal--border style__switch-model--btn___SVFm2 t-button t-button--theme-default t-button--variant-outline';
     promptButton.style.fontWeight = '500';
   }
 

src/content/index.jsx

@@ -23,6 +23,7 @@ function createMountPoint() {
 function initExtension() {
   console.log('Deep Chat Optimize 内容脚本已加载');
   console.log('当前URL:', window.location.href);
+  console.log('当前主机名:', window.location.hostname);
 
   // 检测当前聊天服务
   const chatService = detectChatService();