AI Assistant (Multiple Documents, ViewPager)

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.content.Intent
12
import android.graphics.RectF
13
import android.os.Bundle
14
import androidx.appcompat.app.AppCompatActivity
15
import androidx.appcompat.widget.Toolbar
16
import androidx.fragment.app.Fragment
17
import androidx.viewpager2.adapter.FragmentStateAdapter
18
import androidx.viewpager2.widget.ViewPager2
19
import com.google.android.material.tabs.TabLayoutMediator
20
import com.pspdfkit.ai.showAiAssistant
21
import com.pspdfkit.catalog.R
22
import com.pspdfkit.catalog.SdkExample
23
import com.pspdfkit.catalog.SdkExample.Companion.WELCOME_DOC
24
import com.pspdfkit.catalog.examples.kotlin.AiAssistantComposeActivity.Companion.PREFERENCES_NAME
25
import com.pspdfkit.catalog.examples.kotlin.AiAssistantComposeActivity.Companion.PREF_AI_IP_ADDRESS
26
import com.pspdfkit.catalog.examples.kotlin.AiAssistantViewPagerActivity.Companion.EXTRA_CONFIGURATION
27
import com.pspdfkit.catalog.utils.JwtGenerator
28
import com.pspdfkit.configuration.PdfConfiguration
29
import com.pspdfkit.configuration.activity.PdfActivityConfiguration
30
import com.pspdfkit.document.providers.AssetDataProvider
31
import com.pspdfkit.document.providers.getDataProviderFromDocumentSource
32
import com.pspdfkit.ui.DocumentDescriptor
33
import com.pspdfkit.ui.PdfFragment
34
import com.pspdfkit.utils.getSupportParcelableExtra
35
import io.nutrient.domain.ai.AiAssistant
36
import io.nutrient.domain.ai.AiAssistantProvider
37

38
/**
39
 * Shows how to implement AI Assistant chat supporting multiple documents analysis and interaction inside a ViewPager.
40
 */
41
class AiAssistantViewPagerExample(context: Context) :
42
    SdkExample(
43
        context,
44
        R.string.aiAssistantViewPagerExampleTitle,
45
        R.string.aiAssistantViewPagerExampleDescription,
46
    ) {
47
    override val launchRequirements = setOf(SdkExample.LaunchRequirement.AI_ASSISTANT_SERVER)
48

49
    override fun launchExample(context: Context, configuration: PdfActivityConfiguration.Builder) {
50
        val intent = Intent(context, AiAssistantViewPagerActivity::class.java)
51
        configuration.setAiAssistantEnabled(true)
52
        intent.putExtra(
53
            EXTRA_CONFIGURATION,
54
            configuration.build().configuration,
55
        )
56

57
        context.startActivity(intent)
58
    }
59
}
60

61
class AiAssistantViewPagerActivity :
62
    AppCompatActivity(),
