第一次學 Kotlin Koog AI 就上手 Day 11：讓 AI 讀懂文件：文件內容處理

user {
    text("請分析這份文件")
    attachments {
        binaryFile(Path(filePath), mimeType) // 核心語法
    }
}

/**
 * 根據文件副檔名取得對應的 MIME 類型
 * @param filePath 文件路徑
 * @return 對應的 MIME 類型字串
 */
fun getMimeType(filePath: String): String {
    val extension = kotlin.io.path.Path(filePath).fileName.toString().lowercase().substringAfterLast('.')
    return when (extension) {
        "pdf" -> "application/pdf"
        "txt" -> "text/plain"
        "md" -> "text/markdown"
        else -> "application/octet-stream"
    }
}

class DocumentSummarizer(private val client: GoogleLLMClient) {

    /**
     * 根據文件副檔名取得對應的 MIME 類型
     * @param filePath 文件路徑
     * @return 對應的 MIME 類型字串
     */
    private fun getMimeType(filePath: String): String {
        val extension = kotlin.io.path.Path(filePath).fileName.toString().lowercase().substringAfterLast('.')
        return when (extension) {
            "pdf" -> "application/pdf"
            "txt" -> "text/plain"
            "md" -> "text/markdown"
            else -> "application/octet-stream"
        }
    }

    /**
     * 產生文件摘要
     * @param filePath 文件路徑
     * @return 文件摘要內容
     */
    suspend fun summarizeDocument(filePath: String): String {
        val response = client.execute(
            prompt = prompt {
                user {
                    text("請為這份文件提供簡潔的摘要，重點包括：")
                    text("1. 主要主題和目的")
                    text("2. 核心內容概述")
                    text("3. 重要結論或建議")
                    text("請用繁體中文回答，字數控制在 300 字以內。")
                    attachments {
                        binaryFile(Path(filePath), getMimeType(filePath))
                    }
                }
            },
            model = GoogleModels.Gemini2_5Flash
        )

        return response.first().content
    }

    /**
     * 提取文件關鍵要點
     * 從指定文件中提取 5-10 個最重要的關鍵要點，以條列式呈現
     * @param filePath 文件路徑，支援 PDF、TXT、MD 等格式
     * @return 格式化的關鍵要點列表字串
     */
    suspend fun extractKeyPoints(filePath: String): String {
        val response = client.execute(
            prompt = prompt {
                user {
                    text("請從這份文件中提取 5-10 個最重要的關鍵要點。")
                    text("請以條列式呈現，每個要點簡潔明確。")
                    text("格式：")
                    text("• 要點 1")
                    text("• 要點 2")
                    text("...")
                    attachments {
                        binaryFile(Path(filePath), getMimeType(filePath))
                    }
                }
            },
            model = GoogleModels.Gemini2_5Flash
        )

        return response.first().content
    }

    /**
     * 分析文件結構
     * 分析文件的組織方式、章節結構、資訊邏輯順序等
     * @param filePath 文件路徑，支援各種格式的結構化文件
     * @return 包含文件類型、結構分析的詳細描述
     */
    suspend fun analyzeStructure(filePath: String): String {
        val response = client.execute(
            prompt = prompt {
                user {
                    text("請分析這份文件的結構和組織方式，包括：")
                    text("1. 文件類型和性質")
                    text("2. 主要章節或段落結構")
                    text("3. 資訊的邏輯順序")
                    text("4. 是否包含圖表、表格等特殊元素")
                    attachments {
                        binaryFile(Path(filePath), getMimeType(filePath))
                    }
                }
            },
            model = GoogleModels.Gemini2_5Flash
        )

        return response.first().content
    }

    /**
     * 問答功能
     * 基於文件內容回答用戶提出的特定問題
     * @param filePath 作為知識來源的文件路徑
     * @param question 用戶的具體問題
     * @return 基於文件內容的 AI 回答，如無相關資訊會明確說明
     */
    suspend fun askQuestion(filePath: String, question: String): String {
        val response = client.execute(
            prompt = prompt {
                user {
                    text("請根據這份文件回答以下問題：")
                    text(question)
                    text("請確保回答基於文件內容，如果文件中沒有相關資訊，請明確說明。")
                    attachments {
                        binaryFile(Path(filePath), getMimeType(filePath))
                    }
                }
            },
            model = GoogleModels.Gemini2_5Flash
        )

        return response.first().content
    }
}

suspend fun main() {

    // 初始化 AI 客戶端
    val client = GoogleLLMClient(ApiKeyManager.googleApiKey!!)

    // 建立文件分析器
    val documentSummarizer = DocumentSummarizer(client)

    // 文件路徑範例
    val pdfPath = "/Users/cash/Downloads/file.pdf"
    val txtPath = "/Users/cash/Downloads/file.txt"
    val mdPath = "/Users/cash/Downloads/file.md"

    try {
        // 產生 PDF 文件摘要
        println("=== PDF 文件摘要 ===")
        val pdfSummary = documentSummarizer.summarizeDocument(pdfPath)
        println(pdfSummary)

        // 提取文字文件關鍵要點
        println("\n=== 文字文件關鍵要點 ===")
        val txtKeyPoints = documentSummarizer.extractKeyPoints(txtPath)
        println(txtKeyPoints)

        // 分析 Markdown 文件結構
        println("\n=== Markdown 文件結構分析 ===")
        val mdStructure = documentSummarizer.analyzeStructure(mdPath)
        println(mdStructure)

        // 針對文件提問
        println("\n=== 問答 ===")
        val answer = documentSummarizer.askQuestion(
            pdfPath,
            "這份報告的主要建議是什麼？"
        )
        println(answer)

    } catch (e: Exception) {
        println("處理文件時發生錯誤：${e.message}")
        e.printStackTrace()
    }
}

/**
 * 批量文件處理器
 * 提供批量處理多個文件的功能，支援批量摘要產生和錯誤處理
 */
class BatchDocumentProcessor(private val summarizer: DocumentSummarizer) {

    /**
     * 批量產生文件摘要
     * 依序處理多個文件並產生摘要，包含錯誤處理和 API 頻率限制控制
     * @param filePaths 要處理的文件路徑列表
     * @return 以檔案名稱為鍵、摘要內容為值的對應表，處理失敗的文件會包含錯誤訊息
     */
    suspend fun batchSummarize(filePaths: List<String>): Map<String, String> {
        val results = mutableMapOf<String, String>()

        filePaths.forEach { filePath ->
            try {
                val fileName = File(filePath).name
                println("正在處理：$fileName")

                val summary = summarizer.summarizeDocument(filePath)
                results[fileName] = summary

                println("✓ $fileName 處理完成")

                // 避免 API 頻率限制
                delay(3000)

            } catch (e: Exception) {
                println("✗ 處理 $filePath 時發生錯誤：${e.message}")
                results[filePath] = "處理失敗：${e.message}"
            }
        }

        return results
    }
}