Gemma 4 model integration, language TTS, auto config wrangler URL

.gitignore

@@ -6,3 +6,10 @@ build/
 releases/
 .claude/
 coding-plans/
+
+/worker/wrangler.toml
+
+/worker/.wrangler/
+
+xcuserdata/
+*.xcuserdatad/

leanring-buddy.xcodeproj/project.pbxproj

@@ -411,7 +411,7 @@
 				CODE_SIGN_STYLE = Automatic;
 				COMBINE_HIDPI_IMAGES = YES;
 				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_TEAM = 2UDAY4J48G;
+				DEVELOPMENT_TEAM = ZY7Q367B6W;
 				ENABLE_APP_SANDBOX = NO;
 				ENABLE_HARDENED_RUNTIME = YES;
 				ENABLE_OUTGOING_NETWORK_CONNECTIONS = YES;
@@ -452,7 +452,7 @@
 				CODE_SIGN_STYLE = Automatic;
 				COMBINE_HIDPI_IMAGES = YES;
 				CURRENT_PROJECT_VERSION = 1;
-				DEVELOPMENT_TEAM = 2UDAY4J48G;
+				DEVELOPMENT_TEAM = ZY7Q367B6W;
 				ENABLE_APP_SANDBOX = NO;
 				ENABLE_HARDENED_RUNTIME = YES;
 				ENABLE_OUTGOING_NETWORK_CONNECTIONS = YES;

leanring-buddy/AppleSpeechTranscriptionProvider.swift