63
    AiAssistantProvider {
64
    private lateinit var pager: ViewPager2
65
    private lateinit var fragmentAdapter: ViewPagerPdfFragmentAdapter
66
    private var selectedPage = 0
67
    private val sessionId = AiAssistantViewPagerActivity::class.java.simpleName
68
    var ipAddressValue = ""
69

70
    override fun onCreate(savedInstanceState: Bundle?) {
71
        super.onCreate(savedInstanceState)
72
        setContentView(R.layout.activity_view_pager_ai_assistant)
73
        findViewById<Toolbar>(R.id.main_toolbar).let {
74
            it.setOnMenuItemClickListener { _ ->
75
                showAiAssistant(this@AiAssistantViewPagerActivity)
76
                true
77
            }
78
        }
79
        val preferences = getSharedPreferences(PREFERENCES_NAME, MODE_PRIVATE)
80
        ipAddressValue = preferences.getString(PREF_AI_IP_ADDRESS, "") ?: ""
81
        fragmentAdapter = ViewPagerPdfFragmentAdapter(this@AiAssistantViewPagerActivity, documentDescriptors)
82
        pager =
83
            findViewById<ViewPager2>(R.id.view_pager).apply {
84
                this.adapter = fragmentAdapter
85
                isUserInputEnabled = false
86
                registerOnPageChangeCallback(
87
                    object : ViewPager2.OnPageChangeCallback() {
88
                        override fun onPageSelected(position: Int) {
89
                            super.onPageSelected(position)
90
                            selectedPage = position // Save the selected page.
91
                        }
92

93
                        override fun onPageScrollStateChanged(state: Int) {
94
                            super.onPageScrollStateChanged(state)
95
                            // Refresh the fragment when the user stops scrolling.
96
                            if (state == 0) {
97
                                supportFragmentManager.findFragmentByTag("f$selectedPage")?.let {
98
                                    if (it is PdfFragment) it.refreshPages()
99
                                }
100
                            }
101
                        }
102
                    },
103
                )
104
            }
105

106
        TabLayoutMediator(findViewById(R.id.tab_layout), pager) { tab, position ->
107
            tab.text = "Tab ${position + 1}"
108
        }.attach()
109
    }
110

111
    companion object {
112
        val assetFiles = listOf(WELCOME_DOC, "Scientific-paper.pdf", "Teacher.pdf", "The-Cosmic-Context-for-Life.pdf")
113
        internal const val EXTRA_CONFIGURATION = "Nutrient.AIAssistantViewPagerActivity.configuration"
114
    }
115

116
    var assistant: AiAssistant? = null
117

118
    override fun getAiAssistant(): AiAssistant = assistant ?: initialiseAiAssistant().also {
119
        assistant = it
120
    }
121

122
    val documentDescriptors =
123
        assetFiles.map {
124
            DocumentDescriptor.fromDataProviders(listOf(AssetDataProvider(it)), listOf(), listOf())
125
        }
126

127
    fun initialiseAiAssistant(): AiAssistant = com.pspdfkit.ai.createAiAssistant(
128
        context = this,
129
        documentsDescriptors = documentDescriptors,
130
        serverUrl = "http://$ipAddressValue:4000",
131
        sessionId = sessionId,
132
        jwtToken = { documentIds ->
133
            JwtGenerator.generateJwtToken(
134
                this@AiAssistantViewPagerActivity,
135
                claims =
136
                mapOf(
137
                    "document_ids" to documentIds,
138
                    "session_ids" to listOf(sessionId),
139
                    "request_limit" to
140
                        mapOf(
141
                            "requests" to 160,
142
                            "time_period_s" to 1000 * 60 * 10,
143
                        ),
144
                ),
145
            )
146
        },
147
    )
148

149
    @OptIn(ExperimentalStdlibApi::class)
150
    override fun navigateTo(documentRect: List<RectF>, pageIndex: Int, documentIndex: Int) {
151
        val fragment = fragmentAdapter.fragments[documentIndex]
152
        pager.setCurrentItem(documentIndex, true)
153
        fragment.highlight(this@AiAssistantViewPagerActivity, documentRect, pageIndex)
154
    }
155
}
156

157
/** Minimal [FragmentStateAdapter] to provide [PdfFragment] for each page. */
158
class ViewPagerPdfFragmentAdapter(val activity: AiAssistantViewPagerActivity, descriptors: List<DocumentDescriptor>) :
159
    FragmentStateAdapter(activity) {
160
    val configuration =
161
        activity.intent.getSupportParcelableExtra(EXTRA_CONFIGURATION, PdfConfiguration::class.java)
162
            ?: throw IllegalStateException("Activity Intent was missing configuration extra!")
163

164
    override fun getItemCount(): Int = fragments.size
165

166
    val fragments =
167
        descriptors.map {
168
            val dataProvider = it.documentSource.getDataProviderFromDocumentSource()
169
            PdfFragment.newInstance(dataProvider, null, configuration)
170
        }
171

172
    override fun createFragment(position: Int): Fragment = fragments[position]
173
}