From acaa11923f733ad12c59861ce2501570a9f47345 Mon Sep 17 00:00:00 2001
From: Beingpax <prakashjoshipaxtek@gmail.com>
Date: Thu, 7 Aug 2025 13:45:53 +0545
Subject: [PATCH] Fix AI enhancement icon fallback to use default prompt

---
 VoiceInk/Services/AIEnhancementService.swift     | 6 ++++++
 VoiceInk/Views/Recorder/RecorderComponents.swift | 2 +-
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/VoiceInk/Services/AIEnhancementService.swift b/VoiceInk/Services/AIEnhancementService.swift
index 0d880bc..5ca92f2 100644
--- a/VoiceInk/Services/AIEnhancementService.swift
+++ b/VoiceInk/Services/AIEnhancementService.swift
@@ -161,6 +161,12 @@ class AIEnhancementService: ObservableObject {
         }
         
         guard let activePrompt = activePrompt else {
+            // Use default prompt when none is selected
+            if let defaultPrompt = allPrompts.first(where: { $0.id == PredefinedPrompts.defaultPromptId }) {
+                var systemMessage = String(format: AIPrompts.customPromptTemplate, defaultPrompt.promptText)
+                systemMessage += contextSection
+                return systemMessage
+            }
             return AIPrompts.assistantMode + contextSection
         }
         
diff --git a/VoiceInk/Views/Recorder/RecorderComponents.swift b/VoiceInk/Views/Recorder/RecorderComponents.swift
index 0485a70..f15d07d 100644
--- a/VoiceInk/Views/Recorder/RecorderComponents.swift
+++ b/VoiceInk/Views/Recorder/RecorderComponents.swift
@@ -137,7 +137,7 @@ struct RecorderPromptButton: View {
     var body: some View {
         RecorderToggleButton(
             isEnabled: enhancementService.isEnhancementEnabled,
-            icon: enhancementService.activePrompt?.icon.rawValue ?? "brain",
+            icon: enhancementService.activePrompt?.icon.rawValue ?? enhancementService.allPrompts.first(where: { $0.id == PredefinedPrompts.defaultPromptId })?.icon.rawValue ?? "checkmark.seal.fill",
             color: .blue,
             disabled: false
         ) {