Merge pull request #16 from anrouxel/AddPrescription-View

Add prescription view

app/build.gradle.kts

@@ -116,6 +116,9 @@ dependencies {
     implementation("androidx.room:room-runtime:2.5.0")
     ksp("androidx.room:room-compiler:2.5.0")
 
+    // Alarm
+    implementation("com.github.ColdTea-Projects:SmplrAlarm:2.1.0")
+
     testImplementation("junit:junit:4.13.2")
     androidTestImplementation("androidx.test.ext:junit:1.1.5")
     androidTestImplementation("androidx.test.espresso:espresso-core:3.5.1")

app/src/main/AndroidManifest.xml

@@ -4,9 +4,14 @@
 
     <uses-feature android:name="android.hardware.camera" />
     <uses-permission android:name="android.permission.CAMERA" />
+    <uses-permission android:name="android.permission.INTERNET"/>
+
+    <uses-permission android:name="android.permission.POST_NOTIFICATIONS"/>
+    <uses-permission android:name="android.permission.SCHEDULE_EXACT_ALARM"/>
 
     <application
         android:allowBackup="true"
+        tools:replace="android:allowBackup"
         android:dataExtractionRules="@xml/data_extraction_rules"
         android:fullBackupContent="@xml/backup_rules"
         android:icon="@mipmap/ic_launcher"

app/src/main/java/fr/medicapp/medicapp/MainActivity.kt

@@ -1,20 +1,24 @@
 package fr.medicapp.medicapp
 
+import android.os.Build
 import android.os.Bundle
 import androidx.activity.ComponentActivity
 import androidx.activity.compose.setContent
+import androidx.annotation.RequiresApi
+import androidx.compose.runtime.getValue
 import androidx.navigation.compose.rememberNavController
+import fr.medicapp.medicapp.ai.PrescriptionAI
 import fr.medicapp.medicapp.database.AppDatabase
-import fr.medicapp.medicapp.entity.DoctorEntity
-import fr.medicapp.medicapp.entity.PrescriptionEntity
 import fr.medicapp.medicapp.ui.navigation.RootNavGraph
 import fr.medicapp.medicapp.ui.theme.MedicAppTheme
 
 class MainActivity : ComponentActivity() {
+    @RequiresApi(Build.VERSION_CODES.O)
     override fun onCreate(savedInstanceState: Bundle?) {
         super.onCreate(savedInstanceState)
 
-        val db = AppDatabase.getInstance(this)
+        AppDatabase.getInstance(this)
+        PrescriptionAI.getInstance(this)
 
         setContent {
             MedicAppTheme {

app/src/main/java/fr/medicapp/medicapp/ai/PrescriptionAI.kt

@@ -2,10 +2,14 @@ package fr.medicapp.medicapp.ai
 
 import android.content.Context
 import android.content.res.AssetManager
+import android.net.Uri
 import android.os.Handler
 import android.os.HandlerThread
 import android.util.Log
 import androidx.annotation.WorkerThread
+import com.google.mlkit.vision.common.InputImage
+import com.google.mlkit.vision.text.TextRecognition
+import com.google.mlkit.vision.text.latin.TextRecognizerOptions
 import fr.medicapp.medicapp.tokenization.Feature
 import fr.medicapp.medicapp.tokenization.FeatureConverter
 import org.pytorch.IValue
@@ -39,23 +43,50 @@ class PrescriptionAI(
     private val PREDICT_ANS_NUM = 5
     private val NUM_LITE_THREADS = 4
 
-    /*val modelRunnable = Runnable {
+    private val recognizer = TextRecognition.getClient(TextRecognizerOptions.DEFAULT_OPTIONS)
 
-    }*/
-
-    fun analyse(visionText: String): MutableList<Pair<String, String>> {
-        val latch = CountDownLatch(1)
-        var result = mutableListOf<Pair<String, String>>()
+    init {
         mBackgroundThread = HandlerThread("BackgroundThread")
         mBackgroundThread.start()
         mHandle = Handler(mBackgroundThread.looper)
         mHandle.post {
             loadModel()
-            result = runModel(visionText)
-            latch.countDown() // Decrements the count of the latch, releasing all waiting threads if the count reaches zero.
         }
-        latch.await() // Causes the current thread to wait until the latch has counted down to zero.
-        return result
+    }
+
+    fun analyse(
+        imageUri: Uri,
+        onPrediction: (MutableList<Pair<String, String>>) -> Unit,
+        onDismiss: () -> Unit
+    ) {
+        mHandle.post {
+            while (mModule == null) {
+                Thread.sleep(100)
+            }
+            val visionText = recognizeText(imageUri)
+            if (visionText != null) {
+                val sentenceTokenized = runModel(visionText)
+                onPrediction(sentenceTokenized)
+            }
+            onDismiss()
+        }
+    }
+
+    @WorkerThread
+    private fun recognizeText(imageUri: Uri): String? {
+        var visionText: String? = null
+        val latch = CountDownLatch(1)
+        val image = InputImage.fromFilePath(context, imageUri)
+        recognizer.process(image)
+            .addOnSuccessListener {
+                visionText = it.text
+                latch.countDown()
+            }
+            .addOnFailureListener {
+                latch.countDown()
+            }
+        latch.await()
+        return visionText
     }
 
     @WorkerThread
@@ -76,7 +107,7 @@ class PrescriptionAI(
             }
 
             val moduleFileAbsoluteFilePath: String? = assetFilePath(context.assets)
-            mModule = Module.load(moduleFileAbsoluteFilePath)//, mutableMapOf(), Device.CPU)
+            mModule = Module.load(moduleFileAbsoluteFilePath) // , mutableMapOf(), Device.CPU)
             Log.v(TAG, "Model loaded.")
         }
     }
@@ -150,7 +181,10 @@ class PrescriptionAI(
 
             for (i in predictionsLabelList.indices) {
                 if (predictionsLabelList[i] != "O") {
-                    Log.d("AIModel", "Prediction: ${feature.origTokens[i]} -> ${predictionsLabelList[i]}")
+                    Log.d(
+                        "AIModel",
+                        "Prediction: ${feature.origTokens[i]} -> ${predictionsLabelList[i]}"
+                    )
                     sentenceTokenized.add(Pair(feature.origTokens[i], predictionsLabelList[i]))
                 }
             }
@@ -209,4 +243,4 @@ class PrescriptionAI(
             }
         }
     }
-}
+}

app/src/main/java/fr/medicapp/medicapp/dao/MedicationDAO.kt

@@ -0,0 +1,39 @@
+package fr.medicapp.medicapp.dao
+
+import androidx.room.Dao
+import androidx.room.Delete
+import androidx.room.Insert
+import androidx.room.Query
+import androidx.room.Update
+import fr.medicapp.medicapp.entity.MedicationEntity
+import fr.medicapp.medicapp.entity.UserEntity
+
+@Dao
+interface MedicationDAO {
+    @Query("SELECT * FROM Medications")
+    fun getAll(): List<MedicationEntity>
+
+    @Query("SELECT * FROM Medications WHERE CommercializationStatus NOT LIKE 'NON COMMERCIALISÉE'")
+    fun getAllWithoutNotTreadings(): List<MedicationEntity>
+
+    @Query("SELECT * FROM Medications WHERE cisCode = :id")
+    fun getOne(id: String): MedicationEntity
+
+    @Insert
+    fun add(t: MedicationEntity)
+
+    @Insert
+    fun addAll(vararg t: MedicationEntity)
+
+    @Delete
+    fun delete(t: MedicationEntity)
+
+    @Delete
+    fun deleteAll(vararg t: MedicationEntity)
+
+    @Update
+    fun update(t: MedicationEntity)
+
+    @Update
+    fun updateAll(vararg t: MedicationEntity)
+}

app/src/main/java/fr/medicapp/medicapp/dao/NotificationDAO.kt

@@ -0,0 +1,41 @@
+package fr.medicapp.medicapp.dao
+
+import androidx.room.Dao
+import androidx.room.Delete
+import androidx.room.Insert
+import androidx.room.Query
+import androidx.room.Update
+import fr.medicapp.medicapp.entity.DoctorEntity
+import fr.medicapp.medicapp.entity.NotificationEntity
+import fr.medicapp.medicapp.entity.SideEffectEntity
+
+@Dao
+interface NotificationDAO {
+    @Query("SELECT * FROM NotificationEntity")
+    fun getAll(): List<NotificationEntity>
+
+    @Query("SELECT * FROM NotificationEntity WHERE id = :id")
+    fun getOne(id: String): NotificationEntity
+
+    @Query("SELECT * FROM NotificationEntity WHERE medicationName = :medicament")
+    fun getByMedicament(medicament: String): List<NotificationEntity>
+
+
+    @Insert
+    fun add(t: NotificationEntity)
+
+    @Insert
+    fun addAll(vararg t: NotificationEntity)
+
+    @Delete
+    fun delete(t: NotificationEntity)
+
+    @Delete
+    fun deleteAll(vararg t: NotificationEntity)
+
+    @Update
+    fun update(t: NotificationEntity)
+
+    @Update
+    fun updateAll(vararg t: NotificationEntity)
+}

app/src/main/java/fr/medicapp/medicapp/dao/SideEffectDAO.kt

@@ -15,6 +15,9 @@ interface SideEffectDAO {
     @Query("SELECT * FROM SideEffectEntity WHERE id = :id")
     fun getOne(id: String): SideEffectEntity
 
+    @Query("SELECT * FROM SideEffectEntity WHERE medicament = :medicament")
+    fun getByMedicament(medicament: String): List<SideEffectEntity>
+
     @Insert
     fun add(t: SideEffectEntity)
 

app/src/main/java/fr/medicapp/medicapp/dao/TreatmentDAO.kt

@@ -16,6 +16,9 @@ interface TreatmentDAO {
     @Query("SELECT * FROM TreatmentEntity WHERE id = :id")
     fun getOne(id: String): TreatmentEntity
 
+    @Query("SELECT * FROM TreatmentEntity WHERE notification = 1")
+    fun getWithNotification(): List<TreatmentEntity>
+
     @Insert
     fun add(t: TreatmentEntity)