From fad746e39b5bda0f3324fd708ee99e3e9bd216f7 Mon Sep 17 00:00:00 2001
From: Beingpax <prakashjoshipaxtek@gmail.com>
Date: Wed, 4 Jun 2025 20:40:03 +0545
Subject: [PATCH] Always capture screen when AI enabled

---
 VoiceInk/PowerMode/ActiveWindowService.swift | 12 +++++-------
 1 file changed, 5 insertions(+), 7 deletions(-)

diff --git a/VoiceInk/PowerMode/ActiveWindowService.swift b/VoiceInk/PowerMode/ActiveWindowService.swift
index 4876123..0a2d655 100644
--- a/VoiceInk/PowerMode/ActiveWindowService.swift
+++ b/VoiceInk/PowerMode/ActiveWindowService.swift
@@ -134,14 +134,12 @@ class ActiveWindowService: ObservableObject {
             }
         }
         
+        // Wait for UI changes and model loading to complete first
+        try? await Task.sleep(nanoseconds: 1_500_000_000) // 1.5 seconds
+
+        // Then check if we should capture
         if config.isAIEnhancementEnabled && config.useScreenCapture {
-            let shouldCaptureScreen = !wasEnhancementEnabled || !wasScreenCaptureEnabled
-            
-            if shouldCaptureScreen {
-                // Wait a moment for UI changes and model loading to complete
-                try? await Task.sleep(nanoseconds: 1_500_000_000) // 1.5 seconds
-                await enhancementService.captureScreenContext()
-            }
+            await enhancementService.captureScreenContext()
         }
     }
 }