From ea354bd75e6d1c4c57bd1e853c90fd90135df10d Mon Sep 17 00:00:00 2001
From: Beingpax <prakashjoshipaxtek@gmail.com>
Date: Thu, 9 Oct 2025 10:52:38 +0545
Subject: [PATCH] Enable context awareness in power mode switch

---
 VoiceInk/PowerMode/PowerModeSessionManager.swift | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/VoiceInk/PowerMode/PowerModeSessionManager.swift b/VoiceInk/PowerMode/PowerModeSessionManager.swift
index e40276d..d326538 100644
--- a/VoiceInk/PowerMode/PowerModeSessionManager.swift
+++ b/VoiceInk/PowerMode/PowerModeSessionManager.swift
@@ -101,6 +101,7 @@ class PowerModeSessionManager {
 
         await MainActor.run {
             enhancementService.isEnhancementEnabled = config.isAIEnhancementEnabled
+            enhancementService.useScreenCaptureContext = config.useScreenCapture
 
             if config.isAIEnhancementEnabled {
                 if let promptId = config.selectedPrompt, let uuid = UUID(uuidString: promptId) {
@@ -140,6 +141,7 @@ class PowerModeSessionManager {
 
         await MainActor.run {
             enhancementService.isEnhancementEnabled = state.isEnhancementEnabled
+            enhancementService.useScreenCaptureContext = state.useScreenCaptureContext
             enhancementService.selectedPromptId = state.selectedPromptId.flatMap(UUID.init)
 
             if let aiService = enhancementService.getAIService() {