From 04adabffd5665a59933d56e602947fcdd5763818 Mon Sep 17 00:00:00 2001
From: Beingpax <prakashjoshipaxtek@gmail.com>
Date: Fri, 7 Mar 2025 21:21:37 +0545
Subject: [PATCH] feat: integrate screen capture with recording start in
 WhisperState - Added screen capture trigger when recording starts if
 enhancement and screen capture are enabled

---
 VoiceInk/Whisper/WhisperState.swift | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/VoiceInk/Whisper/WhisperState.swift b/VoiceInk/Whisper/WhisperState.swift
index 508202b..43f4d49 100644
--- a/VoiceInk/Whisper/WhisperState.swift
+++ b/VoiceInk/Whisper/WhisperState.swift
@@ -172,6 +172,13 @@ class WhisperState: NSObject, ObservableObject, AVAudioRecorderDelegate {
                             
                             await ActiveWindowService.shared.applyConfigurationForCurrentApp()
                             
+                            // Trigger screen capture if enhancement and screen capture are enabled
+                            if let enhancementService = self.enhancementService,
+                               enhancementService.isEnhancementEnabled && 
+                               enhancementService.useScreenCaptureContext {
+                                await enhancementService.captureScreenContext()
+                            }
+                            
                             if let currentModel = self.currentModel, self.whisperContext == nil {
                                 do {
                                     try await self.loadModel(currentModel)