Merge pull request #5 from dennisimoo/custom-prompts

Add custom prompt template functionality

Author: rawandahmad698

Recap/UIComponents/Buttons/PillButton.swift

@@ -1,10 +1,3 @@
-//
-//  PillButton.swift
-//  Recap
-//
-//  Created by Rawand Ahmad on 25/07/2025.
-//
-
 import SwiftUI
 
 struct PillButton: View {

Recap/UseCases/Settings/Components/Reusable/CustomTextEditor.swift

@@ -0,0 +1,97 @@
+import SwiftUI
+
+struct CustomTextEditor: View {
+    let title: String
+    let textBinding: Binding<String>
+    let placeholder: String
+    let height: CGFloat
+    
+    @State private var isEditing = false
+    @FocusState private var isFocused: Bool
+    
+    init(
+        title: String,
+        text: Binding<String>,
+        placeholder: String = "",
+        height: CGFloat = 100
+    ) {
+        self.title = title
+        self.textBinding = text
+        self.placeholder = placeholder
+        self.height = height
+    }
+    
+    var body: some View {
+        VStack(alignment: .leading, spacing: 8) {
+            Text(title)
+                .font(.system(size: 11, weight: .medium))
+                .foregroundColor(UIConstants.Colors.textSecondary)
+            
+            ZStack(alignment: .topLeading) {
+                RoundedRectangle(cornerRadius: 8)
+                    .fill(Color(hex: "2A2A2A").opacity(0.3))
+                    .overlay(
+                        RoundedRectangle(cornerRadius: 8)
+                            .stroke(
+                                LinearGradient(
+                                    gradient: Gradient(stops: [
+                                        .init(color: Color(hex: "979797").opacity(isFocused ? 0.4 : 0.2), location: 0),
+                                        .init(color: Color(hex: "979797").opacity(isFocused ? 0.3 : 0.1), location: 1)
+                                    ]),
+                                    startPoint: .top,
+                                    endPoint: .bottom
+                                ),
+                                lineWidth: 0.8
+                            )
+                    )
+                    .frame(height: height)
+                
+                if textBinding.wrappedValue.isEmpty && !isFocused {
+                    Text(placeholder)
+                        .font(.system(size: 12, weight: .medium))
+                        .foregroundColor(UIConstants.Colors.textSecondary.opacity(0.6))
+                        .padding(.horizontal, 12)
+                        .padding(.vertical, 8)
+                        .allowsHitTesting(false)
+                }
+                
+                TextEditor(text: textBinding)
+                    .font(.system(size: 12, weight: .medium))
+                    .foregroundColor(UIConstants.Colors.textPrimary)
+                    .background(Color.clear)
+                    .scrollContentBackground(.hidden)
+                    .padding(.horizontal, 8)
+                    .padding(.vertical, 4)
+                    .focused($isFocused)
+                    .lineLimit(nil)
+                    .textSelection(.enabled)
+                    .onChange(of: isFocused) { _, focused in
+                        withAnimation(.easeInOut(duration: 0.2)) {
+                            isEditing = focused
+                        }
+                    }
+            }
+        }
+    }
+}
+
+#Preview {
+    VStack(spacing: 20) {
+        CustomTextEditor(
+            title: "Custom Prompt",
+            text: .constant(""),
+            placeholder: "Enter your custom prompt template here...",
+            height: 120
+        )
+        
+        CustomTextEditor(
+            title: "With Content",
+            text: .constant(UserPreferencesInfo.defaultPromptTemplate),
+            placeholder: "Enter text...",
+            height: 80
+        )
+    }
+    .frame(width: 400, height: 300)
+    .padding(20)
+    .background(Color.black)
+}

Recap/UseCases/Settings/Components/TabViews/GeneralSettingsView.swift

@@ -86,6 +86,31 @@ struct GeneralSettingsView<ViewModel: GeneralSettingsViewModelType>: View {
                         }
                     }
 
