feat: 添加了自动拉取AI模型到本地的功能

Author: ghost-him

src-tauri/src/commands/utils.rs

@@ -148,3 +148,48 @@ fn export_logs_internal(log_dir: &str, save_path: &str) -> Result<(), String> {
 pub fn command_get_arch() -> String {
     std::env::consts::ARCH.to_string()
 }
+
+#[tauri::command]
+pub fn command_download_model(save_path: String) {
+    let url = "https://github.com/ghost-him/ZeroLaunch-rs/releases/download/model/EmbeddingGemma-300m.zip";
+    let save_path = Path::new(&save_path).join("EmbeddingGemma-300m.zip");
+    let save_path_str = save_path
+        .to_str()
+        .unwrap_or("EmbeddingGemma-300m.zip")
+        .replace("'", "''");
+
+    let script = format!(
+        "Write-Host 'Downloading AI Model...'; \
+         Write-Host (' '); \
+         Write-Host (' '); \
+         Write-Host (' '); \
+         Write-Host (' '); \
+         Write-Host (' '); \
+         Write-Host (' '); \
+         Write-Host (' '); \
+         $url = '{}'; \
+         $output = '{}'; \
+         Write-Host ('Source: ' + $url); \
+         Write-Host ('Destination: ' + $output); \
+         try {{ \
+             $start = Get-Date; \
+             Invoke-WebRequest -Uri $url -OutFile $output; \
+             $end = Get-Date; \
+             $duration = $end - $start; \
+             $file = Get-Item $output; \
+             $sizeMB = $file.Length / 1MB; \
+             Write-Host ('Download Complete!'); \
+             Write-Host ('File Size: {{0:N2}} MB' -f $sizeMB); \
+             Write-Host ('Time Elapsed: ' + $duration); \
+         }} catch {{ \
+             Write-Error $_; \
+         }} \
+         Read-Host 'Press Enter to exit...'",
+        url, save_path_str
+    );
+
+    std::process::Command::new("cmd")
+        .args(["/C", "start", "powershell", "-Command", &script])
+        .spawn()
+        .ok();
+}

src-tauri/src/lib.rs

@@ -215,6 +215,7 @@ pub fn run() {
             command_search_programs_lightweight,
             command_update_program_icon,
             command_get_arch,
+            command_download_model,
             get_everything_icon,
         ])
         .build(tauri::generate_context!())

src-ui/i18n/locales/en.json

@@ -243,6 +243,8 @@
     "semantic_tip_description": "Providing richer description information helps the model deliver more accurate functional search.",
     "semantic_tip_performance": "When enabled, initialization and database updates may use the GPU and significantly increase loading time.",
     "open_model_folder": "Open Model Folder",
+    "download_model": "Download Model",
+    "select_download_folder": "Select Download Folder",
     "open_model_description": "Open Description File",
     "sorting_algorithm_settings": "Sorting Algorithm Parameters",
     "enable_sorting": "Enable Sorting Algorithm",

src-ui/i18n/locales/zh-Hans.json

@@ -243,6 +243,8 @@
     "semantic_tip_description": "如果补充更丰富的描述信息，可让模型提供更准确的功能性搜索。",
     "semantic_tip_performance": "启用后，在初始化与更新数据库阶段可能占用 GPU，并显著延长加载时间。",
     "open_model_folder": "打开模型文件夹",
+    "download_model": "下载模型",
+    "select_download_folder": "选择下载保存的文件夹",
     "open_model_description": "打开描述信息文件",
     "sorting_algorithm_settings": "排序算法参数设置",
     "enable_sorting": "启用排序算法",

src-ui/i18n/locales/zh-Hant.json

@@ -243,6 +243,8 @@
     "semantic_tip_description": "若補充更完整的描述資訊，可讓模型提供更精準的功能性搜尋。",
     "semantic_tip_performance": "啟用後，在初始化與更新資料庫階段可能佔用 GPU，並明顯延長載入時間。",
     "open_model_folder": "開啟模型資料夾",
+    "download_model": "下載模型",
+    "select_download_folder": "選擇下載保存的資料夾",
     "open_model_description": "開啟描述資訊檔案",
     "sorting_algorithm_settings": "排序演算法參數設定",
     "enable_sorting": "啟用排序演算法",

src-ui/windows/settings/programs/Advanced.vue

@@ -125,6 +125,12 @@
               >
                 {{ t('program_index.open_model_folder') || '打开模型文件夹 (TODO)' }}
               </el-button>
+              <el-button
+                type="primary"
+                @click="downloadModel"
+              >
+                {{ t('program_index.download_model') || '下载模型' }}
+              </el-button>
             </div>
           </el-card>
         </div>
@@ -376,6 +382,7 @@ import { useRemoteConfigStore } from '../../../stores/remote_config'
 import { storeToRefs } from 'pinia'
 import { QuestionFilled } from '@element-plus/icons-vue'
 import { invoke } from '@tauri-apps/api/core'
+import { open } from '@tauri-apps/plugin-dialog'
 
 const { t } = useI18n()
 const configStore = useRemoteConfigStore()
@@ -391,6 +398,18 @@ const search_model = computed(() => [
 const openModelFolder = async () => {
     await invoke('command_open_models_dir')
 }
+
+const downloadModel = async () => {
+    const selected = await open({
+        directory: true,
+        multiple: false,
+        title: t('program_index.select_download_folder') || '选择下载保存的文件夹',
+    });
+    
+    if (selected) {
+        await invoke('command_download_model', { savePath: selected });
+    }
+}
 </script>
 
 <style scoped>