Update app.py

app.py

@@ -3,6 +3,7 @@ from pprint import pprint
 
 import gradio as gr
 import easy_db
+from DroidDetective import DroidDetective
 
 def get_db_path():
     return "device_data.db"
@@ -62,11 +63,31 @@ def process_request(id_textbox, installer_textbox,boot_state_textbox,verify_text
 
     apps_and_permissions = app_permissions_textbox.split("&&")
     permissions_dict = {}
+#
     for app_and_permissions in apps_and_permissions:
-        app, permissions = app_and_permissions.split("*******")
-        permissions = permissions.split("-")
-        permissions_dict[app] = permissions
-        
+        try:
+            app, permissions = app_and_permissions.split("*******")
+            permissions = permissions.split(" ")
+            permissions_dict[app] = permissions
+        except ValueError as e:
+            print(e)
+
+    is_malware_list = []
+    for permission in permissions_dict:
+        detector = DroidDetective.APK_Analyser()
+        is_malware = detector.identify(permissions_dict[permission], "DroidDetective/apk_malware.model")[0]
+
+        if is_malware == 0:
+            is_malware = False
+        else:
+            is_malware = True
+
+        is_malware_list.append(is_malware)
+
+    if True in is_malware_list:
+        results.append(True)
+    else:
+        results.append(False)
 
     dict_of_device_data_to_save = {"id": id_textbox,
                                    "installer": installer_textbox,
@@ -86,8 +107,7 @@ def process_request(id_textbox, installer_textbox,boot_state_textbox,verify_text
                                    "adb_enabled": adb_enabled_textbox,
                                    "lock_screen_timeout": lock_screen_timeout_textbox,
                                    "lock_screen_type": lock_screen_type_textbox,
-                                   "notification_visibility": notification_visibility_textbox,
-                                  "app_permissions_textbox":permissions_dict}
+                                   "notification_visibility": notification_visibility_textbox}
     db = easy_db.DataBase('device_data.db')