From adb51193c6ceb9e99554145ee6fef40f8e7aaf89 Mon Sep 17 00:00:00 2001
From: Beingpax <prakashjoshipaxtek@gmail.com>
Date: Wed, 23 Jul 2025 09:30:44 +0545
Subject: [PATCH] Move LocalModel to TranscriptionModel file

---
 VoiceInk/Models/PredefinedModels.swift   | 57 +++++++-----------------
 VoiceInk/Models/TranscriptionModel.swift | 25 +++++++++++
 2 files changed, 42 insertions(+), 40 deletions(-)

diff --git a/VoiceInk/Models/PredefinedModels.swift b/VoiceInk/Models/PredefinedModels.swift
index 96c8ae2..4fb2cd0 100644
--- a/VoiceInk/Models/PredefinedModels.swift
+++ b/VoiceInk/Models/PredefinedModels.swift
@@ -1,31 +1,5 @@
 import Foundation
  
- struct LocalModel: TranscriptionModel {
-     let id = UUID()
-     let name: String
-     let displayName: String
-     let size: String
-     let supportedLanguages: [String: String]
-     let description: String
-     let speed: Double
-     let accuracy: Double
-     let ramUsage: Double
-     let hash: String
-     let provider: ModelProvider = .local
- 
-     var downloadURL: String {
-         "https://huggingface.co/ggerganov/whisper.cpp/resolve/main/\(filename)"
-     }
- 
-     var filename: String {
-         "\(name).bin"
-     }
- 
-     var isMultilingualModel: Bool {
-         supportedLanguages.count > 1
-     }
- }
- 
  enum PredefinedModels {
     static func getLanguageDictionary(isMultilingual: Bool, provider: ModelProvider = .local) -> [String: String] {
         if !isMultilingual {
@@ -121,8 +95,7 @@ import Foundation
              description: "Tiny model, fastest, least accurate",
              speed: 0.95,
              accuracy: 0.6,
-             ramUsage: 0.3,
-             hash: "bd577a113a864445d4c299885e0cb97d4ba92b5f"
+             ramUsage: 0.3
          ),
          LocalModel(
              name: "ggml-tiny.en",
@@ -132,8 +105,17 @@ import Foundation
              description: "Tiny model optimized for English, fastest, least accurate",
              speed: 0.95,
              accuracy: 0.65,
-             ramUsage: 0.3,
-             hash: "c78c86eb1a8faa21b369bcd33207cc90d64ae9df"
+             ramUsage: 0.3
+         ),
+         LocalModel(
+             name: "ggml-base",
+             displayName: "Base",
+             size: "142 MiB",
+             supportedLanguages: getLanguageDictionary(isMultilingual: true, provider: .local),
+             description: "Base model, good balance between speed and accuracy, supports multiple languages",
+             speed: 0.85,
+             accuracy: 0.72,
+             ramUsage: 0.5
          ),
          LocalModel(
              name: "ggml-base.en",
@@ -143,8 +125,7 @@ import Foundation
              description: "Base model optimized for English, good balance between speed and accuracy",
              speed: 0.85,
              accuracy: 0.75,
-             ramUsage: 0.5,
-             hash: "137c40403d78fd54d454da0f9bd998f78703390c"
+             ramUsage: 0.5
          ),
          LocalModel(
              name: "ggml-large-v2",
@@ -154,8 +135,7 @@ import Foundation
              description: "Large model v2, slower than Medium but more accurate",
              speed: 0.3,
              accuracy: 0.96,
-             ramUsage: 3.8,
-             hash: "0f4c8e30f21cf1769f637135f521436792c48186"
+             ramUsage: 3.8
          ),
          LocalModel(
              name: "ggml-large-v3",
@@ -165,8 +145,7 @@ import Foundation
              description: "Large model v3, very slow but most accurate",
              speed: 0.3,
              accuracy: 0.98,
-             ramUsage: 3.9,
-             hash: "ad82bf6a9043ceed055076d0fd39f5f186ff8062"
+             ramUsage: 3.9
          ),
          LocalModel(
              name: "ggml-large-v3-turbo",
@@ -177,8 +156,7 @@ import Foundation
              "Large model v3 Turbo, faster than v3 with similar accuracy",
              speed: 0.75,
              accuracy: 0.97,
-             ramUsage: 1.8,
-             hash: "4af2b29d7ec73d781377bfd1758ca957a807e941"
+             ramUsage: 1.8
          ),
          LocalModel(
              name: "ggml-large-v3-turbo-q5_0",
@@ -188,8 +166,7 @@ import Foundation
              description: "Quantized version of Large v3 Turbo, faster with slightly lower accuracy",
              speed: 0.75,
              accuracy: 0.95,
-             ramUsage: 1.0,
-             hash: "e050f7970618a659205450ad97eb95a18d69c9ee"
+             ramUsage: 1.0
          ),
          
                  // Cloud Models
diff --git a/VoiceInk/Models/TranscriptionModel.swift b/VoiceInk/Models/TranscriptionModel.swift
index ac905ed..5515531 100644
--- a/VoiceInk/Models/TranscriptionModel.swift
+++ b/VoiceInk/Models/TranscriptionModel.swift
@@ -95,4 +95,29 @@ struct CustomCloudModel: TranscriptionModel, Codable {
         self.isMultilingualModel = isMultilingual
         self.supportedLanguages = supportedLanguages ?? PredefinedModels.getLanguageDictionary(isMultilingual: isMultilingual)
     }
+} 
+
+struct LocalModel: TranscriptionModel {
+    let id = UUID()
+    let name: String
+    let displayName: String
+    let size: String
+    let supportedLanguages: [String: String]
+    let description: String
+    let speed: Double
+    let accuracy: Double
+    let ramUsage: Double
+    let provider: ModelProvider = .local
+
+    var downloadURL: String {
+        "https://huggingface.co/ggerganov/whisper.cpp/resolve/main/\(filename)"
+    }
+
+    var filename: String {
+        "\(name).bin"
+    }
+
+    var isMultilingualModel: Bool {
+        supportedLanguages.count > 1
+    }
 } 
\ No newline at end of file