feat: 实现相似度检索工具
- 基于现有search_similar_outfits功能开发独立的相似度检索小工具 - 遵循promptx/tauri-desktop-app-expert开发规范 - 实现完整的前后端架构: * Rust后端命令接口 (similarity_search_commands.rs) * TypeScript类型定义 (similaritySearch.ts) * Zustand状态管理 (similaritySearchStore.ts) * React组件 (SimilaritySearchTool, SimilaritySearchPanel, SimilaritySearchResults, SimilaritySearchCard) * 服务层 (similaritySearchService.ts) 功能特性: - 智能搜索建议和自动完成 - 可调节的相关性阈值 (LOWEST/LOW/MEDIUM/HIGH) - 快速搜索标签 - 响应式网格布局结果展示 - 优雅的加载状态和错误处理 - 遵循UI/UX设计标准的美观界面 技术实现: - 复用现有outfit search API和数据模型 - 简化的搜索配置,专注核心功能 - 完整的TypeScript类型安全 - 现代化的React Hooks和状态管理 - TailwindCSS响应式设计 - 平滑的动画和交互效果 集成: - 添加到快捷工具列表 (/tools/similarity-search) - 配置React Router路由 - 注册Tauri命令处理器
This commit is contained in:
@@ -290,6 +290,10 @@ pub fn run() {
|
|||||||
commands::outfit_search_commands::get_supported_image_formats,
|
commands::outfit_search_commands::get_supported_image_formats,
|
||||||
commands::outfit_search_commands::get_default_search_config,
|
commands::outfit_search_commands::get_default_search_config,
|
||||||
commands::outfit_search_commands::get_outfit_search_config,
|
commands::outfit_search_commands::get_outfit_search_config,
|
||||||
|
// 相似度检索工具命令
|
||||||
|
commands::similarity_search_commands::quick_similarity_search,
|
||||||
|
commands::similarity_search_commands::get_similarity_search_suggestions,
|
||||||
|
commands::similarity_search_commands::get_similarity_search_config,
|
||||||
// 自定义标签管理命令
|
// 自定义标签管理命令
|
||||||
commands::custom_tag_commands::get_custom_tag_categories,
|
commands::custom_tag_commands::get_custom_tag_categories,
|
||||||
commands::custom_tag_commands::create_custom_tag_category,
|
commands::custom_tag_commands::create_custom_tag_category,
|
||||||
|
|||||||
@@ -19,6 +19,7 @@ pub mod export_record_commands;
|
|||||||
pub mod video_generation_commands;
|
pub mod video_generation_commands;
|
||||||
pub mod tools_commands;
|
pub mod tools_commands;
|
||||||
pub mod outfit_search_commands;
|
pub mod outfit_search_commands;
|
||||||
|
pub mod similarity_search_commands;
|
||||||
pub mod custom_tag_commands;
|
pub mod custom_tag_commands;
|
||||||
pub mod tolerant_json_commands;
|
pub mod tolerant_json_commands;
|
||||||
pub mod markdown_commands;
|
pub mod markdown_commands;
|
||||||
|
|||||||
@@ -0,0 +1,148 @@
|
|||||||
|
use tauri::{command, State};
|
||||||
|
use crate::infrastructure::app_state::AppState;
|
||||||
|
use crate::data::models::outfit_search::{SearchRequest, SearchResponse, SearchConfig, RelevanceThreshold};
|
||||||
|
use crate::presentation::commands::outfit_search_commands::search_similar_outfits;
|
||||||
|
|
||||||
|
/// 相似度检索工具命令
|
||||||
|
/// 遵循 Tauri 开发规范的命令设计模式
|
||||||
|
/// 基于现有的 search_similar_outfits 功能,提供简化的接口
|
||||||
|
|
||||||
|
/// 快速相似度搜索
|
||||||
|
/// 使用预设配置进行简化搜索
|
||||||
|
#[command]
|
||||||
|
pub async fn quick_similarity_search(
|
||||||
|
state: State<'_, AppState>,
|
||||||
|
query: String,
|
||||||
|
relevance_threshold: Option<String>,
|
||||||
|
) -> Result<SearchResponse, String> {
|
||||||
|
// 构建默认搜索配置
|
||||||
|
let threshold = match relevance_threshold.as_deref() {
|
||||||
|
Some("LOWEST") => RelevanceThreshold::Lowest,
|
||||||
|
Some("LOW") => RelevanceThreshold::Low,
|
||||||
|
Some("MEDIUM") => RelevanceThreshold::Medium,
|
||||||
|
Some("HIGH") => RelevanceThreshold::High,
|
||||||
|
_ => RelevanceThreshold::Medium, // 默认使用中等阈值
|
||||||
|
};
|
||||||
|
|
||||||
|
let config = SearchConfig {
|
||||||
|
relevance_threshold: threshold,
|
||||||
|
environments: Vec::new(),
|
||||||
|
categories: Vec::new(),
|
||||||
|
color_filters: std::collections::HashMap::new(),
|
||||||
|
design_styles: std::collections::HashMap::new(),
|
||||||
|
max_keywords: 10,
|
||||||
|
};
|
||||||
|
|
||||||
|
let request = SearchRequest {
|
||||||
|
query,
|
||||||
|
config,
|
||||||
|
page_size: 12, // 工具页面显示更多结果
|
||||||
|
page_offset: 0,
|
||||||
|
};
|
||||||
|
|
||||||
|
// 复用现有的搜索功能
|
||||||
|
search_similar_outfits(state, request).await
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 获取搜索建议(简化版)
|
||||||
|
#[command]
|
||||||
|
pub async fn get_similarity_search_suggestions(
|
||||||
|
_state: State<'_, AppState>,
|
||||||
|
query: String,
|
||||||
|
) -> Result<Vec<String>, String> {
|
||||||
|
// 基础搜索建议
|
||||||
|
let base_suggestions = vec![
|
||||||
|
"休闲搭配".to_string(),
|
||||||
|
"正式搭配".to_string(),
|
||||||
|
"运动风格".to_string(),
|
||||||
|
"街头风格".to_string(),
|
||||||
|
"简约风格".to_string(),
|
||||||
|
"复古风格".to_string(),
|
||||||
|
"牛仔裤搭配".to_string(),
|
||||||
|
"连衣裙搭配".to_string(),
|
||||||
|
"外套搭配".to_string(),
|
||||||
|
"夏季搭配".to_string(),
|
||||||
|
"冬季搭配".to_string(),
|
||||||
|
"约会搭配".to_string(),
|
||||||
|
"工作搭配".to_string(),
|
||||||
|
"聚会搭配".to_string(),
|
||||||
|
];
|
||||||
|
|
||||||
|
if query.is_empty() {
|
||||||
|
return Ok(base_suggestions);
|
||||||
|
}
|
||||||
|
|
||||||
|
// 基于查询过滤建议
|
||||||
|
let filtered_suggestions: Vec<String> = base_suggestions
|
||||||
|
.into_iter()
|
||||||
|
.filter(|suggestion| {
|
||||||
|
suggestion.contains(&query) ||
|
||||||
|
query.chars().any(|c| suggestion.contains(c))
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
// 如果没有匹配的建议,返回基础建议的前几个
|
||||||
|
if filtered_suggestions.is_empty() {
|
||||||
|
Ok(base_suggestions.into_iter().take(6).collect())
|
||||||
|
} else {
|
||||||
|
Ok(filtered_suggestions.into_iter().take(8).collect())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 获取相似度检索工具配置信息
|
||||||
|
#[command]
|
||||||
|
pub async fn get_similarity_search_config(
|
||||||
|
_state: State<'_, AppState>,
|
||||||
|
) -> Result<SimilaritySearchConfig, String> {
|
||||||
|
Ok(SimilaritySearchConfig {
|
||||||
|
available_thresholds: vec![
|
||||||
|
ThresholdOption {
|
||||||
|
value: "LOWEST".to_string(),
|
||||||
|
label: "最低 (0.3)".to_string(),
|
||||||
|
description: "显示更多相关结果".to_string(),
|
||||||
|
},
|
||||||
|
ThresholdOption {
|
||||||
|
value: "LOW".to_string(),
|
||||||
|
label: "较低 (0.5)".to_string(),
|
||||||
|
description: "包含较多相关结果".to_string(),
|
||||||
|
},
|
||||||
|
ThresholdOption {
|
||||||
|
value: "MEDIUM".to_string(),
|
||||||
|
label: "中等 (0.7)".to_string(),
|
||||||
|
description: "平衡相关性和数量".to_string(),
|
||||||
|
},
|
||||||
|
ThresholdOption {
|
||||||
|
value: "HIGH".to_string(),
|
||||||
|
label: "较高 (0.9)".to_string(),
|
||||||
|
description: "只显示高度相关结果".to_string(),
|
||||||
|
},
|
||||||
|
],
|
||||||
|
default_threshold: "MEDIUM".to_string(),
|
||||||
|
max_results_per_page: 12,
|
||||||
|
quick_search_tags: vec![
|
||||||
|
"休闲".to_string(),
|
||||||
|
"正式".to_string(),
|
||||||
|
"运动".to_string(),
|
||||||
|
"街头".to_string(),
|
||||||
|
"简约".to_string(),
|
||||||
|
"复古".to_string(),
|
||||||
|
],
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 相似度检索工具配置
|
||||||
|
#[derive(serde::Serialize)]
|
||||||
|
pub struct SimilaritySearchConfig {
|
||||||
|
pub available_thresholds: Vec<ThresholdOption>,
|
||||||
|
pub default_threshold: String,
|
||||||
|
pub max_results_per_page: usize,
|
||||||
|
pub quick_search_tags: Vec<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// 阈值选项
|
||||||
|
#[derive(serde::Serialize)]
|
||||||
|
pub struct ThresholdOption {
|
||||||
|
pub value: String,
|
||||||
|
pub label: String,
|
||||||
|
pub description: String,
|
||||||
|
}
|
||||||
@@ -17,6 +17,7 @@ import DebugPanelTool from './pages/tools/DebugPanelTool';
|
|||||||
import ChatTool from './pages/tools/ChatTool';
|
import ChatTool from './pages/tools/ChatTool';
|
||||||
import ChatTestPage from './pages/tools/ChatTestPage';
|
import ChatTestPage from './pages/tools/ChatTestPage';
|
||||||
import WatermarkTool from './pages/tools/WatermarkTool';
|
import WatermarkTool from './pages/tools/WatermarkTool';
|
||||||
|
import SimilaritySearchTool from './pages/tools/SimilaritySearchTool';
|
||||||
// import BatchThumbnailGenerator from './pages/tools/BatchThumbnailGenerator';
|
// import BatchThumbnailGenerator from './pages/tools/BatchThumbnailGenerator';
|
||||||
|
|
||||||
import Navigation from './components/Navigation';
|
import Navigation from './components/Navigation';
|
||||||
@@ -118,6 +119,7 @@ function App() {
|
|||||||
<Route path="/tools/ai-chat" element={<ChatTool />} />
|
<Route path="/tools/ai-chat" element={<ChatTool />} />
|
||||||
<Route path="/tools/chat-test" element={<ChatTestPage />} />
|
<Route path="/tools/chat-test" element={<ChatTestPage />} />
|
||||||
<Route path="/tools/watermark" element={<WatermarkTool />} />
|
<Route path="/tools/watermark" element={<WatermarkTool />} />
|
||||||
|
<Route path="/tools/similarity-search" element={<SimilaritySearchTool />} />
|
||||||
{/* <Route path="/tools/batch-thumbnail-generator" element={<BatchThumbnailGenerator />} /> */}
|
{/* <Route path="/tools/batch-thumbnail-generator" element={<BatchThumbnailGenerator />} /> */}
|
||||||
</Routes>
|
</Routes>
|
||||||
</div>
|
</div>
|
||||||
|
|||||||
186
apps/desktop/src/components/similarity/SimilaritySearchCard.tsx
Normal file
186
apps/desktop/src/components/similarity/SimilaritySearchCard.tsx
Normal file
@@ -0,0 +1,186 @@
|
|||||||
|
import React, { useState, useCallback } from 'react';
|
||||||
|
import {
|
||||||
|
ExternalLink,
|
||||||
|
Tag,
|
||||||
|
Image as ImageIcon,
|
||||||
|
TrendingUp
|
||||||
|
} from 'lucide-react';
|
||||||
|
import { SimilaritySearchCardProps } from '../../types/similaritySearch';
|
||||||
|
import SimilaritySearchService from '../../services/similaritySearchService';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 相似度检索结果卡片组件
|
||||||
|
* 遵循设计系统规范,提供统一的结果展示界面
|
||||||
|
*/
|
||||||
|
export const SimilaritySearchCard: React.FC<SimilaritySearchCardProps> = ({
|
||||||
|
result,
|
||||||
|
onSelect,
|
||||||
|
showScore = true,
|
||||||
|
compact = false,
|
||||||
|
}) => {
|
||||||
|
const [imageLoaded, setImageLoaded] = useState(false);
|
||||||
|
const [imageError, setImageError] = useState(false);
|
||||||
|
|
||||||
|
// 处理卡片点击
|
||||||
|
const handleCardClick = useCallback(() => {
|
||||||
|
if (onSelect) {
|
||||||
|
onSelect(result);
|
||||||
|
}
|
||||||
|
}, [result, onSelect]);
|
||||||
|
|
||||||
|
// 处理外部链接点击
|
||||||
|
const handleExternalClick = useCallback((e: React.MouseEvent) => {
|
||||||
|
e.stopPropagation();
|
||||||
|
if (result.image_url) {
|
||||||
|
window.open(result.image_url, '_blank', 'noopener,noreferrer');
|
||||||
|
}
|
||||||
|
}, [result.image_url]);
|
||||||
|
|
||||||
|
// 处理图片加载
|
||||||
|
const handleImageLoad = useCallback(() => {
|
||||||
|
setImageLoaded(true);
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
// 处理图片错误
|
||||||
|
const handleImageError = useCallback(() => {
|
||||||
|
setImageError(true);
|
||||||
|
setImageLoaded(true);
|
||||||
|
}, []);
|
||||||
|
|
||||||
|
// 获取相关性评分样式
|
||||||
|
const scoreColor = SimilaritySearchService.getRelevanceScoreColor(result.relevance_score);
|
||||||
|
const formattedScore = SimilaritySearchService.formatRelevanceScore(result.relevance_score);
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div
|
||||||
|
className={`
|
||||||
|
card card-interactive group cursor-pointer animate-fade-in-up
|
||||||
|
relative overflow-hidden bg-gradient-to-br from-white to-gray-50/50
|
||||||
|
hover:from-white hover:to-primary-50/30 transition-all duration-500
|
||||||
|
hover:shadow-lg hover:shadow-primary-500/10 hover:-translate-y-1
|
||||||
|
border border-gray-200 hover:border-primary-300
|
||||||
|
${compact ? 'p-4' : 'p-6'}
|
||||||
|
`}
|
||||||
|
onClick={handleCardClick}
|
||||||
|
>
|
||||||
|
{/* 装饰性背景 */}
|
||||||
|
<div className="absolute top-0 right-0 w-24 h-24 bg-gradient-to-br from-primary-100 to-primary-200 rounded-full -translate-y-12 translate-x-12 opacity-40 group-hover:opacity-60 transition-all duration-500 group-hover:scale-110"></div>
|
||||||
|
|
||||||
|
{/* 相关性评分 */}
|
||||||
|
{showScore && (
|
||||||
|
<div className="absolute top-3 right-3 z-10">
|
||||||
|
<div className={`px-2 py-1 rounded-full text-xs font-medium bg-white shadow-sm border ${scoreColor}`}>
|
||||||
|
<div className="flex items-center gap-1">
|
||||||
|
<TrendingUp className="w-3 h-3" />
|
||||||
|
{formattedScore}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
<div className="relative">
|
||||||
|
{/* 图片区域 */}
|
||||||
|
<div className={`relative ${compact ? 'h-32' : 'h-40'} bg-gray-100 rounded-lg overflow-hidden mb-4`}>
|
||||||
|
{result.image_url && !imageError ? (
|
||||||
|
<>
|
||||||
|
<img
|
||||||
|
src={result.image_url}
|
||||||
|
alt={result.style_description}
|
||||||
|
className={`w-full h-full object-cover transition-all duration-300 ${
|
||||||
|
imageLoaded ? 'opacity-100' : 'opacity-0'
|
||||||
|
}`}
|
||||||
|
onLoad={handleImageLoad}
|
||||||
|
onError={handleImageError}
|
||||||
|
/>
|
||||||
|
{!imageLoaded && (
|
||||||
|
<div className="absolute inset-0 flex items-center justify-center">
|
||||||
|
<div className="w-8 h-8 border-2 border-primary-500 border-t-transparent rounded-full animate-spin"></div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</>
|
||||||
|
) : (
|
||||||
|
<div className="w-full h-full flex items-center justify-center text-gray-400">
|
||||||
|
<ImageIcon className="w-8 h-8" />
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{/* 外部链接按钮 */}
|
||||||
|
{result.image_url && (
|
||||||
|
<button
|
||||||
|
onClick={handleExternalClick}
|
||||||
|
className="absolute top-2 left-2 p-1.5 bg-white/90 hover:bg-white rounded-lg shadow-sm transition-all duration-200 opacity-0 group-hover:opacity-100"
|
||||||
|
>
|
||||||
|
<ExternalLink className="w-3 h-3 text-gray-600" />
|
||||||
|
</button>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* 内容区域 */}
|
||||||
|
<div className="space-y-3">
|
||||||
|
{/* 标题和描述 */}
|
||||||
|
<div>
|
||||||
|
<h3 className={`font-semibold text-high-emphasis line-clamp-2 ${
|
||||||
|
compact ? 'text-sm' : 'text-base'
|
||||||
|
}`}>
|
||||||
|
{result.style_description || '未知风格'}
|
||||||
|
</h3>
|
||||||
|
{result.id && (
|
||||||
|
<p className="text-xs text-medium-emphasis mt-1">
|
||||||
|
ID: {result.id}
|
||||||
|
</p>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* 环境标签 */}
|
||||||
|
{result.environment_tags && result.environment_tags.length > 0 && (
|
||||||
|
<div className="flex flex-wrap gap-1">
|
||||||
|
{result.environment_tags.slice(0, compact ? 2 : 3).map((tag: string, index: number) => (
|
||||||
|
<span
|
||||||
|
key={index}
|
||||||
|
className="inline-flex items-center gap-1 px-2 py-1 bg-blue-100 text-blue-700 text-xs rounded-full"
|
||||||
|
>
|
||||||
|
<Tag className="w-2.5 h-2.5" />
|
||||||
|
{tag}
|
||||||
|
</span>
|
||||||
|
))}
|
||||||
|
{result.environment_tags.length > (compact ? 2 : 3) && (
|
||||||
|
<span className="inline-flex items-center px-2 py-1 bg-gray-100 text-gray-600 text-xs rounded-full">
|
||||||
|
+{result.environment_tags.length - (compact ? 2 : 3)}
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{/* 产品信息 */}
|
||||||
|
{result.products && result.products.length > 0 && (
|
||||||
|
<div className="space-y-2">
|
||||||
|
<div className="text-xs text-medium-emphasis font-medium">产品信息</div>
|
||||||
|
<div className="space-y-1">
|
||||||
|
{result.products.slice(0, compact ? 1 : 2).map((product: any, index: number) => (
|
||||||
|
<div key={index} className="text-xs text-gray-600">
|
||||||
|
<span className="font-medium">{product.category}</span>
|
||||||
|
{product.description && (
|
||||||
|
<span className="ml-2 text-gray-500 line-clamp-1">
|
||||||
|
{product.description}
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
{result.products.length > (compact ? 1 : 2) && (
|
||||||
|
<div className="text-xs text-gray-500">
|
||||||
|
还有 {result.products.length - (compact ? 1 : 2)} 个产品...
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* 悬停效果 */}
|
||||||
|
<div className="absolute inset-0 bg-gradient-to-br from-transparent via-transparent to-primary-50/20 opacity-0 group-hover:opacity-100 transition-opacity duration-500 rounded-lg"></div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
||||||
|
export default SimilaritySearchCard;
|
||||||
253
apps/desktop/src/components/similarity/SimilaritySearchPanel.tsx
Normal file
253
apps/desktop/src/components/similarity/SimilaritySearchPanel.tsx
Normal file
@@ -0,0 +1,253 @@
|
|||||||
|
import React, { useCallback, useRef, useEffect } from 'react';
|
||||||
|
import {
|
||||||
|
Search,
|
||||||
|
Sparkles,
|
||||||
|
Settings,
|
||||||
|
Loader2
|
||||||
|
} from 'lucide-react';
|
||||||
|
import {
|
||||||
|
SimilaritySearchPanelProps,
|
||||||
|
SimilaritySearchRequest
|
||||||
|
} from '../../types/similaritySearch';
|
||||||
|
import SimilaritySearchService from '../../services/similaritySearchService';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 相似度检索搜索面板组件
|
||||||
|
* 遵循 Tauri 开发规范的组件设计原则
|
||||||
|
*/
|
||||||
|
export const SimilaritySearchPanel: React.FC<SimilaritySearchPanelProps> = ({
|
||||||
|
query,
|
||||||
|
selectedThreshold,
|
||||||
|
config,
|
||||||
|
suggestions,
|
||||||
|
showSuggestions,
|
||||||
|
isSearching,
|
||||||
|
onQueryChange,
|
||||||
|
onThresholdChange,
|
||||||
|
onSearch,
|
||||||
|
onSuggestionSelect,
|
||||||
|
onSuggestionsToggle,
|
||||||
|
}) => {
|
||||||
|
const inputRef = useRef<HTMLInputElement>(null);
|
||||||
|
|
||||||
|
// 处理搜索执行
|
||||||
|
const handleSearch = useCallback(() => {
|
||||||
|
if (!SimilaritySearchService.validateQuery(query)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const request: SimilaritySearchRequest = {
|
||||||
|
query: query.trim(),
|
||||||
|
relevance_threshold: selectedThreshold,
|
||||||
|
};
|
||||||
|
|
||||||
|
onSearch(request);
|
||||||
|
}, [query, selectedThreshold, onSearch]);
|
||||||
|
|
||||||
|
// 处理回车键搜索
|
||||||
|
const handleKeyPress = useCallback((e: React.KeyboardEvent) => {
|
||||||
|
if (e.key === 'Enter' && !e.shiftKey) {
|
||||||
|
e.preventDefault();
|
||||||
|
handleSearch();
|
||||||
|
}
|
||||||
|
}, [handleSearch]);
|
||||||
|
|
||||||
|
// 处理建议选择
|
||||||
|
const handleSuggestionClick = useCallback((suggestion: string) => {
|
||||||
|
onSuggestionSelect(suggestion);
|
||||||
|
}, [onSuggestionSelect]);
|
||||||
|
|
||||||
|
// 处理输入框焦点
|
||||||
|
const handleInputFocus = useCallback(() => {
|
||||||
|
if (suggestions.length > 0) {
|
||||||
|
onSuggestionsToggle(true);
|
||||||
|
}
|
||||||
|
}, [suggestions.length, onSuggestionsToggle]);
|
||||||
|
|
||||||
|
// 处理输入框失焦
|
||||||
|
const handleInputBlur = useCallback(() => {
|
||||||
|
// 延迟隐藏建议,允许点击建议
|
||||||
|
setTimeout(() => {
|
||||||
|
onSuggestionsToggle(false);
|
||||||
|
}, 200);
|
||||||
|
}, [onSuggestionsToggle]);
|
||||||
|
|
||||||
|
// 自动聚焦输入框
|
||||||
|
useEffect(() => {
|
||||||
|
if (inputRef.current && !query) {
|
||||||
|
inputRef.current.focus();
|
||||||
|
}
|
||||||
|
}, [query]);
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="space-y-6">
|
||||||
|
{/* 搜索输入区域 */}
|
||||||
|
<div className="card p-6 animate-fade-in">
|
||||||
|
<div className="flex items-center gap-3 mb-4">
|
||||||
|
<div className="icon-container primary w-8 h-8">
|
||||||
|
<Search className="w-4 h-4" />
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<h3 className="text-heading-4 text-high-emphasis">智能搜索</h3>
|
||||||
|
<p className="text-xs text-medium-emphasis">输入关键词进行相似度检索</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="relative">
|
||||||
|
<div className="flex gap-3">
|
||||||
|
<div className="flex-1 relative group">
|
||||||
|
<input
|
||||||
|
ref={inputRef}
|
||||||
|
type="text"
|
||||||
|
value={query}
|
||||||
|
onChange={(e) => onQueryChange(e.target.value)}
|
||||||
|
onKeyDown={handleKeyPress}
|
||||||
|
onFocus={handleInputFocus}
|
||||||
|
onBlur={handleInputBlur}
|
||||||
|
placeholder="输入搜索关键词,如:休闲搭配、牛仔裤、正式风格..."
|
||||||
|
className="form-input pr-12 group-hover:border-primary-300 focus:border-primary-500 transition-colors duration-200"
|
||||||
|
disabled={isSearching}
|
||||||
|
/>
|
||||||
|
<div className="absolute right-3 top-1/2 transform -translate-y-1/2 text-gray-400 group-hover:text-primary-500 transition-colors duration-200">
|
||||||
|
<Sparkles className="w-5 h-5" />
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<button
|
||||||
|
onClick={handleSearch}
|
||||||
|
disabled={isSearching || !SimilaritySearchService.validateQuery(query)}
|
||||||
|
className="btn btn-primary px-6 min-w-[100px] disabled:opacity-50 disabled:cursor-not-allowed"
|
||||||
|
>
|
||||||
|
{isSearching ? (
|
||||||
|
<div className="flex items-center gap-2">
|
||||||
|
<Loader2 className="w-4 h-4 animate-spin" />
|
||||||
|
<span>搜索中</span>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<div className="flex items-center gap-2">
|
||||||
|
<Search className="w-4 h-4" />
|
||||||
|
<span>搜索</span>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* 搜索建议下拉 */}
|
||||||
|
{showSuggestions && suggestions.length > 0 && (
|
||||||
|
<div className="absolute top-full left-0 right-0 mt-2 bg-white border border-gray-200 rounded-xl shadow-lg z-50 animate-slide-in-up">
|
||||||
|
<div className="p-2">
|
||||||
|
<div className="text-xs text-gray-500 px-3 py-2 border-b border-gray-100">
|
||||||
|
搜索建议
|
||||||
|
</div>
|
||||||
|
<div className="max-h-60 overflow-y-auto">
|
||||||
|
{suggestions.map((suggestion, index) => (
|
||||||
|
<button
|
||||||
|
key={index}
|
||||||
|
onClick={() => handleSuggestionClick(suggestion)}
|
||||||
|
className="w-full text-left px-3 py-2 hover:bg-gray-50 rounded-lg transition-colors duration-200 flex items-center gap-2"
|
||||||
|
>
|
||||||
|
<Search className="w-4 h-4 text-gray-400" />
|
||||||
|
<span className="text-sm text-gray-700">{suggestion}</span>
|
||||||
|
</button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* 相关性阈值设置 */}
|
||||||
|
<div className="card p-6 animate-fade-in">
|
||||||
|
<div className="flex items-center gap-3 mb-4">
|
||||||
|
<div className="icon-container orange w-8 h-8">
|
||||||
|
<Settings className="w-4 h-4" />
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<h3 className="text-heading-4 text-high-emphasis">相关性阈值</h3>
|
||||||
|
<p className="text-xs text-medium-emphasis">调整搜索结果的相关性要求</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{config?.available_thresholds && (
|
||||||
|
<div className="space-y-3">
|
||||||
|
{config.available_thresholds.map((threshold) => (
|
||||||
|
<label
|
||||||
|
key={threshold.value}
|
||||||
|
className={`flex items-center gap-3 p-3 rounded-lg border-2 cursor-pointer transition-all duration-200 ${
|
||||||
|
selectedThreshold === threshold.value
|
||||||
|
? 'border-primary-500 bg-primary-50'
|
||||||
|
: 'border-gray-200 hover:border-gray-300 hover:bg-gray-50'
|
||||||
|
}`}
|
||||||
|
>
|
||||||
|
<input
|
||||||
|
type="radio"
|
||||||
|
name="threshold"
|
||||||
|
value={threshold.value}
|
||||||
|
checked={selectedThreshold === threshold.value}
|
||||||
|
onChange={(e) => onThresholdChange(e.target.value)}
|
||||||
|
className="sr-only"
|
||||||
|
/>
|
||||||
|
<div className={`w-4 h-4 rounded-full border-2 flex items-center justify-center ${
|
||||||
|
selectedThreshold === threshold.value
|
||||||
|
? 'border-primary-500 bg-primary-500'
|
||||||
|
: 'border-gray-300'
|
||||||
|
}`}>
|
||||||
|
{selectedThreshold === threshold.value && (
|
||||||
|
<div className="w-2 h-2 rounded-full bg-white"></div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
<div className="flex-1">
|
||||||
|
<div className="flex items-center justify-between">
|
||||||
|
<span className={`font-medium ${
|
||||||
|
selectedThreshold === threshold.value
|
||||||
|
? 'text-primary-900'
|
||||||
|
: 'text-gray-900'
|
||||||
|
}`}>
|
||||||
|
{threshold.label}
|
||||||
|
</span>
|
||||||
|
</div>
|
||||||
|
<p className={`text-sm mt-1 ${
|
||||||
|
selectedThreshold === threshold.value
|
||||||
|
? 'text-primary-700'
|
||||||
|
: 'text-gray-600'
|
||||||
|
}`}>
|
||||||
|
{threshold.description}
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
</label>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* 搜索提示 */}
|
||||||
|
<div className="card p-4 bg-gradient-to-r from-blue-50 to-indigo-50 border border-blue-200 animate-fade-in">
|
||||||
|
<div className="flex items-start gap-3">
|
||||||
|
<div className="icon-container blue w-6 h-6 mt-0.5 shadow-sm">
|
||||||
|
<Sparkles className="w-3 h-3" />
|
||||||
|
</div>
|
||||||
|
<div className="flex-1">
|
||||||
|
<h4 className="text-sm font-semibold text-blue-900 mb-2">💡 搜索提示</h4>
|
||||||
|
<ul className="text-xs text-blue-700 space-y-1.5">
|
||||||
|
<li className="flex items-center gap-2">
|
||||||
|
<div className="w-1 h-1 bg-blue-400 rounded-full"></div>
|
||||||
|
使用具体的关键词获得更精准的结果
|
||||||
|
</li>
|
||||||
|
<li className="flex items-center gap-2">
|
||||||
|
<div className="w-1 h-1 bg-blue-400 rounded-full"></div>
|
||||||
|
可以组合多个关键词,如"休闲 牛仔裤"
|
||||||
|
</li>
|
||||||
|
<li className="flex items-center gap-2">
|
||||||
|
<div className="w-1 h-1 bg-blue-400 rounded-full"></div>
|
||||||
|
调整相关性阈值来控制结果数量
|
||||||
|
</li>
|
||||||
|
</ul>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
||||||
|
export default SimilaritySearchPanel;
|
||||||
@@ -0,0 +1,185 @@
|
|||||||
|
import React, { useCallback } from 'react';
|
||||||
|
import {
|
||||||
|
Grid,
|
||||||
|
Clock,
|
||||||
|
TrendingUp,
|
||||||
|
ExternalLink
|
||||||
|
} from 'lucide-react';
|
||||||
|
import { SimilaritySearchResultsProps } from '../../types/similaritySearch';
|
||||||
|
import { SimilaritySearchCard } from './SimilaritySearchCard';
|
||||||
|
import SimilaritySearchService from '../../services/similaritySearchService';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 相似度检索结果展示组件
|
||||||
|
* 遵循 Tauri 开发规范的组件设计原则
|
||||||
|
*/
|
||||||
|
export const SimilaritySearchResults: React.FC<SimilaritySearchResultsProps> = ({
|
||||||
|
results,
|
||||||
|
totalResults,
|
||||||
|
currentPage,
|
||||||
|
maxResultsPerPage,
|
||||||
|
isLoading,
|
||||||
|
onPageChange,
|
||||||
|
onResultSelect,
|
||||||
|
}) => {
|
||||||
|
// 计算分页信息
|
||||||
|
const totalPages = SimilaritySearchService.getTotalPages(totalResults, maxResultsPerPage);
|
||||||
|
const pageInfo = SimilaritySearchService.getPageRangeInfo(currentPage, maxResultsPerPage, totalResults);
|
||||||
|
|
||||||
|
// 处理结果选择
|
||||||
|
const handleResultSelect = useCallback((result: any) => {
|
||||||
|
if (onResultSelect) {
|
||||||
|
onResultSelect(result);
|
||||||
|
}
|
||||||
|
}, [onResultSelect]);
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
if (isLoading) {
|
||||||
|
return (
|
||||||
|
<div className="card h-full flex flex-col items-center justify-center p-8 animate-fade-in">
|
||||||
|
<div className="icon-container primary w-16 h-16 mb-6">
|
||||||
|
<TrendingUp className="w-8 h-8 animate-pulse" />
|
||||||
|
</div>
|
||||||
|
<h3 className="text-xl font-semibold text-high-emphasis mb-2">正在搜索...</h3>
|
||||||
|
<p className="text-medium-emphasis">AI正在为您寻找最相似的内容</p>
|
||||||
|
|
||||||
|
{/* 加载动画 */}
|
||||||
|
<div className="mt-6 flex space-x-2">
|
||||||
|
{[0, 1, 2].map((i) => (
|
||||||
|
<div
|
||||||
|
key={i}
|
||||||
|
className="w-2 h-2 bg-primary-500 rounded-full animate-bounce"
|
||||||
|
style={{ animationDelay: `${i * 0.1}s` }}
|
||||||
|
></div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (results.length === 0) {
|
||||||
|
return (
|
||||||
|
<div className="card h-full flex flex-col items-center justify-center text-center p-8 animate-fade-in">
|
||||||
|
<div className="icon-container gray w-16 h-16 mb-6">
|
||||||
|
<Grid className="w-8 h-8" />
|
||||||
|
</div>
|
||||||
|
<h3 className="text-xl font-semibold text-high-emphasis mb-2">暂无搜索结果</h3>
|
||||||
|
<p className="text-medium-emphasis max-w-md">
|
||||||
|
没有找到匹配的内容,请尝试:
|
||||||
|
</p>
|
||||||
|
<ul className="text-sm text-medium-emphasis mt-3 space-y-1">
|
||||||
|
<li>• 使用不同的关键词</li>
|
||||||
|
<li>• 降低相关性阈值</li>
|
||||||
|
<li>• 尝试更通用的搜索词</li>
|
||||||
|
</ul>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="space-y-6 animate-fade-in">
|
||||||
|
{/* 结果统计 */}
|
||||||
|
<div className="card p-4 bg-gradient-to-r from-green-50 to-emerald-50 border border-green-200">
|
||||||
|
<div className="flex items-center justify-between">
|
||||||
|
<div className="flex items-center gap-3">
|
||||||
|
<div className="icon-container green w-8 h-8 shadow-sm">
|
||||||
|
<TrendingUp className="w-4 h-4" />
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<h3 className="text-sm font-semibold text-green-900">
|
||||||
|
🎯 找到 {totalResults} 个相似结果
|
||||||
|
</h3>
|
||||||
|
<p className="text-xs text-green-700">
|
||||||
|
显示第 {pageInfo.start}-{pageInfo.end} 个结果
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="flex items-center gap-2 text-xs text-green-600 bg-white px-3 py-1.5 rounded-full border border-green-200">
|
||||||
|
<Clock className="w-3 h-3" />
|
||||||
|
<span>按相关性排序</span>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* 结果网格 */}
|
||||||
|
<div className="grid grid-cols-1 sm:grid-cols-2 lg:grid-cols-3 xl:grid-cols-4 gap-4 lg:gap-6">
|
||||||
|
{results.map((result, index) => (
|
||||||
|
<SimilaritySearchCard
|
||||||
|
key={result.id || index}
|
||||||
|
result={result}
|
||||||
|
onSelect={handleResultSelect}
|
||||||
|
showScore={true}
|
||||||
|
compact={true}
|
||||||
|
/>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* 分页控制 */}
|
||||||
|
{totalPages > 1 && (
|
||||||
|
<div className="card p-4">
|
||||||
|
<div className="flex items-center justify-between">
|
||||||
|
<div className="text-sm text-medium-emphasis">
|
||||||
|
第 {currentPage} 页,共 {totalPages} 页
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="flex items-center gap-2">
|
||||||
|
<button
|
||||||
|
onClick={() => onPageChange(currentPage - 1)}
|
||||||
|
disabled={currentPage <= 1}
|
||||||
|
className="btn btn-secondary btn-sm disabled:opacity-50 disabled:cursor-not-allowed"
|
||||||
|
>
|
||||||
|
上一页
|
||||||
|
</button>
|
||||||
|
|
||||||
|
<div className="flex items-center gap-1">
|
||||||
|
{Array.from({ length: Math.min(5, totalPages) }, (_, i) => {
|
||||||
|
const page = i + 1;
|
||||||
|
return (
|
||||||
|
<button
|
||||||
|
key={page}
|
||||||
|
onClick={() => onPageChange(page)}
|
||||||
|
className={`w-8 h-8 text-sm rounded-lg transition-colors duration-200 ${
|
||||||
|
currentPage === page
|
||||||
|
? 'bg-primary-500 text-white'
|
||||||
|
: 'text-gray-600 hover:bg-gray-100'
|
||||||
|
}`}
|
||||||
|
>
|
||||||
|
{page}
|
||||||
|
</button>
|
||||||
|
);
|
||||||
|
})}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<button
|
||||||
|
onClick={() => onPageChange(currentPage + 1)}
|
||||||
|
disabled={currentPage >= totalPages}
|
||||||
|
className="btn btn-secondary btn-sm disabled:opacity-50 disabled:cursor-not-allowed"
|
||||||
|
>
|
||||||
|
下一页
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{/* 结果操作提示 */}
|
||||||
|
<div className="card p-4 bg-blue-50 border-blue-200">
|
||||||
|
<div className="flex items-start gap-3">
|
||||||
|
<div className="icon-container blue w-6 h-6 mt-0.5">
|
||||||
|
<ExternalLink className="w-3 h-3" />
|
||||||
|
</div>
|
||||||
|
<div className="flex-1">
|
||||||
|
<h4 className="text-sm font-medium text-blue-900 mb-1">操作提示</h4>
|
||||||
|
<p className="text-xs text-blue-700">
|
||||||
|
点击结果卡片查看详细信息,或使用外部链接在新窗口中打开
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
||||||
|
export default SimilaritySearchResults;
|
||||||
@@ -7,7 +7,8 @@ import {
|
|||||||
FileSearch,
|
FileSearch,
|
||||||
MessageCircle,
|
MessageCircle,
|
||||||
Droplets,
|
Droplets,
|
||||||
ImageIcon
|
ImageIcon,
|
||||||
|
Search
|
||||||
} from 'lucide-react';
|
} from 'lucide-react';
|
||||||
import { Tool, ToolCategory, ToolStatus } from '../types/tool';
|
import { Tool, ToolCategory, ToolStatus } from '../types/tool';
|
||||||
|
|
||||||
@@ -101,6 +102,21 @@ export const TOOLS_DATA: Tool[] = [
|
|||||||
isPopular: true,
|
isPopular: true,
|
||||||
version: '1.0.0',
|
version: '1.0.0',
|
||||||
lastUpdated: '2024-01-24'
|
lastUpdated: '2024-01-24'
|
||||||
|
},
|
||||||
|
{
|
||||||
|
id: 'similarity-search',
|
||||||
|
name: '相似度检索工具',
|
||||||
|
description: '基于AI的智能相似度搜索工具,支持多种相关性阈值和快速搜索功能',
|
||||||
|
longDescription: '强大的AI驱动相似度检索工具,基于先进的机器学习算法提供精准的内容匹配。支持可调节的相关性阈值、智能搜索建议、实时结果展示和批量处理功能。适用于图像、文本和多媒体内容的相似性分析。',
|
||||||
|
icon: Search,
|
||||||
|
route: '/tools/similarity-search',
|
||||||
|
category: ToolCategory.AI_TOOLS,
|
||||||
|
status: ToolStatus.STABLE,
|
||||||
|
tags: ['AI搜索', '相似度检索', '智能匹配', '机器学习', '内容分析'],
|
||||||
|
isNew: true,
|
||||||
|
isPopular: true,
|
||||||
|
version: '1.0.0',
|
||||||
|
lastUpdated: '2024-01-25'
|
||||||
}
|
}
|
||||||
];
|
];
|
||||||
|
|
||||||
|
|||||||
229
apps/desktop/src/pages/tools/SimilaritySearchTool.tsx
Normal file
229
apps/desktop/src/pages/tools/SimilaritySearchTool.tsx
Normal file
@@ -0,0 +1,229 @@
|
|||||||
|
import React, { useEffect, useCallback } from 'react';
|
||||||
|
import {
|
||||||
|
Search,
|
||||||
|
Sparkles,
|
||||||
|
RotateCcw,
|
||||||
|
TrendingUp,
|
||||||
|
Zap,
|
||||||
|
ArrowLeft
|
||||||
|
} from 'lucide-react';
|
||||||
|
import { useNavigate } from 'react-router-dom';
|
||||||
|
import {
|
||||||
|
useSimilaritySearchStore,
|
||||||
|
useSimilaritySearchSelectors,
|
||||||
|
useSimilaritySearchActions
|
||||||
|
} from '../../store/similaritySearchStore';
|
||||||
|
import { SimilaritySearchPanel } from '../../components/similarity/SimilaritySearchPanel';
|
||||||
|
import { SimilaritySearchResults } from '../../components/similarity/SimilaritySearchResults';
|
||||||
|
import { SimilaritySearchRequest } from '../../types/similaritySearch';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 相似度检索工具页面
|
||||||
|
* 遵循 Tauri 开发规范和 UI/UX 设计标准
|
||||||
|
*/
|
||||||
|
const SimilaritySearchTool: React.FC = () => {
|
||||||
|
const navigate = useNavigate();
|
||||||
|
|
||||||
|
// 状态管理
|
||||||
|
const {
|
||||||
|
query,
|
||||||
|
selectedThreshold,
|
||||||
|
setQuery,
|
||||||
|
setThreshold,
|
||||||
|
executeSearch,
|
||||||
|
loadConfig,
|
||||||
|
clearError,
|
||||||
|
setShowSuggestions,
|
||||||
|
} = useSimilaritySearchStore();
|
||||||
|
|
||||||
|
// 选择器
|
||||||
|
const { useSearchState, useConfigState, useSuggestionsState } = useSimilaritySearchSelectors();
|
||||||
|
const { quickSearch, resetAll, selectSuggestionAndSearch } = useSimilaritySearchActions();
|
||||||
|
|
||||||
|
// 获取状态
|
||||||
|
const searchState = useSearchState();
|
||||||
|
const configState = useConfigState();
|
||||||
|
const suggestionsState = useSuggestionsState();
|
||||||
|
|
||||||
|
// 页面初始化
|
||||||
|
useEffect(() => {
|
||||||
|
loadConfig();
|
||||||
|
clearError();
|
||||||
|
}, [loadConfig, clearError]);
|
||||||
|
|
||||||
|
// 处理搜索
|
||||||
|
const handleSearch = useCallback((request: SimilaritySearchRequest) => {
|
||||||
|
executeSearch(request);
|
||||||
|
}, [executeSearch]);
|
||||||
|
|
||||||
|
// 处理建议选择
|
||||||
|
const handleSuggestionSelect = useCallback((suggestion: string) => {
|
||||||
|
selectSuggestionAndSearch(suggestion);
|
||||||
|
}, [selectSuggestionAndSearch]);
|
||||||
|
|
||||||
|
// 处理快速搜索标签点击
|
||||||
|
const handleQuickTagClick = useCallback((tag: string) => {
|
||||||
|
quickSearch(tag);
|
||||||
|
}, [quickSearch]);
|
||||||
|
|
||||||
|
// 处理重置
|
||||||
|
const handleReset = useCallback(() => {
|
||||||
|
resetAll();
|
||||||
|
}, [resetAll]);
|
||||||
|
|
||||||
|
// 返回工具列表
|
||||||
|
const handleBackToTools = useCallback(() => {
|
||||||
|
navigate('/tools');
|
||||||
|
}, [navigate]);
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="min-h-screen bg-gradient-to-br from-gray-50 via-white to-gray-50">
|
||||||
|
{/* 页面头部 */}
|
||||||
|
<div className="bg-white border-b border-gray-100 shadow-sm">
|
||||||
|
<div className="container mx-auto px-4 sm:px-6 lg:px-8 py-4 sm:py-6">
|
||||||
|
<div className="flex items-center justify-between">
|
||||||
|
<div className="flex items-center gap-4">
|
||||||
|
<button
|
||||||
|
onClick={handleBackToTools}
|
||||||
|
className="flex items-center gap-2 px-3 py-2 text-gray-600 hover:text-primary-600 hover:bg-gray-50 rounded-lg transition-all duration-200"
|
||||||
|
>
|
||||||
|
<ArrowLeft className="w-4 h-4" />
|
||||||
|
<span className="text-sm font-medium">返回工具</span>
|
||||||
|
</button>
|
||||||
|
|
||||||
|
<div className="h-6 w-px bg-gray-200"></div>
|
||||||
|
|
||||||
|
<div className="flex items-center gap-3">
|
||||||
|
<div className="icon-container primary w-10 h-10">
|
||||||
|
<Search className="w-5 h-5" />
|
||||||
|
</div>
|
||||||
|
<div>
|
||||||
|
<h1 className="text-2xl font-bold text-high-emphasis">相似度检索工具</h1>
|
||||||
|
<p className="text-sm text-medium-emphasis">基于AI的智能相似度搜索</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div className="flex items-center gap-3">
|
||||||
|
<button
|
||||||
|
onClick={handleReset}
|
||||||
|
className="flex items-center gap-2 px-4 py-2 text-gray-600 hover:text-red-600 hover:bg-red-50 rounded-lg transition-all duration-200"
|
||||||
|
>
|
||||||
|
<RotateCcw className="w-4 h-4" />
|
||||||
|
<span className="text-sm font-medium">重置</span>
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* 主要内容 */}
|
||||||
|
<div className="container mx-auto px-4 sm:px-6 lg:px-8 py-6 lg:py-8">
|
||||||
|
{/* 快速开始提示 */}
|
||||||
|
{!searchState.results.length && !searchState.isSearching && (
|
||||||
|
<div className="mb-8 animate-fade-in">
|
||||||
|
<div className="card p-6 bg-gradient-to-r from-primary-50 via-blue-50 to-indigo-50 border border-primary-200 shadow-sm">
|
||||||
|
<div className="flex items-start gap-4">
|
||||||
|
<div className="icon-container primary w-12 h-12 shadow-md">
|
||||||
|
<Sparkles className="w-6 h-6" />
|
||||||
|
</div>
|
||||||
|
<div className="flex-1">
|
||||||
|
<h3 className="text-lg font-semibold text-primary-900 mb-2">🚀 开始您的相似度检索</h3>
|
||||||
|
<p className="text-primary-700 mb-4 leading-relaxed">
|
||||||
|
输入关键词或选择快速标签,AI将为您找到最相似的内容
|
||||||
|
</p>
|
||||||
|
|
||||||
|
{/* 快速标签 */}
|
||||||
|
{configState.config?.quick_search_tags && (
|
||||||
|
<div className="flex flex-wrap gap-2">
|
||||||
|
{configState.config.quick_search_tags.map((tag) => (
|
||||||
|
<button
|
||||||
|
key={tag}
|
||||||
|
onClick={() => handleQuickTagClick(tag)}
|
||||||
|
className="px-3 py-1.5 bg-white hover:bg-primary-100 text-primary-700 text-sm rounded-full border border-primary-200 transition-all duration-200 flex items-center gap-1 hover:shadow-sm hover:-translate-y-0.5"
|
||||||
|
>
|
||||||
|
<Zap className="w-3 h-3" />
|
||||||
|
{tag}
|
||||||
|
</button>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
|
||||||
|
{/* 搜索面板和结果 */}
|
||||||
|
<div className="grid grid-cols-1 lg:grid-cols-[400px_1fr] gap-6 lg:gap-8">
|
||||||
|
{/* 搜索面板 */}
|
||||||
|
<div className="space-y-6 order-2 lg:order-1">
|
||||||
|
<SimilaritySearchPanel
|
||||||
|
query={query}
|
||||||
|
selectedThreshold={selectedThreshold}
|
||||||
|
config={configState.config}
|
||||||
|
suggestions={suggestionsState.suggestions}
|
||||||
|
showSuggestions={suggestionsState.showSuggestions}
|
||||||
|
isSearching={searchState.isSearching}
|
||||||
|
onQueryChange={setQuery}
|
||||||
|
onThresholdChange={setThreshold}
|
||||||
|
onSearch={handleSearch}
|
||||||
|
onSuggestionSelect={handleSuggestionSelect}
|
||||||
|
onSuggestionsToggle={setShowSuggestions}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
{/* 搜索结果 */}
|
||||||
|
<div className="min-h-[400px] sm:min-h-[600px] order-1 lg:order-2">
|
||||||
|
{searchState.results.length > 0 ? (
|
||||||
|
<SimilaritySearchResults
|
||||||
|
results={searchState.results}
|
||||||
|
totalResults={searchState.totalResults}
|
||||||
|
currentPage={searchState.currentPage}
|
||||||
|
maxResultsPerPage={configState.config?.max_results_per_page || 12}
|
||||||
|
isLoading={searchState.isSearching}
|
||||||
|
onPageChange={() => {}} // 暂时不实现分页
|
||||||
|
onResultSelect={(result) => {
|
||||||
|
console.log('Selected result:', result);
|
||||||
|
}}
|
||||||
|
/>
|
||||||
|
) : searchState.error ? (
|
||||||
|
<div className="card h-full flex flex-col items-center justify-center text-center p-6 sm:p-12 animate-fade-in bg-gradient-to-br from-red-50 to-pink-50 border border-red-200">
|
||||||
|
<div className="icon-container red w-16 h-16 mb-6 shadow-md">
|
||||||
|
<Search className="w-8 h-8" />
|
||||||
|
</div>
|
||||||
|
<h3 className="text-2xl font-bold text-red-900 mb-3">❌ 搜索出错了</h3>
|
||||||
|
<p className="text-red-700 mb-6 max-w-md leading-relaxed">{searchState.error}</p>
|
||||||
|
<button
|
||||||
|
onClick={handleReset}
|
||||||
|
className="btn btn-primary hover:shadow-md transition-all duration-200"
|
||||||
|
>
|
||||||
|
🔄 重新开始
|
||||||
|
</button>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<div className="card h-full flex flex-col items-center justify-center text-center p-6 sm:p-12 animate-fade-in bg-gradient-to-br from-purple-50 to-indigo-50 border border-purple-200">
|
||||||
|
<div className="icon-container purple w-16 h-16 mb-6 shadow-md">
|
||||||
|
<TrendingUp className="w-8 h-8" />
|
||||||
|
</div>
|
||||||
|
<h3 className="text-2xl font-bold text-purple-900 mb-3">✨ 准备就绪</h3>
|
||||||
|
<div className="space-y-2 text-purple-700 max-w-md">
|
||||||
|
<p className="flex items-center justify-center gap-2">
|
||||||
|
<Search className="w-4 h-4" />
|
||||||
|
输入关键词开始搜索
|
||||||
|
</p>
|
||||||
|
<p className="flex items-center justify-center gap-2">
|
||||||
|
<Zap className="w-4 h-4" />
|
||||||
|
或选择快速标签快速开始
|
||||||
|
</p>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
||||||
|
export default SimilaritySearchTool;
|
||||||
163
apps/desktop/src/services/similaritySearchService.ts
Normal file
163
apps/desktop/src/services/similaritySearchService.ts
Normal file
@@ -0,0 +1,163 @@
|
|||||||
|
import { invoke } from '@tauri-apps/api/core';
|
||||||
|
import {
|
||||||
|
SimilaritySearchRequest,
|
||||||
|
SimilaritySearchConfig,
|
||||||
|
SearchResponse,
|
||||||
|
} from '../types/similaritySearch';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 相似度检索工具API服务
|
||||||
|
* 遵循 Tauri 开发规范的API服务设计原则
|
||||||
|
*/
|
||||||
|
export class SimilaritySearchService {
|
||||||
|
/**
|
||||||
|
* 执行快速相似度搜索
|
||||||
|
*/
|
||||||
|
static async quickSimilaritySearch(request: SimilaritySearchRequest): Promise<SearchResponse> {
|
||||||
|
try {
|
||||||
|
const response = await invoke<SearchResponse>('quick_similarity_search', {
|
||||||
|
query: request.query,
|
||||||
|
relevanceThreshold: request.relevance_threshold,
|
||||||
|
});
|
||||||
|
return response;
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Failed to perform similarity search:', error);
|
||||||
|
throw new Error(`相似度搜索失败: ${error}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取搜索建议
|
||||||
|
*/
|
||||||
|
static async getSuggestions(query: string): Promise<string[]> {
|
||||||
|
try {
|
||||||
|
const suggestions = await invoke<string[]>('get_similarity_search_suggestions', { query });
|
||||||
|
return suggestions;
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Failed to get search suggestions:', error);
|
||||||
|
return [];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取工具配置
|
||||||
|
*/
|
||||||
|
static async getConfig(): Promise<SimilaritySearchConfig> {
|
||||||
|
try {
|
||||||
|
const config = await invoke<SimilaritySearchConfig>('get_similarity_search_config');
|
||||||
|
return config;
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Failed to get similarity search config:', error);
|
||||||
|
throw new Error(`获取配置失败: ${error}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 验证搜索查询
|
||||||
|
*/
|
||||||
|
static validateQuery(query: string): boolean {
|
||||||
|
return query.trim().length > 0 && query.trim().length <= 200;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 格式化阈值标签
|
||||||
|
*/
|
||||||
|
static formatThresholdLabel(threshold: string): string {
|
||||||
|
const labels: Record<string, string> = {
|
||||||
|
'LOWEST': '最低',
|
||||||
|
'LOW': '较低',
|
||||||
|
'MEDIUM': '中等',
|
||||||
|
'HIGH': '较高',
|
||||||
|
};
|
||||||
|
return labels[threshold] || threshold;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取阈值描述
|
||||||
|
*/
|
||||||
|
static getThresholdDescription(threshold: string): string {
|
||||||
|
const descriptions: Record<string, string> = {
|
||||||
|
'LOWEST': '显示更多相关结果,包含相关性较低的内容',
|
||||||
|
'LOW': '包含较多相关结果,适合探索性搜索',
|
||||||
|
'MEDIUM': '平衡相关性和数量,推荐使用',
|
||||||
|
'HIGH': '只显示高度相关结果,结果更精准',
|
||||||
|
};
|
||||||
|
return descriptions[threshold] || '未知阈值';
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 生成搜索建议
|
||||||
|
*/
|
||||||
|
static generateSearchSuggestions(query: string, baseList: string[]): string[] {
|
||||||
|
if (!query || query.trim().length === 0) {
|
||||||
|
return baseList.slice(0, 6);
|
||||||
|
}
|
||||||
|
|
||||||
|
const lowercaseQuery = query.toLowerCase();
|
||||||
|
const filtered = baseList.filter(item =>
|
||||||
|
item.toLowerCase().includes(lowercaseQuery) ||
|
||||||
|
lowercaseQuery.split('').some(char => item.includes(char))
|
||||||
|
);
|
||||||
|
|
||||||
|
return filtered.length > 0 ? filtered.slice(0, 8) : baseList.slice(0, 6);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 计算总页数
|
||||||
|
*/
|
||||||
|
static getTotalPages(totalResults: number, pageSize: number): number {
|
||||||
|
return Math.ceil(totalResults / pageSize);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 检查是否有更多结果
|
||||||
|
*/
|
||||||
|
static hasMoreResults(currentPage: number, totalPages: number): boolean {
|
||||||
|
return currentPage < totalPages;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取页面范围信息
|
||||||
|
*/
|
||||||
|
static getPageRangeInfo(currentPage: number, pageSize: number, totalResults: number): {
|
||||||
|
start: number;
|
||||||
|
end: number;
|
||||||
|
total: number;
|
||||||
|
} {
|
||||||
|
const start = (currentPage - 1) * pageSize + 1;
|
||||||
|
const end = Math.min(currentPage * pageSize, totalResults);
|
||||||
|
return { start, end, total: totalResults };
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 格式化搜索时间
|
||||||
|
*/
|
||||||
|
static formatSearchTime(timeMs: number): string {
|
||||||
|
if (timeMs < 1000) {
|
||||||
|
return `${timeMs}ms`;
|
||||||
|
} else if (timeMs < 60000) {
|
||||||
|
return `${(timeMs / 1000).toFixed(1)}s`;
|
||||||
|
} else {
|
||||||
|
return `${(timeMs / 60000).toFixed(1)}min`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 获取相关性评分颜色
|
||||||
|
*/
|
||||||
|
static getRelevanceScoreColor(score: number): string {
|
||||||
|
if (score >= 0.9) return 'text-green-600';
|
||||||
|
if (score >= 0.7) return 'text-blue-600';
|
||||||
|
if (score >= 0.5) return 'text-yellow-600';
|
||||||
|
return 'text-gray-600';
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 格式化相关性评分
|
||||||
|
*/
|
||||||
|
static formatRelevanceScore(score: number): string {
|
||||||
|
return `${(score * 100).toFixed(0)}%`;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export default SimilaritySearchService;
|
||||||
208
apps/desktop/src/store/similaritySearchStore.ts
Normal file
208
apps/desktop/src/store/similaritySearchStore.ts
Normal file
@@ -0,0 +1,208 @@
|
|||||||
|
import { create } from 'zustand';
|
||||||
|
import {
|
||||||
|
SimilaritySearchState,
|
||||||
|
SimilaritySearchRequest,
|
||||||
|
DEFAULT_SIMILARITY_SEARCH_CONFIG,
|
||||||
|
} from '../types/similaritySearch';
|
||||||
|
import SimilaritySearchService from '../services/similaritySearchService';
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 相似度检索工具状态管理
|
||||||
|
* 遵循 Tauri 开发规范的状态管理模式
|
||||||
|
*/
|
||||||
|
export const useSimilaritySearchStore = create<SimilaritySearchState>((set, get) => ({
|
||||||
|
// 搜索状态
|
||||||
|
query: '',
|
||||||
|
selectedThreshold: DEFAULT_SIMILARITY_SEARCH_CONFIG.default_threshold || 'MEDIUM',
|
||||||
|
searchResults: [],
|
||||||
|
isSearching: false,
|
||||||
|
searchError: null,
|
||||||
|
|
||||||
|
// 配置
|
||||||
|
config: null,
|
||||||
|
isLoadingConfig: false,
|
||||||
|
|
||||||
|
// 建议
|
||||||
|
suggestions: [],
|
||||||
|
showSuggestions: false,
|
||||||
|
|
||||||
|
// 分页
|
||||||
|
currentPage: 1,
|
||||||
|
totalResults: 0,
|
||||||
|
|
||||||
|
// 操作方法
|
||||||
|
setQuery: (query: string) => {
|
||||||
|
set({ query });
|
||||||
|
|
||||||
|
// 自动加载建议
|
||||||
|
if (query.trim().length >= 2) {
|
||||||
|
get().loadSuggestions(query);
|
||||||
|
} else {
|
||||||
|
set({ suggestions: [], showSuggestions: false });
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
setThreshold: (threshold: string) => {
|
||||||
|
set({ selectedThreshold: threshold });
|
||||||
|
},
|
||||||
|
|
||||||
|
executeSearch: async (request: SimilaritySearchRequest) => {
|
||||||
|
const { query } = request;
|
||||||
|
|
||||||
|
// 验证查询
|
||||||
|
if (!SimilaritySearchService.validateQuery(query)) {
|
||||||
|
set({ searchError: '请输入有效的搜索关键词' });
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
set({
|
||||||
|
isSearching: true,
|
||||||
|
searchError: null,
|
||||||
|
showSuggestions: false,
|
||||||
|
currentPage: 1,
|
||||||
|
});
|
||||||
|
|
||||||
|
try {
|
||||||
|
const response = await SimilaritySearchService.quickSimilaritySearch(request);
|
||||||
|
|
||||||
|
set({
|
||||||
|
searchResults: response.results,
|
||||||
|
totalResults: response.total_size,
|
||||||
|
isSearching: false,
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
set({
|
||||||
|
searchError: error instanceof Error ? error.message : '搜索失败',
|
||||||
|
isSearching: false,
|
||||||
|
searchResults: [],
|
||||||
|
totalResults: 0,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
loadConfig: async () => {
|
||||||
|
set({ isLoadingConfig: true });
|
||||||
|
|
||||||
|
try {
|
||||||
|
const config = await SimilaritySearchService.getConfig();
|
||||||
|
set({
|
||||||
|
config,
|
||||||
|
selectedThreshold: config.default_threshold,
|
||||||
|
isLoadingConfig: false,
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Failed to load config:', error);
|
||||||
|
set({
|
||||||
|
isLoadingConfig: false,
|
||||||
|
// 使用默认配置
|
||||||
|
config: {
|
||||||
|
available_thresholds: [
|
||||||
|
{ value: 'LOWEST', label: '最低 (0.3)', description: '显示更多相关结果' },
|
||||||
|
{ value: 'LOW', label: '较低 (0.5)', description: '包含较多相关结果' },
|
||||||
|
{ value: 'MEDIUM', label: '中等 (0.7)', description: '平衡相关性和数量' },
|
||||||
|
{ value: 'HIGH', label: '较高 (0.9)', description: '只显示高度相关结果' },
|
||||||
|
],
|
||||||
|
default_threshold: 'MEDIUM',
|
||||||
|
max_results_per_page: 12,
|
||||||
|
quick_search_tags: ['休闲', '正式', '运动', '街头', '简约', '复古'],
|
||||||
|
},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
loadSuggestions: async (query: string) => {
|
||||||
|
try {
|
||||||
|
const suggestions = await SimilaritySearchService.getSuggestions(query);
|
||||||
|
set({
|
||||||
|
suggestions,
|
||||||
|
showSuggestions: suggestions.length > 0,
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
console.error('Failed to load suggestions:', error);
|
||||||
|
set({ suggestions: [], showSuggestions: false });
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
clearResults: () => {
|
||||||
|
set({
|
||||||
|
searchResults: [],
|
||||||
|
totalResults: 0,
|
||||||
|
currentPage: 1,
|
||||||
|
searchError: null,
|
||||||
|
});
|
||||||
|
},
|
||||||
|
|
||||||
|
clearError: () => {
|
||||||
|
set({ searchError: null });
|
||||||
|
},
|
||||||
|
|
||||||
|
setShowSuggestions: (show: boolean) => {
|
||||||
|
set({ showSuggestions: show });
|
||||||
|
},
|
||||||
|
}));
|
||||||
|
|
||||||
|
// 选择器 hooks
|
||||||
|
export const useSimilaritySearchSelectors = () => {
|
||||||
|
const store = useSimilaritySearchStore();
|
||||||
|
|
||||||
|
return {
|
||||||
|
// 搜索状态选择器
|
||||||
|
useSearchState: () => ({
|
||||||
|
query: store.query,
|
||||||
|
selectedThreshold: store.selectedThreshold,
|
||||||
|
results: store.searchResults,
|
||||||
|
isSearching: store.isSearching,
|
||||||
|
error: store.searchError,
|
||||||
|
totalResults: store.totalResults,
|
||||||
|
currentPage: store.currentPage,
|
||||||
|
}),
|
||||||
|
|
||||||
|
// 配置状态选择器
|
||||||
|
useConfigState: () => ({
|
||||||
|
config: store.config,
|
||||||
|
isLoadingConfig: store.isLoadingConfig,
|
||||||
|
}),
|
||||||
|
|
||||||
|
// 建议状态选择器
|
||||||
|
useSuggestionsState: () => ({
|
||||||
|
suggestions: store.suggestions,
|
||||||
|
showSuggestions: store.showSuggestions,
|
||||||
|
}),
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
// 操作 hooks
|
||||||
|
export const useSimilaritySearchActions = () => {
|
||||||
|
const store = useSimilaritySearchStore();
|
||||||
|
|
||||||
|
return {
|
||||||
|
// 快速搜索
|
||||||
|
quickSearch: async (query: string, threshold?: string) => {
|
||||||
|
const request: SimilaritySearchRequest = {
|
||||||
|
query,
|
||||||
|
relevance_threshold: threshold || store.selectedThreshold,
|
||||||
|
};
|
||||||
|
await store.executeSearch(request);
|
||||||
|
},
|
||||||
|
|
||||||
|
// 重置所有状态
|
||||||
|
resetAll: () => {
|
||||||
|
store.clearResults();
|
||||||
|
store.setQuery('');
|
||||||
|
store.setShowSuggestions(false);
|
||||||
|
store.clearError();
|
||||||
|
},
|
||||||
|
|
||||||
|
// 选择建议并搜索
|
||||||
|
selectSuggestionAndSearch: async (suggestion: string) => {
|
||||||
|
store.setQuery(suggestion);
|
||||||
|
store.setShowSuggestions(false);
|
||||||
|
|
||||||
|
const request: SimilaritySearchRequest = {
|
||||||
|
query: suggestion,
|
||||||
|
relevance_threshold: store.selectedThreshold,
|
||||||
|
};
|
||||||
|
await store.executeSearch(request);
|
||||||
|
},
|
||||||
|
};
|
||||||
|
};
|
||||||
133
apps/desktop/src/types/similaritySearch.ts
Normal file
133
apps/desktop/src/types/similaritySearch.ts
Normal file
@@ -0,0 +1,133 @@
|
|||||||
|
/**
|
||||||
|
* 相似度检索工具类型定义
|
||||||
|
* 遵循 Tauri 开发规范的类型定义模式
|
||||||
|
*/
|
||||||
|
|
||||||
|
// 重用现有的搜索结果类型
|
||||||
|
import type { SearchResult, SearchResponse } from './outfitSearch';
|
||||||
|
export type { SearchResult, SearchResponse };
|
||||||
|
|
||||||
|
// 阈值选项
|
||||||
|
export interface ThresholdOption {
|
||||||
|
value: string;
|
||||||
|
label: string;
|
||||||
|
description: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 相似度检索工具配置
|
||||||
|
export interface SimilaritySearchConfig {
|
||||||
|
available_thresholds: ThresholdOption[];
|
||||||
|
default_threshold: string;
|
||||||
|
max_results_per_page: number;
|
||||||
|
quick_search_tags: string[];
|
||||||
|
}
|
||||||
|
|
||||||
|
// 搜索请求(简化版)
|
||||||
|
export interface SimilaritySearchRequest {
|
||||||
|
query: string;
|
||||||
|
relevance_threshold?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 搜索状态
|
||||||
|
export interface SimilaritySearchState {
|
||||||
|
// 搜索状态
|
||||||
|
query: string;
|
||||||
|
selectedThreshold: string;
|
||||||
|
searchResults: SearchResult[];
|
||||||
|
isSearching: boolean;
|
||||||
|
searchError: string | null;
|
||||||
|
|
||||||
|
// 配置
|
||||||
|
config: SimilaritySearchConfig | null;
|
||||||
|
isLoadingConfig: boolean;
|
||||||
|
|
||||||
|
// 建议
|
||||||
|
suggestions: string[];
|
||||||
|
showSuggestions: boolean;
|
||||||
|
|
||||||
|
// 分页
|
||||||
|
currentPage: number;
|
||||||
|
totalResults: number;
|
||||||
|
|
||||||
|
// 操作方法
|
||||||
|
setQuery: (query: string) => void;
|
||||||
|
setThreshold: (threshold: string) => void;
|
||||||
|
executeSearch: (request: SimilaritySearchRequest) => Promise<void>;
|
||||||
|
loadConfig: () => Promise<void>;
|
||||||
|
loadSuggestions: (query: string) => Promise<void>;
|
||||||
|
clearResults: () => void;
|
||||||
|
clearError: () => void;
|
||||||
|
setShowSuggestions: (show: boolean) => void;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 组件属性接口
|
||||||
|
export interface SimilaritySearchPanelProps {
|
||||||
|
query: string;
|
||||||
|
selectedThreshold: string;
|
||||||
|
config: SimilaritySearchConfig | null;
|
||||||
|
suggestions: string[];
|
||||||
|
showSuggestions: boolean;
|
||||||
|
isSearching: boolean;
|
||||||
|
onQueryChange: (query: string) => void;
|
||||||
|
onThresholdChange: (threshold: string) => void;
|
||||||
|
onSearch: (request: SimilaritySearchRequest) => void;
|
||||||
|
onSuggestionSelect: (suggestion: string) => void;
|
||||||
|
onSuggestionsToggle: (show: boolean) => void;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface SimilaritySearchResultsProps {
|
||||||
|
results: SearchResult[];
|
||||||
|
totalResults: number;
|
||||||
|
currentPage: number;
|
||||||
|
maxResultsPerPage: number;
|
||||||
|
isLoading: boolean;
|
||||||
|
onPageChange: (page: number) => void;
|
||||||
|
onResultSelect?: (result: SearchResult) => void;
|
||||||
|
}
|
||||||
|
|
||||||
|
export interface SimilaritySearchCardProps {
|
||||||
|
result: SearchResult;
|
||||||
|
onSelect?: (result: SearchResult) => void;
|
||||||
|
showScore?: boolean;
|
||||||
|
compact?: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
// 默认配置
|
||||||
|
export const DEFAULT_SIMILARITY_SEARCH_CONFIG: Partial<SimilaritySearchConfig> = {
|
||||||
|
default_threshold: 'MEDIUM',
|
||||||
|
max_results_per_page: 12,
|
||||||
|
quick_search_tags: [
|
||||||
|
'休闲',
|
||||||
|
'正式',
|
||||||
|
'运动',
|
||||||
|
'街头',
|
||||||
|
'简约',
|
||||||
|
'复古'
|
||||||
|
],
|
||||||
|
};
|
||||||
|
|
||||||
|
// 常用搜索关键词
|
||||||
|
export const COMMON_SEARCH_KEYWORDS = [
|
||||||
|
'休闲搭配',
|
||||||
|
'正式搭配',
|
||||||
|
'运动风格',
|
||||||
|
'街头风格',
|
||||||
|
'简约风格',
|
||||||
|
'复古风格',
|
||||||
|
'牛仔裤搭配',
|
||||||
|
'连衣裙搭配',
|
||||||
|
'外套搭配',
|
||||||
|
'夏季搭配',
|
||||||
|
'冬季搭配',
|
||||||
|
'约会搭配',
|
||||||
|
'工作搭配',
|
||||||
|
'聚会搭配',
|
||||||
|
];
|
||||||
|
|
||||||
|
// 工具函数类型
|
||||||
|
export interface SimilaritySearchUtils {
|
||||||
|
formatThresholdLabel: (threshold: string) => string;
|
||||||
|
getThresholdDescription: (threshold: string) => string;
|
||||||
|
validateSearchQuery: (query: string) => boolean;
|
||||||
|
generateSearchSuggestions: (query: string, baseList: string[]) => string[];
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user