Merge pull request #4 from Realtime-Coding/feature/ai-gemini

Feature/ai gemini

Author: v-shahzadahmad

composeApp/build.gradle.kts

@@ -4,12 +4,15 @@ import org.jetbrains.kotlin.gradle.ExperimentalKotlinGradlePluginApi
 import org.jetbrains.kotlin.gradle.dsl.JvmTarget
 import org.jetbrains.kotlin.gradle.targets.js.dsl.ExperimentalWasmDsl
 import org.jetbrains.kotlin.gradle.targets.js.webpack.KotlinWebpackConfig
+import com.codingfeline.buildkonfig.compiler.FieldSpec
+import java.util.Properties
 
 plugins {
     alias(libs.plugins.kotlinMultiplatform)
     alias(libs.plugins.androidApplication)
     alias(libs.plugins.jetbrainsCompose)
     alias(libs.plugins.compose.compiler)
+    alias(libs.plugins.buildkonfig)
 }
 
 kotlin {
@@ -29,11 +32,12 @@ kotlin {
         }
         binaries.executable()
     }
-    
+
     androidTarget {
-        @OptIn(ExperimentalKotlinGradlePluginApi::class)
-        compilerOptions {
-            jvmTarget.set(JvmTarget.JVM_17)
+        compilations.all {
+            kotlinOptions {
+                jvmTarget = "17"
+            }
         }
     }
 
@@ -67,7 +71,12 @@ kotlin {
             implementation(libs.coil.network.ktor)
             implementation(libs.voyager.navigator)
             implementation(libs.voyager.tab.navigator)
+            implementation(libs.voyager.screenmodel)
             implementation(libs.lifecycle.viewmodel.compose)
+            implementation(libs.markdown.renderer)
+            api(libs.compose.window.size)
+            api(libs.generativeai)
+            implementation(libs.filekit.compose)
         }
         desktopMain.dependencies {
             implementation(compose.desktop.currentOs)
@@ -147,3 +156,25 @@ compose.desktop {
         }
     }
 }
+
+buildkonfig {
+    packageName = "com.travel.buddy"
+
+    val localPropsFile = rootProject.file("local.properties")
+    val localProperties = Properties()
+    if (localPropsFile.exists()) {
+        runCatching {
+            localProperties.load(localPropsFile.inputStream())
+        }.getOrElse {
+            it.printStackTrace()
+        }
+    }
+    defaultConfigs {
+        buildConfigField(
+            FieldSpec.Type.STRING,
+            "GEMINI_API_KEY",
+            localProperties["gemini_api_key"]?.toString() ?: ""
+        )
+    }
+
+}

composeApp/src/androidMain/kotlin/App.android.kt

@@ -1,5 +1,8 @@
 import android.content.Intent
+import android.graphics.BitmapFactory
 import android.net.Uri
+import androidx.compose.ui.graphics.ImageBitmap
+import androidx.compose.ui.graphics.asImageBitmap
 import com.travel.buddy.MuseumApp
 
 internal actual fun openUrl(url: String?) {
@@ -10,4 +13,8 @@ internal actual fun openUrl(url: String?) {
         addFlags(Intent.FLAG_ACTIVITY_NEW_TASK)
     }
     MuseumApp.INSTANCE.startActivity(intent)
+}
+
+actual fun ByteArray.toComposeImageBitmap(): ImageBitmap {
+    return BitmapFactory.decodeByteArray(this, 0, size).asImageBitmap()
 }

composeApp/src/commonMain/composeResources/drawable/chat_bot.png

@@ -2,7 +2,7 @@
 <resources>
     <string name="home_tab">Home</string>
     <string name="fav_tab">Favourite</string>
-    <string name="profile_tab">Profile</string>
+    <string name="profile_tab">Gemini</string>
     <string name="cart_tab">Cart</string>
     <string name="ratting">RATING</string>
     <string name="type">TYPE</string>
@@ -18,4 +18,5 @@
     <string name="category">Category</string>
     <string name="all">All</string>
     <string name="num_of_people">Number Of People</string>
+    <string name="gemini">Gemini ChatBot</string>
 </resources>

composeApp/src/commonMain/composeResources/values/string.xml

@@ -4,11 +4,10 @@ import androidx.compose.material3.Scaffold
 import androidx.compose.runtime.*
 import androidx.compose.ui.Alignment
 import androidx.compose.ui.Modifier
-import androidx.lifecycle.viewmodel.compose.viewModel
+import androidx.compose.ui.graphics.ImageBitmap
 import cafe.adriel.voyager.navigator.LocalNavigator
 import cafe.adriel.voyager.navigator.currentOrThrow
 import cafe.adriel.voyager.navigator.tab.CurrentTab
-import cafe.adriel.voyager.navigator.tab.LocalTabNavigator
 import cafe.adriel.voyager.navigator.tab.TabNavigator
 import coil3.ImageLoader
 import coil3.PlatformContext
@@ -18,27 +17,27 @@ import coil3.memory.MemoryCache
 import coil3.request.CachePolicy
 import coil3.request.crossfade
 import coil3.util.DebugLogger
+import di.HomeScreenModelProvider
 import okio.FileSystem
 import theme.TravelAppTheme
 import ui.component.BottomMenuBar
 import ui.component.tabs
 import ui.screen.CartTab
 import ui.screen.FavoriteTab
 import ui.screen.HomeTab
-import ui.screen.ProfileTab
-import ui.viewmodel.HomeViewModel
+import ui.screen.GeminiTab
 import util.AnimateVisibility
 
 @Composable
