Update app.py

app.py

@@ -1,33 +1,209 @@
-import gradio as gr
 import tensorflow as tf
+import gradio as gr
+from joblib import load
 import numpy as np 
-import cv2
-
-new_model = tf.keras.models.load_model('best_model_mammography.h5')
-
-def classify_image(file_name):
-    img1 = cv2.imread(file_name.name.replace("\\",'/'),0)
-    img = cv2.resize(img1, (224,224))
-    img = img.reshape(img.shape[0],img.shape[1],1)
-    pred = new_model.predict(np.array([img]))
-    pred = np.round(pred,1)
-    if pred == 0:
-        pred = "Vôtre cancer est Begnine"
+
+
+
+model_multi_path = "bestmodel_attack.h5"
+model_bin_path = "model_binary.joblib"
+
+model_multi= tf.keras.models.load_model('bestmodel_attack.h5')
+model_bin = load("model_binary.joblib")
+# protocol icmp tcp udp
+
+# service IRC X11 Z3950 aol auth bgp courier csnet_ns ctf daytime discard domain domain_u echo eco_i ecr_i
+# efs exec finger ftp ftp_data gopher harvest hostnames http http_2784 http_443 http_8001 imap4
+# iso_tsap klogin kshell ldap link login mtp name netbios_dgm netbios_ns netbios_ssn 
+# netstat nnsp nntp ntp_u other pm_dump pop_2 pop_3 printer private red_i remote_job rje 
+# shell smtp sql_net ssh sunrpc supdup systat telnet tftp_u tim_i time urh_i urp_i uucp
+# uucp_path vmnet whois 
+
+
+# flag 
+# OTH RSTO RSTOS0 RSTR S0 S1 S2 S3 SF SH 
+
+# duration
+
+# src_bytes
+
+# dst_bytes
+
+# urgent
+
+# root_shell 
+
+# su_attempted
+
+# num_file_creations
+
+# num_shells
+
+# num_access_files
+
+# is_guest_login
+
+data_mining = {
+    "protocol": ["icmp", "tcp", "udp"],
+    "service":['IRC', 'X11', 'Z39_50', 'aol','auth', 'bgp', 'courier', 'csnet_ns','ctf', 'daytime', 
+               'discard', 'domain','domain_u', 'echo', 'eco_i', 'ecr_i','efs', 'exec', 'finger', 'ftp',
+               'ftp_data', 'gopher', 'harvest','hostnames', 'http', 'http_2784','http_443', 'http_8001', 
+               'imap4','iso_tsap', 'klogin', 'kshell', 'ldap','link', 'login', 'mtp', 'name','netbios_dgm', 
+               'netbios_ns', 'netbios_ssn','netstat', 'nnsp', 'nntp', 'ntp_u','other', 'pm_dump', 
+               'pop_2', 'pop_3','printer', 'private', 'red_i','remote_job', 'rje', 'shell', 'smtp',
+               'sql_net', 'ssh', 'sunrpc', 'supdup','systat', 'telnet', 'tftp_u', 'tim_i','time', 
+               'urh_i', 'urp_i', 'uucp','uucp_path', 'vmnet', 'whois'],
+    'flag':['OTH', 'REJ', 'RSTO', 'RSTOS0', 'RSTR','S0', 'S1', 'S2', 'S3', 'SF', 'SH']
+}
+def greet(choix,protocol,services, flag,duration,src_bytes,dst_bytes,land,wrong_fragment,urgent,hot, # ok
+num_failed_logins,logged_in,num_compromised,root_shell,su_attempted,num_root,num_file_creations,num_shells # ok
+,num_access_files,num_outbound_cmds,is_host_login,is_guest_login,count,srv_count,serror_rate,srv_serror_rate #ok
+,rerror_rate,srv_rerror_rate,same_srv_rate,diff_srv_rate,srv_diff_host_rate,dst_host_count,dst_host_srv_count #ok
+,dst_host_same_srv_rate,dst_host_diff_srv_rate,dst_host_same_src_port_rate,dst_host_srv_diff_host_rate,dst_host_serror_rate
+,dst_host_srv_serror_rate,dst_host_rerror_rate,dst_host_srv_rerror_rate):
+    liste_head = [0]* len(data_mining["protocol"])
+    liste_head[data_mining["protocol"].index(protocol)] = 1
+    liste_head2 = [0]* len(data_mining["service"])
+    liste_head2[data_mining["service"].index(services)] = 1
+    liste_head3 = [0]* len(data_mining["flag"])
+    liste_head3[data_mining["flag"].index(flag)] = 1
+    list1 = list()
+    pred = None
+    name = str()
+
+    list1 = liste_head + liste_head2 + liste_head3 + [duration,src_bytes,dst_bytes,land,wrong_fragment,urgent,hot,
+num_failed_logins,logged_in,num_compromised,root_shell,su_attempted,num_root,num_file_creations,num_shells
+,num_access_files,num_outbound_cmds,is_host_login,is_guest_login,count,srv_count,serror_rate,srv_serror_rate
+,rerror_rate,srv_rerror_rate,same_srv_rate,diff_srv_rate,srv_diff_host_rate,dst_host_count,dst_host_srv_count
+,dst_host_same_srv_rate,dst_host_diff_srv_rate,dst_host_same_src_port_rate,dst_host_srv_diff_host_rate,dst_host_serror_rate
+,dst_host_srv_serror_rate,dst_host_rerror_rate,dst_host_srv_rerror_rate]
+    arr = np.array(list1)
+    if choix == "Binaire":
+        pred = int(model_bin.predict(arr.reshape(1,arr.shape[0])))
+        if pred == 0:
+            name = "Aucune Attaque détectée"
+        else:
+            name= "Attaque détectée"
     else:
-        pred= "Vôtre cancer est maline"
-    return pred
-
-
-image = gr.inputs.File( file_count="single",type="file", label="Fichier à Traiter (sous fichier .pgm)")
-# label = gr.outputs.Label(num_top_classes=3)
-
-gr.Interface(
-    fn=classify_image, 
-    inputs=image, 
-    outputs="text", 
-    interpretation="default",
-    live=True,
-    theme="dark-peach",
-    title="API BREASTNET de Test de diagnostique du Cancer de Sein",
-    description="Cette API est utilisé pour dire si le Cancer de sein est Maline ou Pas"
-).launch()
+        pred = int(model_multi.predict(arr.reshape(1,arr.shape[0])))
+        if pred == 0:
+            name = "Aucune Attaque détectée"
+        elif pred == 1:
+            name = "Attaque DOS détecté"
+        elif pred == 2:
+            name = "Attaque Probe détecté"
+        elif pred == 3:
+            name = "Attaque de Privilège détecté"
+        else:
+            name = "Attaque d'accès détecter"
+    return f"La prédiction est : {name} "
+
+
+
+iface = gr.Interface(
+    fn=greet, 
+    title="API de Test de pénétration",
+    description="Cette API est utilisé pour deux actions binaires et multidetection assisté d'un environnement test continuel",
+    inputs=[
+        gr.inputs.Radio(["Binaire", "Catégorisation"],label="Choix de Prediction"), # choix
+        gr.inputs.Radio(["icmp", "tcp", "udp"],label="protocol de communication"), # protocol
+        gr.inputs.Radio(['IRC', 'X11', 'Z39_50', 'aol','auth', 'bgp', 'courier', 'csnet_ns','ctf', 'daytime', 
+                        'discard', 'domain','domain_u', 'echo', 'eco_i', 'ecr_i','efs', 'exec', 'finger', 'ftp',
+                        'ftp_data', 'gopher', 'harvest','hostnames', 'http', 'http_2784','http_443', 'http_8001', 
+                        'imap4','iso_tsap', 'klogin', 'kshell', 'ldap','link', 'login', 'mtp', 'name','netbios_dgm', 
+                        'netbios_ns', 'netbios_ssn','netstat', 'nnsp', 'nntp', 'ntp_u','other', 'pm_dump', 
+                        'pop_2', 'pop_3','printer', 'private', 'red_i','remote_job', 'rje', 'shell', 'smtp',
+                        'sql_net', 'ssh', 'sunrpc', 'supdup','systat', 'telnet', 'tftp_u', 'tim_i','time', 
+                        'urh_i', 'urp_i', 'uucp','uucp_path', 'vmnet', 'whois']
+                        ,label="services de communication"), # service
+        gr.inputs.Radio(['OTH', 'REJ', 'RSTO', 'RSTOS0', 'RSTR','S0', 'S1', 'S2', 'S3', 'SF', 'SH'], # 10 
+                        label="drapeau de communication"), # flag
+        gr.inputs.Slider(minimum=0, maximum=360,label="durée de l'execution"), # duration
+        gr.inputs.Slider(minimum=0, maximum=360,label="nombre de bits sources"), # src_bytes
+        gr.inputs.Slider(minimum=0, maximum=360,label="nombre de bits destination"), # dst_bytes
+        gr.inputs.Slider(minimum=0, maximum=360,label="land"), # land 
+        gr.inputs.Slider(minimum=0, maximum=360,label="wrong_fragment"), # wrong_fragment
+        gr.inputs.Slider(minimum=0, maximum=150,step=1,label="l'urgence de l'execution"), # urgent
+        gr.inputs.Slider(minimum=0, maximum=45,step=1,label="hot"), # hot
+        gr.inputs.Slider(minimum=0, maximum=100,step=1,label="num_failed_logins"), # num_failed_logins
+        gr.inputs.Slider(minimum=0, maximum=3,step=1,label="logged_in"), # logged_in
+        gr.inputs.Slider(minimum=0, maximum=350,step=1,label="num_compromised"), # num_compromised
+        gr.inputs.Slider(minimum=0, maximum=35,step=1,label="ouverture du shell en super utilisateur"), # root_shell
+        gr.inputs.Slider(minimum=0, maximum=50,step=1,label="utilisation du mode super utilisateur"), # su_attempted
+        gr.inputs.Slider(minimum=0, maximum=315,step=1,label="num_root"), # num_root
+        gr.inputs.Slider(minimum=0, maximum=200,step=1,label="nombre de fichier en création"), # num_file_creations
+        gr.inputs.Slider(minimum=0, maximum=200,step=1,label="nombre de shell"), # num_shells
+        gr.inputs.Slider(minimum=0, maximum=200,step=1,label="nombre d'accès de fichier"), # num_access_files
+        gr.inputs.Slider(minimum=0, maximum=1,step=1,label="num_outbound_cmds"), # num_outbound_cmds
+        gr.inputs.Slider(minimum=0, maximum=400,step=1,label="is_host_login"), # is_host_login
+        gr.inputs.Slider(minimum=0, maximum=15,step=1,label="is_guest_login"), # is_guest_login
+        gr.inputs.Slider(minimum=0, maximum=5,step=1,label="count"), # count
+        gr.inputs.Slider(minimum=0, maximum=7,step=1,label="srv_count"), # srv_count
+        gr.inputs.Slider(minimum=0, maximum=1.6,step=.1,label="serror_rate"), # serror_rate
+        gr.inputs.Slider(minimum=0, maximum=2,step=.1,label="srv_serror_rate"), # srv_rerror_rate
+        gr.inputs.Slider(minimum=0, maximum=3,step=.1,label="rerror_rate"), # rerror_rate
+        gr.inputs.Slider(minimum=0, maximum=4,step=.1,label="srv_rerror_rate"), # srv_rerror_rate
+        gr.inputs.Slider(minimum=0, maximum=1,step=.1,label="same_srv_rate"), # same_srv_rate
+        gr.inputs.Slider(minimum=0, maximum=6,step=1,label="diff_srv_rate"), # diff_srv_rate
+        gr.inputs.Slider(minimum=0, maximum=6,step=1,label="srv_diff_host_rate"), # srv_diff_host_rate
+        gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_count"), # dst_host_count
+        gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_srv_count"), # dst_host_srv_count
+        gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_same_srv_rate"), # dst_host_same_srv_rate
+        gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_same_srv_rate"), # dst_host_diff_srv_rate
+        gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_same_src_port_rate"), # dst_host_same_src_port_rate
+        gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_srv_diff_host_rate"), # dst_host_srv_diff_host_rate
+        gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_serror_rate"), # dst_host_serror_rate
+        gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_srv_serror_rate"), # dst_host_srv_serror_rate
+        gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_rerror_rate"), # dst_host_rerror_rate
+        gr.inputs.Slider(minimum=0, maximum=10,step=1,label="dst_host_srv_rerror_rate") #, # dst_host_srv_rerror_rate
+        # gr.inputs.Number(label="sources bytes de la chaine d'execution"),
+        # gr.inputs.Number(label="destination bytes de la chaine d'execution"),
+        # gr.inputs.Slider(minimum=0, maximum=80,step=1,label="land"), # land
+        # gr.inputs.Number(label="wrong fragment"),
+        # gr.inputs.Slider(minimum=0, maximum=150,step=1,label="l'urgence de l'execution"),
+        # gr.inputs.Slider(minimum=0, maximum=45,step=1,label="hot"),
+        # gr.inputs.Slider(minimum=0, maximum=100,step=1,label="num_failed_logins"),
+        # gr.inputs.Slider(minimum=0, maximum=3,step=1,label="logged_in"),
+        # gr.inputs.Slider(minimum=0, maximum=350,step=1,label="num_compromised"),
+        # gr.inputs.Slider(minimum=0, maximum=35,step=1,label="ouverture du shell en super utilisateur"),
+        # gr.inputs.Slider(minimum=0, maximum=50,step=1,label="utilisation du mode super utilisateur"),
+        # gr.inputs.Slider(minimum=0, maximum=315,step=1,label=",num_root"),
+        # gr.inputs.Slider(minimum=0, maximum=200,step=1,label=",nombre de fichier en création"),
+        # gr.inputs.Slider(minimum=0, maximum=200,step=1,label=",nombre de shell"),
+        # gr.inputs.Slider(minimum=0, maximum=200,step=1,label=",nombre d'accès de fichier"),
+        # gr.inputs.Slider(minimum=0, maximum=1,step=1,label="num_outbound_cmds"),
+        # gr.inputs.Slider(minimum=0, maximum=400,step=1,label=",is_host_login"),
+        # gr.inputs.Slider(minimum=0, maximum=15,step=1,label="i,s_guest_login"),
+        # gr.inputs.Slider(minimum=0, maximum=5,step=1,label="co,unt"),
+        # gr.inputs.Slider(minimum=0, maximum=7,step=1,label="sr,v_count"),
+        # gr.inputs.Slider(minimum=0, maximum=1.6,step=.1,label=,"serror_rate"),
+        # gr.inputs.Slider(minimum=0, maximum=2,step=.1,label="s,rv_serror_rate"),
+        # gr.inputs.Slider(minimum=0, maximum=3,step=.1,label="r,error_rate"), 
+        # gr.inputs.Slider(minimum=0, maximum=4,step=.1,label="s,rv_rerror_rate"), 
+        # gr.inputs.Slider(minimum=0, maximum=1,step=.1,label="s,ame_srv_rate"), 
+        # gr.inputs.Slider(minimum=0, maximum=6,step=1,label="di,ff_srv_rate"),
+        # gr.inputs.Slider(minimum=0, maximum=6,step=1,label="sr,v_diff_host_rate"),
+        # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_count"),
+        # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_srv_count"),
+        # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_same_srv_rate"),
+        # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_same_src_port_rate"),
+        # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_srv_diff_host_rate"),
+        # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_serror_rate"),
+        # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_srv_serror_rate"),
+        # gr.inputs.Slider(minimum=0, maximum=10,step=1,label="d,st_host_srv_rerror_rate")#,
+        # gr.inputs.Slider(minimum=0, maximum=15,step=1,label=,"nombre de shell crée"),
+        # gr.inputs.Slider(minimum=0, maximum=15,step=1,label=,"accès aux fichiers systèmes"),
+        # gr.inputs.Slider(minimum=0, maximum=1,step=1,label="est en login")
+        ], # ok 
+        
+    outputs="text",
+    examples=[
+        ["Binaire", "icmp", 'IRC','OTH',4000,45,35,2,0,1,20,5,7,0],
+        ["Catégorisation", "tcp", 'netstat','REJ',3500,25,15,0,1,0,17,10,2,0],
+        ["Catégorisation", "udp", 'login','S0',150,30,5,0,1,20,2,15,1,1]
+    ],
+    flagging_options=["this", "or", "that"],
+    theme="dark-seafoam",
+    thumbnail='images.jpg'
+    )
+iface.launch(inbrowser=True,favicon_path='images.jpg',share= True)# share= True,auth=("admin", "pass1234"))