+                    SettingsCard(title: "Custom Prompt") {
+                        VStack(alignment: .leading, spacing: 12) {
+                            CustomTextEditor(
+                                title: "Prompt Template",
+                                text: viewModel.customPromptTemplate,
+                                placeholder: "Enter your custom prompt template here...",
+                                height: 120
+                            )
+                            
+                            HStack {
+                                Text("Customize how AI summarizes your meeting transcripts")
+                                    .font(.system(size: 11, weight: .regular))
+                                    .foregroundColor(UIConstants.Colors.textSecondary)
+                                
+                                Spacer()
+                                
+                                PillButton(text: "Reset to Default") {
+                                    Task {
+                                        await viewModel.resetToDefaultPrompt()
+                                    }
+                                }
+                            }
+                        }
+                    }
+                    
                 }
                 .padding(.horizontal, 20)
                 .padding(.vertical, 20)
@@ -121,12 +146,15 @@ struct GeneralSettingsView<ViewModel: GeneralSettingsViewModelType>: View {
 }
 
 #Preview {
-    GeneralSettingsView<PreviewGeneralSettingsViewModel>(viewModel: PreviewGeneralSettingsViewModel())
+    GeneralSettingsView(viewModel: PreviewGeneralSettingsViewModel())
         .frame(width: 550, height: 500)
         .background(Color.black)
 }
 