-internal fun App(
-    viewModel: HomeViewModel = viewModel { HomeViewModel() }
-) {
+internal fun App() {
     TravelAppTheme {
 
         setSingletonImageLoaderFactory { context ->
             getAsyncImageLoader(context)
         }
 
+        val viewModel = HomeScreenModelProvider.homeScreenModel
+
         val bottomNavBarVisibility by viewModel.bottomNavBarVisible.collectAsState()
 
         TabNavigator(HomeTab) {
@@ -57,7 +56,7 @@ internal fun App(
                                 HomeTab -> LocalNavigator.currentOrThrow.push(HomeTab)
                                 FavoriteTab -> LocalNavigator.currentOrThrow.push(FavoriteTab)
                                 CartTab -> LocalNavigator.currentOrThrow.push(CartTab)
-                                ProfileTab -> LocalNavigator.currentOrThrow.push(ProfileTab)
+                                GeminiTab -> LocalNavigator.currentOrThrow.push(GeminiTab)
                             }
                         }
                     }
@@ -81,4 +80,6 @@ fun newDiskCache(): DiskCache {
     return DiskCache.Builder().directory(FileSystem.SYSTEM_TEMPORARY_DIRECTORY / "image_cache")
         .maxSizeBytes(1024L * 1024 * 1024) // 512MB
         .build()
-}
+}
+
+expect fun ByteArray.toComposeImageBitmap(): ImageBitmap

composeApp/src/commonMain/kotlin/App.kt

@@ -0,0 +1,47 @@
+package data
+
+import com.travel.buddy.BuildKonfig
+import dev.shreyaspatil.ai.client.generativeai.GenerativeModel
+import dev.shreyaspatil.ai.client.generativeai.type.GenerateContentResponse
+import dev.shreyaspatil.ai.client.generativeai.type.PlatformImage
+import dev.shreyaspatil.ai.client.generativeai.type.content
+import kotlinx.coroutines.flow.Flow
+import kotlin.io.encoding.ExperimentalEncodingApi
+
+class GeminiApi {
+    companion object {
+        const val PROMPT_GENERATE_UI = "Act as an Android app developer. " +
+                "For the image provided, use Jetpack Compose to build the screen so that " +
+                "the Compose Preview is as close to this image as possible. Also make sure " +
+                "to include imports and use Material3. Only give code part without any extra " +
+                "text or description neither at start or end, your response should contain " +
+                "only code without any explanation."
+    }
+
+
+    private val apiKey = BuildKonfig.GEMINI_API_KEY
+
+
+    val generativeVisionModel = GenerativeModel(
+        modelName = "gemini-1.5-flash",
+        apiKey = apiKey
+    )
+
+    val generativeModel = GenerativeModel(
+        modelName = "gemini-pro",
+        apiKey = apiKey
+    )
+
+    fun generateContent(prompt: String): Flow<GenerateContentResponse> {
+        return generativeModel.generateContentStream(prompt)
+    }
+
+    @OptIn(ExperimentalEncodingApi::class)
+    fun generateContent(prompt: String, imageData: ByteArray): Flow<GenerateContentResponse> {
+        val content = content {
+            image(PlatformImage(imageData))
+            text(prompt)
+        }
+        return generativeVisionModel.generateContentStream(content)
+    }
+}

composeApp/src/commonMain/kotlin/data/GeminiApi.kt

@@ -1,5 +1,7 @@
 package di
 
+import ui.viewmodel.HomeScreenModel
+
 //import org.koin.core.context.startKoin
 //import org.koin.dsl.module
 //
@@ -14,3 +16,9 @@ package di
 //        )
 //    }
 //}
+
+object HomeScreenModelProvider {
+    val homeScreenModel: HomeScreenModel by lazy {
+        HomeScreenModel()
+    }
+}

composeApp/src/commonMain/kotlin/di/Koin.kt

@@ -0,0 +1,3 @@
+package model
+
+data class ChatMessage(val text: String, val isSender: Boolean)

composeApp/src/commonMain/kotlin/model/ChatMessage.kt

@@ -18,4 +18,7 @@ val ReviewBodyBg = Color(0xFFF6F8FA)
 val Yellow = Color(0xFFF8E545)
 val Red = Color(0xFFFF2828)
 val CategoryChipBg = Color(0xFF4D5652)
-val BorderColor = Color(0xFFAFAFAF)
+val BorderColor = Color(0xFFDDDDDD) // Light grey
+val LinkColor = Color(0xFF0000FF) // Blue
+val CodeBackground = Color(0xFFf9f9f9)
+

composeApp/src/commonMain/kotlin/theme/Color.kt

@@ -10,7 +10,6 @@ import androidx.compose.foundation.layout.padding
 import androidx.compose.foundation.layout.size
 import androidx.compose.foundation.layout.width
 import androidx.compose.foundation.layout.wrapContentSize
-import androidx.compose.foundation.lazy.LazyItemScope
 import androidx.compose.foundation.lazy.LazyRow
 import androidx.compose.foundation.lazy.items
 import androidx.compose.foundation.shape.RoundedCornerShape
@@ -40,7 +39,7 @@ import theme.SecondTextColor
 import ui.screen.CartTab
 import ui.screen.FavoriteTab
 import ui.screen.HomeTab
-import ui.screen.ProfileTab
+import ui.screen.GeminiTab
 
 interface Tabx: Tab {
     fun defaultTitle(): StringResource
@@ -51,7 +50,7 @@ val tabs = arrayListOf<Tabx>().apply {
     add(HomeTab)
     add(FavoriteTab)
     add(CartTab)
-    add(ProfileTab)
+    add(GeminiTab)
 }
 
 @Composable