Update chatchain.py

chatchain.py

@@ -5,6 +5,7 @@ import hashlib
 import datetime
 from huggingface_hub import (upload_file,HfApi)
 
+import crypt
 token_self = os.environ['HF_TOKEN']
 api = HfApi(token=token_self)
 chain_d="chain1.json"
@@ -235,6 +236,7 @@ class MyChainSend:
             'recipient': recipient,
             'message': message,
         }
+        transaction = crypt.encrypt_text(transaction)
         self.pending_transactions.append(transaction)
     def proof_of_work(self, previous_proof):
         new_proof = 1
@@ -362,14 +364,14 @@ class MyChainSend:
 
 class MyChainRec:
 
-    def __init__(self,chain_load, load=None,create=None):
+    def __init__(self,chain_load, load=None,create=None,pub_key=None):
         global main_chain_rec
         main_chain_rec=chain_load
         
         self.pending_transactions = []
         if load == None or load=="":
             self.chain = []
-            self.create_block(proof=1, previous_hash='0',chain_n=create)
+            self.create_block(proof=1, previous_hash='0',chain_n=create,pub_key=pub_key)
         if load != None and load !="":
             #r = requests.get(load) 
             lod = json.loads(load)
@@ -378,7 +380,7 @@ class MyChainRec:
         self.chain = []
         self.pending_transactions = []
         self.create_block(proof=1, previous_hash='0',chain_n=create)
-    def create_block(self, proof, previous_hash,chain_r=None,chain_n=None):
+    def create_block(self, proof, previous_hash,chain_r=None,chain_n=None,pub_key=None):
         if chain_r=="" or chain_r==None:
             chain_r=f"{main_chain_rec.split('datasets/',1)[1].split('/raw',1)[0]}"
         if chain_n !="" and chain_n !=None:
@@ -389,11 +391,19 @@ class MyChainRec:
             
         #prev_block = len(blockchain.chain)-1
         #previous_block = blockchain.chain[prev_block]             
-        block = {'index': len(self.chain) + 1,
-                'timestamp': str(datetime.datetime.now()),
-                'message': self.pending_transactions,
-                'proof': proof,
-                'previous_hash': previous_hash}
+        if pub_key !=None and pub_key!="":
+            block = {'index': len(self.chain) + 1,
+                    'timestamp': str(datetime.datetime.now()),
+                    'message': pub_key,
+                    'proof': proof,
+                    'previous_hash': previous_hash}
+        else:
+            block = {'index': len(self.chain) + 1,
+                    'timestamp': str(datetime.datetime.now()),
+                    'message': self.pending_transactions,
+                    'proof': proof,
+                    'previous_hash': previous_hash}
+        
         if self.block_valid(block) == True:
             self.pending_transactions = []
             self.chain.append(block)