Update app.py

app.py

@@ -44,28 +44,13 @@ def encrypt(passw,mes=None,img=None,doc=None,trans_s=None,trans_r=None,trans_a=N
     fernet = Fernet(key)
     enc_mes=""
     enc_file=None
-    if mes != None and mes != "":
-        bytes_m = mes.encode()
-        enc_mes = fernet.encrypt(bytes_m)
-        enc_mes = f'{enc_mes}+aaa+'
-    if img != None:
-        with open(f'{img}', "rb") as image_file:
-            bytes_i = base64.b64encode(image_file.read()) 
-        if mes != None and mes != "":
-            im_bytes = fernet.encrypt(bytes_i)
-            enc_mes = f'{enc_mes}{im_bytes}+bbb+'
-        else:
-            im_bytes = fernet.encrypt(bytes_i)
-            enc_mes = f'{im_bytes}+bbb+'
+ 
     if trans_s != None and trans_r != None and trans_a != None and trans_s != "" and trans_r != "" and trans_a != "":
         trans = trans_block(trans_s,trans_r,trans_a)
         bytes_j = trans.encode()
-        if mes != None or mes != "" or img != None or img !="":
-            json_bytes = fernet.encrypt(bytes_j)
-            enc_mes = f'{enc_mes}{json_bytes}+ccc+'
-        else:    
-            json_bytes = fernet.encrypt(bytes_j)
-            enc_mes = f'{json_bytes}+ccc+'
+
+        json_bytes = fernet.encrypt(bytes_j)
+        enc_mes = f'{json_bytes}+ccc+'
 
     #enc_qr = qr.make_qr(txt=(enc_mes.strip('"')))
     m = hashlib.sha256()
@@ -76,26 +61,8 @@ def encrypt(passw,mes=None,img=None,doc=None,trans_s=None,trans_r=None,trans_a=N
     print(out)
     qr_link = out
     enc_qr = stegan.conv_im(qr_link=qr_link,data=enc_mes)
-    
-    if doc != None:
-        #print(dir(doc))
-        #print(doc.name)
-        doc_name = doc.name
-        doc_name = doc_name.split("/",4)[4]
-        with open(doc.name, "rb") as file:
-            # read all file data
-            file_data = file.read()
-        enc_doc = fernet.encrypt(file_data)
-        og_name = doc.name
-        og_end = og_name.split(".",1)[1]
-        og_front=og_name.split(".",1)[0]
-        
-        
-        enc_file=f'{doc.name}.ocrpt'
-        with open(enc_file, "wb") as file:
-            file.write(enc_doc)
-    #current_url.url_return("www.google.com")
-    return enc_mes,enc_file,enc_qr,enc_qr
+  
+    return enc_mes,enc_qr,enc_qr
     
 def decrypt(passw,enc_in=None):
     key = create_key(passw)
@@ -246,6 +213,6 @@ with gr.Blocks() as app:
 
     dec_qr_btn.click(decode_qr,[dec_qr_im,pass_out],[dec_qr_im_out,dec_qr_txt,d_json])
     dec_doc_btn.click(decode_doc,[pass_out,dec_doc_in],dec_doc_out)
-    en_btn.click(encrypt,[pass_in,mes,im,doc,send,rec,am],[enc_out,enc_doc_out,enc_qr_out,dec_qr_im])
+    en_btn.click(encrypt,[pass_in,mes,im,doc,send,rec,am],[enc_out,enc_qr_out,dec_qr_im])
     d_btn.click(decrypt,[pass_out,enc_in],[d_im,d_txt,d_json])
 app.launch()