-private final class PreviewGeneralSettingsViewModel: ObservableObject, GeneralSettingsViewModelType {
+private final class PreviewGeneralSettingsViewModel: GeneralSettingsViewModelType {
+    var customPromptTemplate: Binding<String> {
+        .constant(UserPreferencesInfo.defaultPromptTemplate)
+    }
     @Published var availableModels: [LLMModelInfo] = [
         LLMModelInfo(name: "llama3.2", provider: "ollama"),
         LLMModelInfo(name: "codellama", provider: "ollama")
@@ -170,4 +198,8 @@ private final class PreviewGeneralSettingsViewModel: ObservableObject, GeneralSe
     func toggleAutoStopRecording(_ enabled: Bool) async {
         isAutoStopRecording = enabled
     }
+    
+    func updateCustomPromptTemplate(_ template: String) async {}
+    
+    func resetToDefaultPrompt() async {}
 }

Recap/UseCases/Settings/SettingsView.swift

@@ -150,9 +150,11 @@ struct SettingsView<GeneralViewModel: GeneralSettingsViewModelType>: View {
     .frame(width: 550, height: 500)
 }
 
-private final class PreviewGeneralSettingsViewModel: ObservableObject, GeneralSettingsViewModelType {
-    var activeWarnings: [WarningItem] = []
-    
+
+private final class PreviewGeneralSettingsViewModel: GeneralSettingsViewModelType {
+    var customPromptTemplate: Binding<String> {
+        .constant(UserPreferencesInfo.defaultPromptTemplate)
+    }
     @Published var availableModels: [LLMModelInfo] = [
         LLMModelInfo(name: "llama3.2", provider: "ollama"),
         LLMModelInfo(name: "codellama", provider: "ollama")
@@ -165,6 +167,15 @@ private final class PreviewGeneralSettingsViewModel: ObservableObject, GeneralSe
     @Published var errorMessage: String?
     @Published var showToast = false
     @Published var toastMessage = ""
+    @Published var activeWarnings: [WarningItem] = [
+        WarningItem(
+            id: "ollama",
+            title: "Ollama Not Running",
+            message: "Please start Ollama to use local AI models for summarization.",
+            icon: "server.rack",
+            severity: .warning
+        )
+    ]
 
     var hasModels: Bool {
         !availableModels.isEmpty
@@ -187,4 +198,8 @@ private final class PreviewGeneralSettingsViewModel: ObservableObject, GeneralSe
     func toggleAutoStopRecording(_ enabled: Bool) async {
         isAutoStopRecording = enabled
     }
+    
+    func updateCustomPromptTemplate(_ template: String) async {}
+    
+    func resetToDefaultPrompt() async {}
 }

Recap/UseCases/Settings/ViewModels/General/GeneralSettingsViewModel.swift

@@ -1,13 +1,27 @@
 import Foundation
 import Combine
+import SwiftUI
 
 @MainActor
-final class GeneralSettingsViewModel: ObservableObject, GeneralSettingsViewModelType {
+final class GeneralSettingsViewModel: GeneralSettingsViewModelType {
     @Published private(set) var availableModels: [LLMModelInfo] = []
     @Published private(set) var selectedModel: LLMModelInfo?
     @Published private(set) var selectedProvider: LLMProvider = .default
     @Published private(set) var autoDetectMeetings: Bool = false
     @Published private(set) var isAutoStopRecording: Bool = false
+    @Published private var customPromptTemplateValue: String = ""
+    
+    var customPromptTemplate: Binding<String> {
+        Binding(
+            get: { self.customPromptTemplateValue },
+            set: { newValue in
+                Task {
+                    await self.updateCustomPromptTemplate(newValue)
+                }
+            }
+        )
+    }
+
     @Published private(set) var isLoading = false
     @Published private(set) var errorMessage: String?
     @Published private(set) var showToast = false
@@ -31,7 +45,7 @@ final class GeneralSettingsViewModel: ObservableObject, GeneralSettingsViewModel
     init(
         llmService: LLMServiceType,
         userPreferencesRepository: UserPreferencesRepositoryType,
-        environmentValidator: EnvironmentValidatorType = EnvironmentValidator(),
+        environmentValidator: EnvironmentValidatorType,
         warningManager: WarningManagerType
     ) {
         self.llmService = llmService
@@ -58,10 +72,12 @@ final class GeneralSettingsViewModel: ObservableObject, GeneralSettingsViewModel
             selectedProvider = preferences.selectedProvider
             autoDetectMeetings = preferences.autoDetectMeetings
             isAutoStopRecording = preferences.autoStopRecording
+            customPromptTemplateValue = preferences.summaryPromptTemplate ?? UserPreferencesInfo.defaultPromptTemplate
         } catch {
             selectedProvider = .default
             autoDetectMeetings = false
             isAutoStopRecording = false
+            customPromptTemplateValue = UserPreferencesInfo.defaultPromptTemplate
         }
         await loadModels()
     }
@@ -167,4 +183,19 @@ final class GeneralSettingsViewModel: ObservableObject, GeneralSettingsViewModel
             isAutoStopRecording = !enabled
         }
     }
-}
+    
+    func updateCustomPromptTemplate(_ template: String) async {
+        customPromptTemplateValue = template
+        
+        do {
+            let templateToSave = template.isEmpty ? nil : template
+            try await userPreferencesRepository.updateSummaryPromptTemplate(templateToSave)
+        } catch {
+            errorMessage = error.localizedDescription
+        }
+    }
+    
+    func resetToDefaultPrompt() async {
+        await updateCustomPromptTemplate(UserPreferencesInfo.defaultPromptTemplate)
+    }
+}

Recap/UseCases/Settings/ViewModels/General/GeneralSettingsViewModelType.swift

@@ -1,5 +1,6 @@
 import Foundation
 import Combine
+import SwiftUI
 
 @MainActor
 protocol GeneralSettingsViewModelType: ObservableObject {
@@ -15,10 +16,13 @@ protocol GeneralSettingsViewModelType: ObservableObject {
     var showToast: Bool { get }
     var toastMessage: String { get }
     var activeWarnings: [WarningItem] { get }
+    var customPromptTemplate: Binding<String> { get }
 
     func loadModels() async
     func selectModel(_ model: LLMModelInfo) async
     func selectProvider(_ provider: LLMProvider) async
     func toggleAutoDetectMeetings(_ enabled: Bool) async
     func toggleAutoStopRecording(_ enabled: Bool) async
+    func updateCustomPromptTemplate(_ template: String) async
+    func resetToDefaultPrompt() async
 }