CheetahApp.swift

import SwiftUI
import Combine
import LibWhisper
import CheetahIPC
import Sparkle

enumAnswerRequest{
case none
case answerQuestion
case refineAnswer(selection:Range<String.Index>?)
case analyzeCode
}

letdefaultWhisperModel="ggml-medium.en"

classAppViewModel:ObservableObject{
@AppStorage("authToken")varauthToken:String?
@AppStorage("useGPT4")varuseGPT4:Bool?

@Publishedvardevices=[CaptureDevice]()
@PublishedvarselectedDevice:CaptureDevice?

@PublishedvarwhisperModel= defaultWhisperModel
@PublishedvardownloadState=ModelDownloader.State.pending

@Publishedvaranalyzer:ConversationAnalyzer?
@PublishedvaranswerRequest=AnswerRequest.none
@PublishedvarerrorDescription:String?

@Publishedvartranscript:String?
@Publishedvaranswer:String?
@PublishedvarcodeAnswer:String?

@PublishedvarbuttonsAlwaysEnabled=false
}

@main
structCheetahApp:App{
@AppStorage("whisperModel")varpreferredWhisperModel:String?

@ObservedObjectvarviewModel=AppViewModel()

@Statevardownload:ModelDownloader?
@Statevarstream:WhisperStream?
@StatevaripcServer:IPCServer?

varextensionState=BrowserExtensionState()

letupdaterController=SPUStandardUpdaterController(startingUpdater:true, updaterDelegate:nil, userDriverDelegate:nil)

func start()async{
 viewModel.devices =try!CaptureDevice.devices

letdownloadConfig=URLSessionConfiguration.default
 downloadConfig.allowsExpensiveNetworkAccess =false
 downloadConfig.waitsForConnectivity =true

 viewModel.whisperModel = preferredWhisperModel ?? defaultWhisperModel
letdownload=ModelDownloader(modelName: viewModel.whisperModel, configuration: downloadConfig)
 download.$state.assign(to:&viewModel.$downloadState)
 download.resume()
self.download = download

 // Handle messages from ExtensionHelper
letserver=IPCServer()
 server.delegate = extensionState
 server.addSourceForNewLocalMessagePort(name:MessagePortName.browserExtensionServer.rawValue,
 toRunLoop:RunLoop.main.getCFRunLoop())
self.ipcServer = server

 // Install manifest needed for the browser extension to talk to ExtensionHelper
 _ =try?installNativeMessagingManifest()

whiletrue{
do{
fortryawaitrequestin viewModel.$answerRequest.receive(on:RunLoop.main).values {
iflet analyzer = viewModel.analyzer {
switch request {
case.answerQuestion:
tryawait analyzer.answer()
 viewModel.answer = analyzer.context[.answer]
 viewModel.codeAnswer = analyzer.context[.codeAnswer]
 viewModel.answerRequest =.none

case.refineAnswer(let selection):
tryawait analyzer.answer(refine:true, selection: selection)
 viewModel.answer = analyzer.context[.answer]
 viewModel.codeAnswer = analyzer.context[.codeAnswer]
 viewModel.answerRequest =.none

case.analyzeCode:
tryawait analyzer.analyzeCode(extensionState: extensionState)
 viewModel.answer = analyzer.context[.answer]
 viewModel.answerRequest =.none

case.none:
break
}
}
}
}catchlet error as ErrorResult{
 viewModel.errorDescription = error.message
 viewModel.answerRequest =.none
}catch{
 viewModel.errorDescription = error.localizedDescription
 viewModel.answerRequest =.none
}
}
}

varbody:someScene{
WindowGroup{
ContentView(viewModel: viewModel)
.task{
awaitstart()
}
.onChange(of: viewModel.selectedDevice){
setCaptureDevice($0)
}
}
.windowResizability(.contentSize)
.windowStyle(.hiddenTitleBar)
.commands{
CommandGroup(after:.appInfo){
CheckForUpdatesView(updater: updaterController.updater)
}
CommandGroup(replacing:.appSettings){
Button(action:{
 viewModel.authToken =nil
resetAfterSettingsChanged()
}){
Text("Change API Key…")
}
Button(action:{
if viewModel.useGPT4 ==true{
 viewModel.useGPT4 =false
}else{
 viewModel.useGPT4 =true
}
resetAfterSettingsChanged()
}){
Text("Use GPT-4")
if viewModel.useGPT4 ==true{
Image(systemName:"checkmark")
}
}
}
}
}

func resetAfterSettingsChanged(){
 viewModel.selectedDevice =nil
 viewModel.analyzer =nil
}

func setCaptureDevice(_ device:CaptureDevice?){
 stream?.cancel()

guardlet device = device,
let authToken = viewModel.authToken,
let modelURL = download?.modelURL else{
return
}

letstream=WhisperStream(model: modelURL, device: device)
 stream.start()
self.stream = stream

 stream.$segments
.receive(on:RunLoop.main)
.map{String($0.text)}
.assign(to:&viewModel.$transcript)

 viewModel.analyzer =ConversationAnalyzer(
 stream: stream,
 generator:PromptGenerator(),
 executor:.init(authToken: authToken, useGPT4: viewModel.useGPT4 ??false))
}
}