chat: add gemini and perplexity (fixes #3360) (#3363)

Co-authored-by: dogi <[email protected]>

app/build.gradle

@@ -9,8 +9,8 @@ android {
         applicationId "org.ole.planet.myplanet"
         minSdkVersion 21
         targetSdkVersion 34
-        versionCode 1475
-        versionName "0.14.75"
+        versionCode 1476
+        versionName "0.14.76"
         ndkVersion '21.3.6528147'
         testInstrumentationRunner "androidx.test.runner.AndroidJUnitRunner"
         vectorDrawables.useSupportLibrary = true

app/src/main/java/org/ole/planet/myplanet/datamanager/ApiInterface.kt

@@ -5,7 +5,7 @@ import okhttp3.RequestBody
 import okhttp3.ResponseBody
 import org.ole.planet.myplanet.model.DocumentResponse
 import org.ole.planet.myplanet.model.MyPlanet
-import org.ole.planet.myplanet.ui.chat.ChatModel
+import org.ole.planet.myplanet.model.ChatModel
 import retrofit2.Call
 import retrofit2.http.Body
 import retrofit2.http.GET
@@ -56,4 +56,7 @@ interface ApiInterface {
 
     @POST
     fun chatGpt(@Url url: String?, @Body requestBody: RequestBody?): Call<ChatModel>
+
+    @GET
+    fun checkAiProviders(@Url url: String?): Call<ResponseBody>
 }

app/src/main/java/org/ole/planet/myplanet/model/ChatRequestModel.kt

@@ -0,0 +1,32 @@
+package org.ole.planet.myplanet.model
+
+import com.google.gson.annotations.SerializedName
+
+data class ChatRequestModel(
+    @SerializedName("data") val data: ContentData,
+    @SerializedName("save") val save: Boolean
+)
+
+data class ContentData(
+    @SerializedName("user") var user: String,
+    @SerializedName("content") val content: String,
+    @SerializedName("aiProvider") val aiProvider: AiProvider
+)
+
+data class ContinueChatModel(
+    @SerializedName("data") val data: Data,
+    @SerializedName("save") val save: Boolean
+)
+
+data class Data(
+    @SerializedName("user") var user: String,
+    @SerializedName("content") var content: String,
+    @SerializedName("aiProvider") val aiProvider: AiProvider,
+    @SerializedName("_id") var Id: String,
+    @SerializedName("_rev") var rev: String
+)
+
+data class AiProvider (
+    @SerializedName("name") val name: String,
+    @SerializedName("model") val model: String
+)

app/src/main/java/org/ole/planet/myplanet/ui/chat/ChatResponseModel.kt → app/src/main/java/org/ole/planet/myplanet/model/ChatResponseModel.kt

@@ -1,4 +1,4 @@
-package org.ole.planet.myplanet.ui.chat
+package org.ole.planet.myplanet.model
 
 import com.google.gson.annotations.SerializedName
 

app/src/main/java/org/ole/planet/myplanet/ui/chat/ChatViewModel.kt → app/src/main/java/org/ole/planet/myplanet/model/ChatViewModel.kt

@@ -1,4 +1,4 @@
-package org.ole.planet.myplanet.ui.chat
+package org.ole.planet.myplanet.model
 
 import androidx.lifecycle.LiveData
 import androidx.lifecycle.MutableLiveData

app/src/main/java/org/ole/planet/myplanet/ui/chat/AiProviderModel.kt

@@ -0,0 +1,7 @@
+package org.ole.planet.myplanet.ui.chat
+
+data class AiProvidersResponse(
+    val openai: Boolean,
+    val perplexity: Boolean,
+    val gemini: Boolean
+)

app/src/main/java/org/ole/planet/myplanet/ui/chat/ChatDetailFragment.kt

@@ -1,30 +1,21 @@
 package org.ole.planet.myplanet.ui.chat
 
 import android.os.Bundle
-import android.text.Editable
-import android.text.TextUtils
-import android.text.TextWatcher
-import android.view.KeyEvent
-import android.view.LayoutInflater
-import android.view.View
-import android.view.ViewGroup
+import android.text.*
+import android.view.*
+import androidx.core.content.ContextCompat
 import androidx.fragment.app.Fragment
 import androidx.lifecycle.ViewModelProvider
-import androidx.recyclerview.widget.LinearLayoutManager
-import androidx.recyclerview.widget.RecyclerView
-import com.google.gson.Gson
-import com.google.gson.JsonArray
-import com.google.gson.JsonObject
+import androidx.recyclerview.widget.*
+import com.google.gson.*
 import io.realm.Realm
-import okhttp3.MediaType
-import okhttp3.RequestBody
+import okhttp3.*
 import org.ole.planet.myplanet.R
 import org.ole.planet.myplanet.databinding.FragmentChatDetailBinding
-import org.ole.planet.myplanet.datamanager.ApiClient
-import org.ole.planet.myplanet.datamanager.ApiInterface
-import org.ole.planet.myplanet.datamanager.DatabaseService
-import org.ole.planet.myplanet.model.RealmChatHistory
+import org.ole.planet.myplanet.datamanager.*
+import org.ole.planet.myplanet.model.*
 import org.ole.planet.myplanet.model.RealmChatHistory.Companion.addConversationToChatHistory
+import org.ole.planet.myplanet.service.UserProfileDbHandler
 import org.ole.planet.myplanet.utilities.Utilities
 import retrofit2.Call
 import retrofit2.Callback
@@ -36,6 +27,8 @@ class ChatDetailFragment : Fragment() {
     private lateinit var sharedViewModel: ChatViewModel
     private var _id: String = ""
     private var _rev: String = ""
+    private var aiName: String = ""
+    private var aiModel: String = ""
     private lateinit var mRealm: Realm
 
     override fun onCreate(savedInstanceState: Bundle?) {
@@ -50,34 +43,35 @@ class ChatDetailFragment : Fragment() {
     override fun onViewCreated(view: View, savedInstanceState: Bundle?) {
         super.onViewCreated(view, savedInstanceState)
         mRealm = DatabaseService(requireContext()).realmInstance
-
+        val user = UserProfileDbHandler(requireContext()).userModel
         mAdapter = ChatAdapter(ArrayList(), requireContext(), fragmentChatDetailBinding.recyclerGchat)
         fragmentChatDetailBinding.recyclerGchat.adapter = mAdapter
         val layoutManager: RecyclerView.LayoutManager = LinearLayoutManager(requireContext())
         fragmentChatDetailBinding.recyclerGchat.layoutManager = layoutManager
         fragmentChatDetailBinding.recyclerGchat.isNestedScrollingEnabled = true
         fragmentChatDetailBinding.recyclerGchat.setHasFixedSize(true)
-
+        checkAiProviders()
         if (mAdapter.itemCount > 0) {
             fragmentChatDetailBinding.recyclerGchat.scrollToPosition(mAdapter.itemCount - 1)
             fragmentChatDetailBinding.recyclerGchat.smoothScrollToPosition(mAdapter.itemCount - 1)
         }
 
         fragmentChatDetailBinding.buttonGchatSend.setOnClickListener {
+            val aiProvider = AiProvider(name = aiName, model = aiModel)
             fragmentChatDetailBinding.textGchatIndicator.visibility = View.GONE
-            if (TextUtils.isEmpty(fragmentChatDetailBinding.editGchatMessage.text.toString().trim())) {
+            if (TextUtils.isEmpty("${fragmentChatDetailBinding.editGchatMessage.text}".trim())) {
                 fragmentChatDetailBinding.textGchatIndicator.visibility = View.VISIBLE
-                fragmentChatDetailBinding.textGchatIndicator.text = "Kindly enter message"
+                fragmentChatDetailBinding.textGchatIndicator.text = getString(R.string.kindly_enter_message)
             } else {
                 val message = "${fragmentChatDetailBinding.editGchatMessage.text}".replace("\n", " ")
                 mAdapter.addQuery(message)
                 if (_id != "") {
-                    val continueChatData = ContinueChatModel(data = Data(message, _id, _rev), save = true)
+                    val continueChatData = ContinueChatModel(data = Data("${user?.name}", message, aiProvider, _id, _rev), save = true)
                     val jsonContent = Gson().toJson(continueChatData)
                     val requestBody = RequestBody.create(MediaType.parse("application/json"), jsonContent)
                     continueChatRequest(requestBody, _id, message)
                 } else {
-                    val chatData = ChatRequestModel(data = ContentData(message), save = true)
+                    val chatData = ChatRequestModel(data = ContentData("${user?.name}", message, aiProvider), save = true)
                     val jsonContent = Gson().toJson(chatData)
                     val requestBody = RequestBody.create(MediaType.parse("application/json"), jsonContent)
                     makePostRequest(requestBody, message)
@@ -90,11 +84,9 @@ class ChatDetailFragment : Fragment() {
         fragmentChatDetailBinding.editGchatMessage.setOnKeyListener { _, _, event ->
             if (event.action == KeyEvent.ACTION_DOWN) {
                 if (event.keyCode == KeyEvent.KEYCODE_ENTER && event.isShiftPressed) {
-                    // Insert a new line when Shift + Enter is pressed
                     fragmentChatDetailBinding.editGchatMessage.append("\n")
                     return@setOnKeyListener true
                 } else if (event.keyCode == KeyEvent.KEYCODE_ENTER) {
-                    // Call the functionality inside buttonGchatSend click listener
                     fragmentChatDetailBinding.buttonGchatSend.performClick()
                     return@setOnKeyListener true
                 }
@@ -143,6 +135,115 @@ class ChatDetailFragment : Fragment() {
         }
     }
 
+    private fun checkAiProviders() {
+        val apiInterface = ApiClient.client?.create(ApiInterface::class.java)
+        apiInterface?.checkAiProviders("${Utilities.hostUrl}/checkproviders")?.enqueue(object : Callback<ResponseBody> {
+            override fun onResponse(call: Call<ResponseBody>, response: Response<ResponseBody>) {
+                if (response.isSuccessful) {
+                    response.body()?.let { responseBody ->
+                        val gson = Gson()
+                        val aiProvidersResponse = gson.fromJson(responseBody.string(), AiProvidersResponse::class.java)
+                        if (aiProvidersResponse.openai) {
+                            fragmentChatDetailBinding.tvOpenai.visibility = View.VISIBLE
+                            fragmentChatDetailBinding.view1.visibility = View.VISIBLE
+
+                            aiName = getString(R.string.openai)
+                            aiModel = "gpt-3.5-turbo"
+                            fragmentChatDetailBinding.tvOpenai.setBackgroundColor(ContextCompat.getColor(requireContext(), R.color.colorPrimary))
+                            fragmentChatDetailBinding.tvOpenai.setTextColor(ContextCompat.getColor(requireContext(), R.color.textColorPrimary))
+
+                            fragmentChatDetailBinding.tvOpenai.setOnClickListener {
+                                fragmentChatDetailBinding.tvOpenai.setBackgroundColor(ContextCompat.getColor(requireContext(), R.color.colorPrimary))
+                                fragmentChatDetailBinding.tvOpenai.setTextColor(ContextCompat.getColor(requireContext(), R.color.textColorPrimary))
+
+                                fragmentChatDetailBinding.tvPerplexity.setBackgroundColor(ContextCompat.getColor(requireContext(), R.color.disable_color))
+                                fragmentChatDetailBinding.tvPerplexity.setTextColor(ContextCompat.getColor(requireContext(), R.color.md_black_1000))
+
+                                fragmentChatDetailBinding.tvGemini.setBackgroundColor(ContextCompat.getColor(requireContext(), R.color.disable_color))
+                                fragmentChatDetailBinding.tvGemini.setTextColor(ContextCompat.getColor(requireContext(), R.color.md_black_1000))
+
+                                clearChatDetail()
+                                fragmentChatDetailBinding.textGchatIndicator.visibility = View.GONE
+                                aiName = getString(R.string.openai)
+                                aiModel = "gpt-3.5-turbo"
+                            }
+                        } else {
+                            fragmentChatDetailBinding.tvOpenai.visibility = View.GONE
+                            fragmentChatDetailBinding.view1.visibility = View.GONE
+                        }
+
+                        if (aiProvidersResponse.perplexity) {
+                            fragmentChatDetailBinding.tvPerplexity.visibility = View.VISIBLE
+                            fragmentChatDetailBinding.view2.visibility = View.VISIBLE
+
+                            if (!aiProvidersResponse.openai) {
+                                aiName = getString(R.string.perplexity)
+                                aiModel= "pplx-7b-online"
+                                fragmentChatDetailBinding.tvPerplexity.setBackgroundColor(ContextCompat.getColor(requireContext(), R.color.colorPrimary))
+                                fragmentChatDetailBinding.tvPerplexity.setTextColor(ContextCompat.getColor(requireContext(), R.color.textColorPrimary))
+                            }
+
+                            fragmentChatDetailBinding.tvPerplexity.setOnClickListener {
+                                fragmentChatDetailBinding.tvPerplexity.setBackgroundColor(ContextCompat.getColor(requireContext(), R.color.colorPrimary))
+                                fragmentChatDetailBinding.tvPerplexity.setTextColor(ContextCompat.getColor(requireContext(), R.color.textColorPrimary))
+
+                                fragmentChatDetailBinding.tvOpenai.setBackgroundColor(ContextCompat.getColor(requireContext(), R.color.disable_color))
+                                fragmentChatDetailBinding.tvOpenai.setTextColor(ContextCompat.getColor(requireContext(), R.color.md_black_1000))
+
+                                fragmentChatDetailBinding.tvGemini.setBackgroundColor(ContextCompat.getColor(requireContext(), R.color.disable_color))
+                                fragmentChatDetailBinding.tvGemini.setTextColor(ContextCompat.getColor(requireContext(), R.color.md_black_1000))
+
+                                clearChatDetail()
+                                fragmentChatDetailBinding.textGchatIndicator.visibility = View.GONE
+                                aiName = getString(R.string.perplexity)
+                                aiModel= "pplx-7b-online"
+                            }
+                        } else {
+                            fragmentChatDetailBinding.tvPerplexity.visibility = View.GONE
+                            fragmentChatDetailBinding.view2.visibility = View.GONE
+                        }
+
+                        if (aiProvidersResponse.gemini) {
+                            if (!aiProvidersResponse.openai && !aiProvidersResponse.perplexity) {
+                                aiName= getString(R.string.gemini)
+                                aiModel= "gemini-pro"
+                                fragmentChatDetailBinding.tvGemini.setBackgroundColor(ContextCompat.getColor(requireContext(), R.color.colorPrimary))
+                                fragmentChatDetailBinding.tvGemini.setTextColor(ContextCompat.getColor(requireContext(), R.color.textColorPrimary))
+                            }
+
+                            fragmentChatDetailBinding.tvGemini.visibility = View.VISIBLE
+
+                            fragmentChatDetailBinding.tvGemini.setOnClickListener {
+                                fragmentChatDetailBinding.tvGemini.setBackgroundColor(ContextCompat.getColor(requireContext(), R.color.colorPrimary))
+                                fragmentChatDetailBinding.tvGemini.setTextColor(ContextCompat.getColor(requireContext(), R.color.textColorPrimary))
+
+                                fragmentChatDetailBinding.tvPerplexity.setBackgroundColor(ContextCompat.getColor(requireContext(), R.color.disable_color))
+                                fragmentChatDetailBinding.tvPerplexity.setTextColor(ContextCompat.getColor(requireContext(), R.color.md_black_1000))
+
+                                fragmentChatDetailBinding.tvOpenai.setBackgroundColor(ContextCompat.getColor(requireContext(), R.color.disable_color))
+                                fragmentChatDetailBinding.tvOpenai.setTextColor(ContextCompat.getColor(requireContext(), R.color.md_black_1000))
+
+                                clearChatDetail()
+                                fragmentChatDetailBinding.textGchatIndicator.visibility = View.GONE
+                                aiName= getString(R.string.gemini)
+                                aiModel= "gemini-pro"
+                            }
+                        } else {
+                            fragmentChatDetailBinding.tvGemini.visibility = View.GONE
+                        }
+                    }
+                }
+            }
+
+            override fun onFailure(call: Call<ResponseBody>, t: Throwable) {
+                fragmentChatDetailBinding.textGchatIndicator.visibility = View.VISIBLE
+                fragmentChatDetailBinding.textGchatIndicator.text = getString(R.string.virtual_assistant_currently_not_available)
+                fragmentChatDetailBinding.editGchatMessage.isEnabled = false
+                fragmentChatDetailBinding.buttonGchatSend.isEnabled = false
+            }
+        })
+    }
+
     private fun makePostRequest(content: RequestBody, query: String) {
         fragmentChatDetailBinding.buttonGchatSend.isEnabled = false
         fragmentChatDetailBinding.editGchatMessage.isEnabled = false
@@ -198,7 +299,11 @@ class ChatDetailFragment : Fragment() {
                     }
                 } else {
                     fragmentChatDetailBinding.textGchatIndicator.visibility = View.VISIBLE
-                    fragmentChatDetailBinding.textGchatIndicator.text = getString(R.string.request_failed_please_retry)
+                    fragmentChatDetailBinding.textGchatIndicator.text = if (response.message() == "null"){
+                        getString(R.string.request_failed_please_retry)
+                    } else {
+                        response.message()
+                    }
                     val jsonObject = JsonObject()
                     jsonObject.addProperty("_rev", "")
                     jsonObject.addProperty("_id", "")

app/src/main/java/org/ole/planet/myplanet/ui/chat/ChatHistoryListFragment.kt

@@ -1,11 +1,8 @@
 package org.ole.planet.myplanet.ui.chat
 
 import android.os.Bundle
-import android.text.Editable
-import android.text.TextWatcher
-import android.view.LayoutInflater
-import android.view.View
-import android.view.ViewGroup
+import android.text.*
+import android.view.*
 import androidx.activity.OnBackPressedCallback
 import androidx.fragment.app.Fragment
 import androidx.lifecycle.ViewModelProvider
@@ -14,8 +11,7 @@ import io.realm.RealmList
 import org.ole.planet.myplanet.R
 import org.ole.planet.myplanet.databinding.FragmentChatHistoryListBinding
 import org.ole.planet.myplanet.datamanager.DatabaseService
-import org.ole.planet.myplanet.model.Conversation
-import org.ole.planet.myplanet.model.RealmChatHistory
+import org.ole.planet.myplanet.model.*
 
 class ChatHistoryListFragment : Fragment() {
     private lateinit var fragmentChatHistoryListBinding: FragmentChatHistoryListBinding