expo-background-remover 0.1.2 → 0.2.2

package/android/build.gradle

@@ -2,7 +2,7 @@ apply plugin: 'com.android.library'
 apply plugin: 'kotlin-android'
 
 group = 'expo.modules.backgroundremover'
-version = '0.1.2'
+version = '0.2.2'
 
 def expoModulesCorePlugin = new File(project(":expo-modules-core").projectDir.absolutePath, "ExpoModulesCorePlugin.gradle")
 apply from: expoModulesCorePlugin
@@ -37,7 +37,7 @@ android {
   namespace "expo.modules.backgroundremover"
   defaultConfig {
     versionCode 1
-    versionName "0.1.2"
+    versionName "0.2.2"
   }
   lintOptions {
     abortOnError false
@@ -65,5 +65,6 @@ dependencies {
     implementation "androidx.lifecycle:lifecycle-runtime-ktx:2.7.0"
 
     implementation("com.google.android.gms:play-services-mlkit-subject-segmentation:16.0.0-beta1")
+    implementation("androidx.exifinterface:exifinterface:1.4.2")
 }
 

package/android/src/main/java/expo/modules/backgroundremover/BackgroundRemoverProcessor.kt

@@ -3,10 +3,13 @@ package expo.modules.backgroundremover
 import android.content.Context
 import android.graphics.*
 import android.net.Uri
+import androidx.exifinterface.media.ExifInterface // Added for getRotation
 import com.google.mlkit.vision.common.InputImage
 import com.google.mlkit.vision.segmentation.subject.SubjectSegmentation
 import com.google.mlkit.vision.segmentation.subject.SubjectSegmenterOptions
+import kotlinx.coroutines.Dispatchers
 import kotlinx.coroutines.tasks.await
+import kotlinx.coroutines.withContext
 import java.io.File
 import java.io.FileOutputStream
 import java.util.UUID
@@ -17,59 +20,80 @@ class BackgroundRemoverProcessor(private val context: Context) {
 
     private val segmenter = SubjectSegmentation.getClient(
         SubjectSegmenterOptions.Builder()
-            .enableForegroundConfidenceMask() // High quality edges
+            .enableForegroundConfidenceMask()
             .build()
     )
 
-    suspend fun processImage(uriString: String): String {
-        // 1. Efficiently load and downscale to max 2048px
+    suspend fun processImage(uriString: String): String = withContext(Dispatchers.IO) {
         val bitmap = loadAndResizeBitmap(uriString, 2048)
-        val inputImage = InputImage.fromBitmap(bitmap, 0)
+        
+        // FIX 1: Defined getRotation helper
+        val inputImage = InputImage.fromBitmap(bitmap, getRotation(uriString))
 
-        // 2. Perform Segmentation
         val result = segmenter.process(inputImage).await()
         
-        // 3. Retrieve global mask (Includes People + Objects automatically)
         val maskBuffer = result.foregroundConfidenceMask 
             ?: throw Exception("Could not detect subjects")
-            
-        // 4. Create Mask Bitmap and Blend
-        val maskBitmap = createMaskFromBuffer(maskBuffer, bitmap.width, bitmap.height)
+
+        // FIX 2: Use bitmap dimensions (result does not have width/height)
+        val totalPixels = maskBuffer.remaining()
+        val maskWidth = bitmap.width
+        val maskHeight = totalPixels / maskWidth
+
+        val maskBitmap = createMaskFromBuffer(maskBuffer, maskWidth, maskHeight)
         val outputBitmap = applyMaskToBitmap(bitmap, maskBitmap)
 
-        return saveResult(outputBitmap)
+        saveResult(outputBitmap)
+    }
+
+    // Helper to handle image rotation logic
+    private fun getRotation(uriString: String): Int {
+        return try {
+            val inputStream = context.contentResolver.openInputStream(Uri.parse(uriString))
+            val exifInterface = inputStream?.use { ExifInterface(it) }
+            when (exifInterface?.getAttributeInt(ExifInterface.TAG_ORIENTATION, ExifInterface.ORIENTATION_NORMAL)) {
+                ExifInterface.ORIENTATION_ROTATE_90 -> 90
+                ExifInterface.ORIENTATION_ROTATE_180 -> 180
+                ExifInterface.ORIENTATION_ROTATE_270 -> 270
+                else -> 0
+            }
+        } catch (e: Exception) {
+            0
+        }
     }
 
     private fun loadAndResizeBitmap(uriString: String, maxDimension: Int): Bitmap {
         val uri = Uri.parse(uriString)
-        
-        // Stage A: Get dimensions only
         val options = BitmapFactory.Options().apply { inJustDecodeBounds = true }
-        context.contentResolver.openInputStream(uri)?.use { stream ->
-          BitmapFactory.decodeStream(stream, null, loadOptions)
-         }?: throw Exception("Failed to open input stream")
         
-        // Stage B: Calculate optimal inSampleSize (power of 2)
+        context.contentResolver.openInputStream(uri)?.use { 
+            BitmapFactory.decodeStream(it, null, options) 
+        } ?: throw Exception("Failed to open input stream")
+        
         var sampleSize = 1
         while ((options.outWidth / (sampleSize * 2)) >= maxDimension && 
                (options.outHeight / (sampleSize * 2)) >= maxDimension) {
             sampleSize *= 2
         }
         
-        // Stage C: Load the subsampled bitmap
         val loadOptions = BitmapFactory.Options().apply { inSampleSize = sampleSize }
-        val subsampledBitmap = context.contentResolver.openInputStream(uri).use { 
+        val subsampledBitmap = context.contentResolver.openInputStream(uri)?.use { 
             BitmapFactory.decodeStream(it, null, loadOptions) 
         } ?: throw Exception("Failed to decode image")
 
-        // Stage D: Precise scaling to exactly fit within 2048px while preserving aspect ratio
         val scale = minOf(maxDimension.toFloat() / subsampledBitmap.width, maxDimension.toFloat() / subsampledBitmap.height)
         if (scale >= 1f) return subsampledBitmap
 
-        val targetWidth = (subsampledBitmap.width * scale).toInt()
-        val targetHeight = (subsampledBitmap.height * scale).toInt()
-        
-        return Bitmap.createScaledBitmap(subsampledBitmap, targetWidth, targetHeight, true)
+        val scaledBitmap = Bitmap.createScaledBitmap(
+        subsampledBitmap,
+        (subsampledBitmap.width * scale).toInt(),
+        (subsampledBitmap.height * scale).toInt(),
+        true
+        )
+
+        subsampledBitmap.recycle()  
+
+        return scaledBitmap
     }
 
     private fun createMaskFromBuffer(buffer: java.nio.FloatBuffer, width: Int, height: Int): Bitmap {
@@ -77,7 +101,8 @@ class BackgroundRemoverProcessor(private val context: Context) {
         buffer.rewind()
         val pixels = IntArray(width * height)
         for (i in 0 until width * height) {
-            val alpha = (buffer.get() * 255).toInt()
+            // Mask transparency based on confidence (0.0 to 1.0)
+            val alpha = (buffer.get().coerceIn(0f, 1f) * 255).toInt()
             pixels[i] = Color.argb(alpha, 0, 0, 0)
         }
         mask.setPixels(pixels, 0, width, 0, 0, width, height)
@@ -101,4 +126,8 @@ class BackgroundRemoverProcessor(private val context: Context) {
         FileOutputStream(file).use { bitmap.compress(Bitmap.CompressFormat.PNG, 100, it) }
         return Uri.fromFile(file).toString()
     }
-}
+    
+    fun close() {
+        segmenter.close()
+    }
+}

package/android/src/main/java/expo/modules/backgroundremover/ExpoBackgroundRemoverModule.kt

@@ -3,6 +3,7 @@ package expo.modules.backgroundremover
 import expo.modules.kotlin.modules.Module
 import expo.modules.kotlin.modules.ModuleDefinition
 import java.net.URL
+ 
 
 class ExpoBackgroundRemoverModule : Module() {
   // Each module class must implement the definition function. The definition consists of components
@@ -28,9 +29,13 @@ class ExpoBackgroundRemoverModule : Module() {
     }
 
     AsyncFunction("removeBackgroundAsync") { imageUri: String ->
-      // Pass the internal context to the processor
-      val processor = BackgroundRemoverProcessor(appContext.reactContext!!)
-      return@AsyncFunction processor.processImage(imageUri)
+       val processor = BackgroundRemoverProcessor(appContext.reactContext!!)
+
+       try {
+         return@AsyncFunction processor.processImage(imageUri)
+       } finally {
+          processor.close() //ensure cleanup
+       }
     }
 
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "expo-background-remover",
-  "version": "0.1.2",
+  "version": "0.2.2",
   "description": "Expo Module For High Quality Background Remover",
   "main": "build/index.js",
   "types": "build/index.d.ts",