Ahmad Hammoudeh commited on
Commit
373662a
·
1 Parent(s): fde02c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +35 -1
app.py CHANGED
@@ -1,7 +1,41 @@
1
  import gradio as gr
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
  def greet(name):
4
  return "Hello " + name + "!!"
5
 
6
  iface = gr.Interface(fn=greet, inputs="text", outputs="text")
7
- iface.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ from datetime import datetime
3
+ import random
4
+ import math
5
+ import base64
6
+ import hashlib
7
+ from Crypto import Random
8
+ from Crypto.Cipher import AES
9
+ import rsa
10
+ import string
11
+
12
+
13
+
14
 
15
  def greet(name):
16
  return "Hello " + name + "!!"
17
 
18
  iface = gr.Interface(fn=greet, inputs="text", outputs="text")
19
+ iface.launch()
20
+
21
+
22
+
23
+ class AESCipher(object):
24
+ def __init__(self, key):
25
+ self.bs = AES.block_size
26
+ self.key = hashlib.sha256(key.encode()).digest()
27
+ def encrypt(self, raw):
28
+ raw = self._pad(raw)
29
+ iv = Random.new().read(AES.block_size)
30
+ cipher = AES.new(self.key, AES.MODE_CBC, iv)
31
+ return base64.b64encode(iv + cipher.encrypt(raw.encode()))
32
+ def decrypt(self, enc):
33
+ enc = base64.b64decode(enc)
34
+ iv = enc[:AES.block_size]
35
+ cipher = AES.new(self.key, AES.MODE_CBC, iv)
36
+ return AESCipher._unpad(cipher.decrypt(enc[AES.block_size:])).decode('utf-8')
37
+ def _pad(self, s):
38
+ return s + (self.bs - len(s) % self.bs) * chr(self.bs - len(s) % self.bs)
39
+ @staticmethod
40
+ def _unpad(s):
41
+ return s[:-ord(s[len(s)-1:])]