AI Assistant Multiple Documents: DocumentCoordinator

1
/*
2
 *   Copyright © 2025-2026 PSPDFKit GmbH. All rights reserved.
3
 *
4
 *   The PSPDFKit Sample applications are licensed with a modified BSD license.
5
 *   Please see License for details. This notice may not be removed from this file.
6
 */
7

8
package com.pspdfkit.catalog.examples.kotlin
9

10
import android.content.Context
11
import android.graphics.RectF
12
import android.net.Uri
13
import android.os.Bundle
14
import com.pspdfkit.catalog.R
15
import com.pspdfkit.catalog.SdkExample
16
import com.pspdfkit.catalog.SdkExample.Companion.WELCOME_DOC
17
import com.pspdfkit.catalog.examples.kotlin.AiAssistantComposeActivity.Companion.PREFERENCES_NAME
18
import com.pspdfkit.catalog.examples.kotlin.AiAssistantComposeActivity.Companion.PREF_AI_IP_ADDRESS
19
import com.pspdfkit.catalog.tasks.ExtractAssetTask.extractAsync
20
import com.pspdfkit.catalog.utils.JwtGenerator
21
import com.pspdfkit.configuration.activity.PdfActivityConfiguration
22
import com.pspdfkit.ui.DocumentDescriptor
23
import com.pspdfkit.ui.PdfActivity
24
import com.pspdfkit.ui.PdfActivityIntentBuilder
25
import io.nutrient.domain.ai.AiAssistant
26
import io.nutrient.domain.ai.AiAssistantProvider
27
import io.reactivex.rxjava3.android.schedulers.AndroidSchedulers
28
import io.reactivex.rxjava3.core.Observable
29
import kotlinx.coroutines.CoroutineScope
30
import kotlinx.coroutines.Dispatchers
31
import kotlinx.coroutines.launch
32

33
/**
34
 * Shows how to implement AI Assistant chat supporting multiple tabbed documents analysis and interaction.
35
 */
36
class AiAssistantDocumentTabsExample(context: Context) : SdkExample(context, R.string.aiAssistantTabbedDocumentExampleTitle, R.string.aiAssistantTabbedDocumentExampleDescription) {
37

38
    override fun launchExample(context: Context, configuration: PdfActivityConfiguration.Builder) {
39
        configuration.setAiAssistantEnabled(true)
40

41
        Observable.fromIterable(AiAssistantDocumentTabsActivity.assetFiles)
42
            .flatMapSingle { assetName -> extractAsync(assetName, assetName, context, false, null) }
43
            .map { file ->
44
                DocumentDescriptor.fromUri(Uri.fromFile(file))
45
            }
46
            .toList()
47
            .observeOn(AndroidSchedulers.mainThread())
48
            .doOnSuccess { documentDescriptors ->
49
                launchExampleActivity(context, documentDescriptors, configuration)
50
            }.subscribe()
51
    }
52

53
    private fun launchExampleActivity(context: Context, documentDescriptors: List<DocumentDescriptor>, configuration: PdfActivityConfiguration.Builder) {
54
        val intentBuilder = if (documentDescriptors.isEmpty()) {
55
            PdfActivityIntentBuilder.emptyActivity(context)
56
        } else {
57
            PdfActivityIntentBuilder.fromDocumentDescriptor(context, *documentDescriptors.toTypedArray())
58
        }
59
        intentBuilder
60
            .configuration(configuration.build())
61
            .activityClass(AiAssistantDocumentTabsActivity::class.java)
62

63
        context.startActivity(intentBuilder.build())
64
    }
65
}
66

67
class AiAssistantDocumentTabsActivity : PdfActivity(), AiAssistantProvider {
68

69
    var ipAddressValue = ""
70
    private val sessionId = AiAssistantDocumentTabsActivity::class.java.simpleName
71

72
    companion object {
73
        val assetFiles = listOf("Classbook.pdf", WELCOME_DOC, "Aviation.pdf", "Annotations.pdf")
74
    }
75

76
    var assistant: AiAssistant? = null
77

78
    override fun onCreate(savedInstanceState: Bundle?) {
79
        super.onCreate(savedInstanceState)
80
        val preferences = getSharedPreferences(PREFERENCES_NAME, MODE_PRIVATE)
81
        ipAddressValue = preferences.getString(PREF_AI_IP_ADDRESS, "") ?: ""
82
        assistant = getAiAssistant()
83
        CoroutineScope(Dispatchers.IO).launch {
84
            // ingesting the documents as soon as the activity is created
85
            assistant?.initialize(false)
86
        }
87
    }
88

89
    fun getAiAssistantInstance() = com.pspdfkit.ai.createAiAssistant(
90
        context = this,
91
        documentsDescriptors = documentCoordinator.documents,
92
        serverUrl = "http://$ipAddressValue:4000",
93
        sessionId = sessionId,
94
        jwtToken = { documentIds ->
95
            JwtGenerator.generateJwtToken(
96
                this@AiAssistantDocumentTabsActivity,
97
                claims = mapOf(
98
                    "document_ids" to documentIds,
99
                    "session_ids" to listOf(sessionId),
100
                    "request_limit" to mapOf(
101
                        "requests" to 160,
102
                        "time_period_s" to 1000 * 60 * 10
103
                    )
104
                )
105
            )
106
        }
107
    )
108

109
    override fun getAiAssistant(): AiAssistant {
110
        return assistant ?: getAiAssistantInstance().also {
111
            assistant = it
112
        }
113
    }
114

115
    override fun navigateTo(
116
        documentRect: List<RectF>,
117
        pageIndex: Int,
118
        documentIndex: Int
119
    ) {
120
        val descriptor = documentCoordinator.getDocuments()[documentIndex]
121
        documentCoordinator.setVisibleDocument(descriptor)
122
        pdfFragment?.highlight(this@AiAssistantDocumentTabsActivity, documentRect, pageIndex)
123
    }
124
}