jake/vOOice
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Revamp word replacement with direct regex implementation

Browse Source
This commit is contained in:
Beingpax 2025-03-18 10:47:08 +05:45
parent 4ceccb0990
commit 147daa5f5f
5 changed files with 76 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
VoiceInk/Services/AudioTranscriptionManager.swift
View File

@ -98,6 +98,11 @@ class AudioTranscriptionManager: ObservableObject {
var text = await whisperContext?.getTranscription() ?? ""
text = text.trimmingCharacters(in: .whitespacesAndNewlines)
// Apply word replacements if enabled
if UserDefaults.standard.bool(forKey: "IsWordReplacementEnabled") {
text = WordReplacementService.shared.applyReplacements(to: text)
}
// Handle enhancement if enabled
if let enhancementService = whisperState.enhancementService,
enhancementService.isEnhancementEnabled,

6
VoiceInk/Services/AudioTranscriptionService.swift
View File

@ -103,6 +103,12 @@ class AudioTranscriptionService: ObservableObject {
text = text.trimmingCharacters(in: .whitespacesAndNewlines)
logger.notice("✅ Retranscription completed successfully, length: \(text.count) characters")
// Apply word replacements if enabled
if UserDefaults.standard.bool(forKey: "IsWordReplacementEnabled") {
text = WordReplacementService.shared.applyReplacements(to: text)
logger.notice("✅ Word replacements applied")
}
// Apply AI enhancement if enabled - using the same enhancement service as WhisperState
if let enhancementService = enhancementService,
enhancementService.isEnhancementEnabled,

33
VoiceInk/Services/WordReplacementService.swift Normal file
View File

@ -0,0 +1,33 @@
import Foundation
class WordReplacementService {
static let shared = WordReplacementService()
private init() {}
func applyReplacements(to text: String) -> String {
guard let replacements = UserDefaults.standard.dictionary(forKey: "wordReplacements") as? [String: String],
!replacements.isEmpty else {
return text // No replacements to apply
}
var modifiedText = text
// Apply each replacement (case-insensitive, whole word)
for (original, replacement) in replacements {
// Create a regular expression that matches the word boundaries
let pattern = "\\b\(NSRegularExpression.escapedPattern(for: original))\\b"
if let regex = try? NSRegularExpression(pattern: pattern, options: .caseInsensitive) {
let range = NSRange(modifiedText.startIndex..., in: modifiedText)
modifiedText = regex.stringByReplacingMatches(
in: modifiedText,
options: [],
range: range,
withTemplate: replacement
)
}
}
return modifiedText
}
}

36
VoiceInk/Views/Dictionary/WordReplacementView.swift
View File

@ -7,8 +7,15 @@ class WordReplacementManager: ObservableObject {
}
}
@Published var isEnabled: Bool {
didSet {
UserDefaults.standard.set(isEnabled, forKey: "IsWordReplacementEnabled")
}
}
init() {
self.replacements = UserDefaults.standard.dictionary(forKey: "wordReplacements") as? [String: String] ?? [:]
self.isEnabled = UserDefaults.standard.bool(forKey: "IsWordReplacementEnabled")
}
func addReplacement(original: String, replacement: String) {
@ -28,17 +35,26 @@ struct WordReplacementView: View {
var body: some View {
VStack(alignment: .leading, spacing: 20) {
// Info Section
// Info Section with Toggle
GroupBox {
Label {
Text("Define word replacements to automatically replace specific words or phrases during AI enhancement")
.font(.system(size: 12))
.foregroundColor(.secondary)
.fixedSize(horizontal: false, vertical: true)
.frame(alignment: .leading)
} icon: {
Image(systemName: "info.circle.fill")
.foregroundColor(.blue)
HStack {
Label {
Text("Define word replacements to automatically replace specific words or phrases")
.font(.system(size: 12))
.foregroundColor(.secondary)
.fixedSize(horizontal: false, vertical: true)
.frame(alignment: .leading)
} icon: {
Image(systemName: "info.circle.fill")
.foregroundColor(.blue)
}
Spacer()
Toggle("Enable", isOn: $manager.isEnabled)
.toggleStyle(.switch)
.labelsHidden()
.help("Enable automatic word replacement after transcription")
}
}

14
VoiceInk/Whisper/WhisperState.swift
View File

@ -401,6 +401,12 @@ class WhisperState: NSObject, ObservableObject, AVAudioRecorderDelegate {
text = text.trimmingCharacters(in: .whitespacesAndNewlines)
logger.notice("✅ Transcription completed successfully, length: \(text.count) characters")
// Apply word replacements if enabled
if UserDefaults.standard.bool(forKey: "IsWordReplacementEnabled") {
text = WordReplacementService.shared.applyReplacements(to: text)
logger.notice("✅ Word replacements applied")
}
if let enhancementService = enhancementService,
enhancementService.isEnhancementEnabled,
enhancementService.isConfigured {
@ -444,14 +450,6 @@ class WhisperState: NSObject, ObservableObject, AVAudioRecorderDelegate {
try? modelContext.save()
}
if case .trialExpired = licenseViewModel.licenseState {
text = """
Your trial has expired. Upgrade to VoiceInk Pro at tryvoiceink.com/buy
\(text)
"""
}
messageLog += "Done: \(text)\n"
SoundManager.shared.playStopSound()