Update app.py

app.py

@@ -11,13 +11,14 @@ classifier = pipeline("text-classification", model="BaranKanat/BerTurk-SpamSMS")
 AES_KEY = b"ThisIsASecretKey"  # 16-byte (128-bit)
 
 # PKCS7 padding
-def pad(text):
-    pad_len = 16 - len(text) % 16
-    return text + chr(pad_len) * pad_len
+def pad(data: bytes) -> bytes:
+    pad_len = 16 - len(data) % 16
+    return data + bytes([pad_len] * pad_len)
+
+def unpad(data: bytes) -> bytes:
+    pad_len = data[-1]
+    return data[:-pad_len]
 
-def unpad(text):
-    pad_len = ord(text[-1])
-    return text[:-pad_len]
 
 # AES ile şifreleme (CBC mode), BASE64 çıktısı verir
 def encrypt(text):
@@ -25,10 +26,12 @@ def encrypt(text):
         raise ValueError("Şifrelenecek metin boş olamaz.")
     iv = get_random_bytes(16)
     cipher = AES.new(AES_KEY, AES.MODE_CBC, iv)
-    padded_text = pad(text)
-    encrypted_bytes = cipher.encrypt(padded_text.encode('utf-8'))
+    data_bytes = text.encode('utf-8')  # Türkçe karakterleri de destekler
+    padded_data = pad(data_bytes)
+    encrypted_bytes = cipher.encrypt(padded_data)
     encrypted_message = base64.b64encode(iv + encrypted_bytes).decode('utf-8')
     return encrypted_message
+    
 
 
 # AES ile şifre çözme (BASE64 giriş)
@@ -38,8 +41,9 @@ def decrypt(encrypted_text):
         iv = encrypted_data[:16]
         encrypted_msg = encrypted_data[16:]
         cipher = AES.new(AES_KEY, AES.MODE_CBC, iv)
-        decrypted = cipher.decrypt(encrypted_msg).decode('utf-8')
-        return unpad(decrypted)
+        decrypted_bytes = cipher.decrypt(encrypted_msg)
+        unpadded_bytes = unpad(decrypted_bytes)
+        return unpadded_bytes.decode('utf-8')
     except Exception as e:
         return f"Hata: {str(e)}"