ContentView.swift

//
//  Created by Artem Novichkov on 17.08.2025.
//

import SwiftUI
import FoundationModels
import TranscriptDebugMenu

struct ContentView: View {
    @State private var isLoading = false
    @State private var text = ""
    @State private var session = LanguageModelSession(tools: [MoodTool()]) {
        "You're a helpful assistant that generates haikus. Always use `generateMood` tool to get a random mood for the haiku."
    }
    @State private var showTranscript = false

    var body: some View {
        NavigationStack {
            VStack {
                if isLoading {
                    ProgressView("Loading...")
                } else {
                    Text(text)
                }
            }
            .navigationTitle("Haiku")
            .transcriptDebugMenu(session, isPresented: $showTranscript)
            .toolbar {
                ToolbarItem {
                    Button {
                        showTranscript.toggle()
                    } label: {
                        Label("Transcript", systemImage: "gear")
                    }
                }
            }
            .onAppear {
                Task {
                    do {
                        isLoading = true
                        let response = try await session.respond(to: "Generate a haiku about Swift")
                        text = response.content
                        isLoading = false
                    } catch {
                        isLoading = false
                        print("Error: \(error)")
                    }
                }
            }
        }
    }
}

@Generable
enum Mood: String, CaseIterable {
    case happy, sad, thoughtful, excited, calm
}

final class MoodTool: Tool {
    let name = "generateMood"
    let description = "Generates a random mood for haiku"

    @Generable
    struct Arguments {}

    func call(arguments: Arguments) async throws -> Mood? {
        .allCases.randomElement()
    }
}

@Generable
struct Haiku {
    let text: String
}

#Preview {
    ContentView()
}