fastrepl
diff --git a/‎Cargo.lock‎
Lines changed: 3 additions & 0 deletions b/‎Cargo.lock‎
Lines changed: 3 additions & 0 deletions
diff --git a/‎apps/desktop/src/components/settings/components/ai/stt-view-local.tsx‎
Lines changed: 39 additions & 19 deletions b/‎apps/desktop/src/components/settings/components/ai/stt-view-local.tsx‎
Lines changed: 39 additions & 19 deletions
diff --git a/‎crates/am/Cargo.toml‎
Lines changed: 4 additions & 1 deletion b/‎crates/am/Cargo.toml‎
Lines changed: 4 additions & 1 deletion
diff --git a/‎crates/am/src/client.rs‎
Lines changed: 13 additions & 3 deletions b/‎crates/am/src/client.rs‎
Lines changed: 13 additions & 3 deletions
diff --git a/‎crates/am/src/error.rs‎
Lines changed: 6 additions & 0 deletions b/‎crates/am/src/error.rs‎
Lines changed: 6 additions & 0 deletions
@@ -112,6 +112,10 @@ export function STTViewLocal({
       {amAvailable && (
         <ProModelsManagement
           on={!!servers.data?.external}
+          selectedSTTModel={selectedSTTModel}
+          setSelectedSTTModel={setSelectedSTTModel}
+          downloadingModels={downloadingModels}
+          handleModelDownload={handleModelDownload}
         />
       )}
     </div>
@@ -142,7 +146,7 @@ function BasicModelsManagement({
       <div className="flex flex-col mb-3">
         <div className={cn(["text-sm font-semibold text-gray-700 flex items-center gap-2"])}>
           <h3>Basic Models</h3>
-          <span className={cn(["w-2 h-2 rounded-full", on ? "bg-blue-300 animate-pulse" : "bg-red-300"])} />
+          <span className={cn(["w-2 h-2 rounded-full", on ? "bg-blue-300 animate-pulse" : "bg-gray-100"])} />
         </div>
         <p className="text-xs text-gray-500">Default inference mode powered by Whisper.cpp.</p>
       </div>
@@ -164,14 +168,36 @@ function BasicModelsManagement({
   );
 }
 
-function ProModelsManagement({ on }: { on: boolean }) {
+function ProModelsManagement(
+  { on, selectedSTTModel, setSelectedSTTModel, downloadingModels, handleModelDownload }: {
+    on: boolean;
+    selectedSTTModel: string;
+    setSelectedSTTModel: (model: string) => void;
+    downloadingModels: Set<string>;
+    handleModelDownload: (model: string) => void;
+  },
+) {
+  const handleShowFileLocation = async () => {
+    localSttCommands.modelsDir().then((path) => openPath(path));
+  };
+
   const proModels = useQuery({
     queryKey: ["pro-models"],
     queryFn: async () => {
-      const models = await localSttCommands.listSupportedModels();
-      return models.filter((model) =>
-        model.key === "am-whisper-small-en" || model.key === "am-whisper-large-v3" || model.key === "am-parakeet-v2"
+      const models = await localSttCommands.listSupportedModels().then((models) =>
+        models.filter((model) => model.key === "am-whisper-large-v3" || model.key === "am-parakeet-v2")
       );
+      const downloaded = await Promise.all(
+        models.map(({ key }) => localSttCommands.isModelDownloaded(key)),
+      );
+
+      return models.map((model, index) => ({
+        name: model.display_name,
+        key: model.key,
+        downloaded: downloaded[index],
+        size: `${(model.size_bytes / 1024 / 1024).toFixed(0)} MB`,
+        fileName: "",
+      }));
     },
   });
 
@@ -181,7 +207,7 @@ function ProModelsManagement({ on }: { on: boolean }) {
         <div className="flex flex-col mb-3">
           <div className="text-sm font-semibold text-gray-700 flex items-center gap-2">
             <h3>Pro Models</h3>
-            <span className={cn(["w-2 h-2 rounded-full", on ? "bg-blue-300 animate-pulse" : "bg-red-300"])} />
+            <span className={cn(["w-2 h-2 rounded-full", on ? "bg-blue-300 animate-pulse" : "bg-gray-100"])} />
           </div>
           <p className="text-xs text-gray-500">
             Latency and resource optimized. Only for pro plan users.
@@ -192,19 +218,13 @@ function ProModelsManagement({ on }: { on: boolean }) {
           {proModels.data?.map((model) => (
             <ModelEntry
               key={model.key}
-              disabled={true}
-              model={{
-                name: model.display_name,
-                key: model.key,
-                downloaded: false,
-                size: `${(model.size_bytes / 1024 / 1024).toFixed(0)} MB`,
-                fileName: "",
-              }}
-              selectedSTTModel={""}
-              setSelectedSTTModel={() => {}}
-              downloadingModels={new Set()}
-              handleModelDownload={() => {}}
-              handleShowFileLocation={() => {}}
+              disabled={false}
+              model={model}
+              selectedSTTModel={selectedSTTModel}
+              setSelectedSTTModel={setSelectedSTTModel}
+              downloadingModels={downloadingModels}
+              handleModelDownload={handleModelDownload}
+              handleShowFileLocation={handleShowFileLocation}
             />
           ))}
         </div>
 
@@ -4,17 +4,20 @@ version = "0.1.0"
 edition = "2021"
 
 [dev-dependencies]
+hf-hub = "0.4.3"
 tokio = { workspace = true, features = ["rt", "macros"] }
 
 [dependencies]
 hypr-download-interface = { workspace = true }
+hypr-file = { workspace = true }
 
-hf-hub = "0.4.3"
 reqwest = { workspace = true, features = ["json"] }
+tracing = { workspace = true }
 
 dirs = { workspace = true }
 serde = { workspace = true, features = ["derive"] }
 serde_json = { workspace = true }
 specta = { workspace = true, features = ["derive"] }
 strum = { workspace = true, features = ["derive"] }
+tar = "0.4.44"
 thiserror = { workspace = true }
@@ -100,15 +100,25 @@ impl InitRequest {
         Self {
             api_key: api_key.into(),
             model: None,
-            download_base: None,
             model_repo: None,
             model_folder: None,
         }
     }
 
-    pub fn with_model(mut self, model: crate::AmModel) -> Self {
-        self.model = Some(model.model_key().to_string());
+    pub fn with_model(
+        mut self,
+        model: crate::AmModel,
+        base_dir: impl AsRef<std::path::Path>,
+    ) -> Self {
+        self.model = Some(model.model_dir().to_string());
         self.model_repo = Some(model.repo_name().to_string());
+        self.model_folder = Some(
+            base_dir
+                .as_ref()
+                .join(model.model_dir())
+                .to_string_lossy()
+                .to_string(),
+        );
         self
     }
 }
@@ -6,6 +6,9 @@ pub enum Error {
     #[error(transparent)]
     Request(#[from] reqwest::Error),
 
+    #[error(transparent)]
+    HyprFile(#[from] hypr_file::Error),
+
     #[error("Could not find home directory")]
     NoHomeDir,
 
@@ -17,4 +20,7 @@ pub enum Error {
 
     #[error("Unexpected response from server")]
     UnexpectedResponse,
+
+    #[error("Checksum mismatch")]
+    ChecksumMismatch,
 }