From e2bc4139bac717d5ff9465311d782a660f57a289 Mon Sep 17 00:00:00 2001
From: KOSASIH <kosasihg88@gmail.com>
Date: Thu, 8 Aug 2024 16:38:54 +0700
Subject: [PATCH] Create main.py

---
 projects/PiHE/crypto/demo/main.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 projects/PiHE/crypto/demo/main.py

diff --git a/projects/PiHE/crypto/demo/main.py b/projects/PiHE/crypto/demo/main.py
new file mode 100644
index 000000000..3ac39908a
--- /dev/null
+++ b/projects/PiHE/crypto/demo/main.py
@@ -0,0 +1,28 @@
+from crypto.key_manager import KeyManager
+from crypto.homomorphic_encryption import HomomorphicEncryption
+
+def main():
+    key_manager = KeyManager(key_size=2048, ciphertext_size=2048)
+    key_manager.generate_keys()
+
+    he = HomomorphicEncryption(key_manager.get_public_key(), key_manager.get_private_key())
+
+    plaintext1 = 5
+    plaintext2 = 3
+
+    ciphertext1 = he.encrypt(plaintext1)
+    ciphertext2 = he.encrypt(plaintext2)
+
+    ciphertext_add = he.add(ciphertext1, ciphertext2)
+    ciphertext_multiply = he.multiply(ciphertext1, ciphertext2)
+
+    decrypted_add = he.decrypt(ciphertext_add)
+    decrypted_multiply = he.decrypt(ciphertext_multiply)
+
+    print(f"Plaintext 1: {plaintext1}")
+    print(f"Plaintext 2: {plaintext2}")
+    print(f"Encrypted addition: {decrypted_add}")
+    print(f"Encrypted multiplication: {decrypted_multiply}")
+
+if __name__ == "__main__":
+    main()