@@ -24,12 +24,13 @@ final class AppleSpeechTranscriptionProvider: BuddyTranscriptionProvider {
     let unavailableExplanation: String? = nil
 
     func startStreamingSession(
+        languageCode: String,
         keyterms: [String],
         onTranscriptUpdate: @escaping (String) -> Void,
         onFinalTranscriptReady: @escaping (String) -> Void,
         onError: @escaping (Error) -> Void
     ) async throws -> any BuddyStreamingTranscriptionSession {
-        guard let speechRecognizer = Self.makeBestAvailableSpeechRecognizer() else {
+        guard let speechRecognizer = Self.makeBestAvailableSpeechRecognizer(languageCode: languageCode) else {
             throw AppleSpeechTranscriptionProviderError(message: "dictation is not available on this mac.")
         }
 
@@ -41,8 +42,9 @@ final class AppleSpeechTranscriptionProvider: BuddyTranscriptionProvider {
         )
     }
 
-    private static func makeBestAvailableSpeechRecognizer() -> SFSpeechRecognizer? {
+    private static func makeBestAvailableSpeechRecognizer(languageCode: String) -> SFSpeechRecognizer? {
         let preferredLocales = [
+            Locale(identifier: languageCode),
             Locale.autoupdatingCurrent,
             Locale(identifier: "en-US")
         ]

leanring-buddy/AssemblyAIStreamingTranscriptionProvider.swift

@@ -17,10 +17,6 @@ struct AssemblyAIStreamingTranscriptionProviderError: LocalizedError {
 }
 
 final class AssemblyAIStreamingTranscriptionProvider: BuddyTranscriptionProvider {
-    /// URL for the Cloudflare Worker endpoint that returns a short-lived
-    /// AssemblyAI streaming token. The real API key never leaves the server.
-    private static let tokenProxyURL = "https://your-worker-name.your-subdomain.workers.dev/transcribe-token"
-
     let displayName = "AssemblyAI"
     let requiresSpeechRecognitionPermission = false
 
@@ -34,6 +30,7 @@ final class AssemblyAIStreamingTranscriptionProvider: BuddyTranscriptionProvider
     private let sharedWebSocketURLSession = URLSession(configuration: .default)
 
     func startStreamingSession(
+        languageCode: String,
         keyterms: [String],
         onTranscriptUpdate: @escaping (String) -> Void,
         onFinalTranscriptReady: @escaping (String) -> Void,
@@ -48,6 +45,7 @@ final class AssemblyAIStreamingTranscriptionProvider: BuddyTranscriptionProvider
             temporaryToken: temporaryToken,
             urlSession: sharedWebSocketURLSession,
             keyterms: keyterms,
+            languageCode: languageCode,
             onTranscriptUpdate: onTranscriptUpdate,
             onFinalTranscriptReady: onFinalTranscriptReady,
             onError: onError
@@ -59,7 +57,8 @@ final class AssemblyAIStreamingTranscriptionProvider: BuddyTranscriptionProvider
 
     /// Calls the Cloudflare Worker to get a short-lived AssemblyAI token.
     private func fetchTemporaryToken() async throws -> String {
-        var request = URLRequest(url: URL(string: Self.tokenProxyURL)!)
+        let baseURL = await WorkerEnvironment.shared.getBaseURL()
+        var request = URLRequest(url: URL(string: "\(baseURL)/transcribe-token")!)
         request.httpMethod = "POST"
 
         let (data, response) = try await URLSession.shared.data(for: request)
@@ -117,6 +116,7 @@ private final class AssemblyAIStreamingTranscriptionSession: NSObject, BuddyStre
     private let apiKey: String?
     private let temporaryToken: String?
     private let keyterms: [String]
+    private let languageCode: String
     private let onTranscriptUpdate: (String) -> Void
     private let onFinalTranscriptReady: (String) -> Void
     private let onError: (Error) -> Void
@@ -142,6 +142,7 @@ private final class AssemblyAIStreamingTranscriptionSession: NSObject, BuddyStre
         temporaryToken: String?,
         urlSession: URLSession,
         keyterms: [String],
+        languageCode: String,
         onTranscriptUpdate: @escaping (String) -> Void,
         onFinalTranscriptReady: @escaping (String) -> Void,
         onError: @escaping (Error) -> Void
@@ -150,6 +151,7 @@ private final class AssemblyAIStreamingTranscriptionSession: NSObject, BuddyStre
         self.temporaryToken = temporaryToken
         self.urlSession = urlSession
         self.keyterms = keyterms
+        self.languageCode = languageCode
         self.onTranscriptUpdate = onTranscriptUpdate
         self.onFinalTranscriptReady = onFinalTranscriptReady
         self.onError = onError
@@ -158,7 +160,8 @@ private final class AssemblyAIStreamingTranscriptionSession: NSObject, BuddyStre
     func open() async throws {
         let websocketURL = try Self.makeWebsocketURL(
             temporaryToken: temporaryToken,
-            keyterms: keyterms
+            keyterms: keyterms,
+            languageCode: languageCode
         )
 
         var websocketRequest = URLRequest(url: websocketURL)
@@ -436,7 +439,8 @@ private final class AssemblyAIStreamingTranscriptionSession: NSObject, BuddyStre
 
     private static func makeWebsocketURL(
         temporaryToken: String?,
-        keyterms: [String]
+        keyterms: [String],
+        languageCode: String
     ) throws -> URL {
         guard var websocketURLComponents = URLComponents(string: websocketBaseURLString) else {
             throw AssemblyAIStreamingTranscriptionProviderError(
@@ -451,6 +455,12 @@ private final class AssemblyAIStreamingTranscriptionSession: NSObject, BuddyStre
             URLQueryItem(name: "speech_model", value: "u3-rt-pro")
         ]
 
+        if languageCode != "en" {
+            // AssemblyAI supports multiple language codes.
+            queryItems.append(URLQueryItem(name: "language_code", value: languageCode))
+            // u3-rt-pro generally supports multi lingual. If any errors happen, we might need a different speech_model or drop it.
+        }
+
         let normalizedKeyterms = keyterms
             .map { $0.trimmingCharacters(in: .whitespacesAndNewlines) }
             .filter { !$0.isEmpty }

leanring-buddy/BuddyDictationManager.swift

@@ -517,7 +517,9 @@ final class BuddyDictationManager: NSObject, ObservableObject {
 
         print("🎙️ BuddyDictationManager: opening transcription provider \(transcriptionProvider.displayName)")
 
+        let languageCode = UserDefaults.standard.string(forKey: "selectedLanguageCode") ?? "en"
         let activeTranscriptionSession = try await transcriptionProvider.startStreamingSession(
+            languageCode: languageCode,
             keyterms: buildTranscriptionKeyterms(),
             onTranscriptUpdate: { [weak self] transcriptText in
                 Task { @MainActor in

leanring-buddy/BuddyTranscriptionProvider.swift

@@ -22,6 +22,7 @@ protocol BuddyTranscriptionProvider {
     var unavailableExplanation: String? { get }
 
     func startStreamingSession(
+        languageCode: String,
         keyterms: [String],
         onTranscriptUpdate: @escaping (String) -> Void,
         onFinalTranscriptReady: @escaping (String) -> Void,

leanring-buddy/ClaudeAPI.swift

@@ -10,12 +10,13 @@ class ClaudeAPI {
     private static let tlsWarmupLock = NSLock()
     private static var hasStartedTLSWarmup = false
 
-    private let apiURL: URL
+    var proxyURL: String
+    private var apiURL: URL { URL(string: proxyURL)! }
     var model: String
     private let session: URLSession
 
     init(proxyURL: String, model: String = "claude-sonnet-4-6") {
-        self.apiURL = URL(string: proxyURL)!
+        self.proxyURL = proxyURL
         self.model = model
 
         // Use .default instead of .ephemeral so TLS session tickets are cached.
@@ -29,11 +30,6 @@ class ClaudeAPI {
         config.urlCache = nil
         config.httpCookieStorage = nil
         self.session = URLSession(configuration: config)
-
-        // Fire a lightweight HEAD request in the background to pre-establish the TLS
-        // connection. This caches the TLS session ticket so the first real API call
-        // (which carries a large image payload) doesn't need a cold TLS handshake.
-        warmUpTLSConnectionIfNeeded()
     }
 
     private func makeAPIRequest() -> URLRequest {
@@ -63,7 +59,7 @@ class ClaudeAPI {
 
     /// Sends a no-op HEAD request to the API host to establish and cache a TLS session.
     /// Failures are silently ignored — this is purely an optimization.
-    private func warmUpTLSConnectionIfNeeded() {
+    func warmUpTLSConnectionIfNeeded() {
         Self.tlsWarmupLock.lock()
         let shouldStartTLSWarmup = !Self.hasStartedTLSWarmup
         if shouldStartTLSWarmup {