Added blockchain

.idea/Zamark.iml

@@ -2,7 +2,7 @@
 <module type="PYTHON_MODULE" version="4">
   <component name="NewModuleRootManager">
     <content url="file://$MODULE_DIR$" />
-    <orderEntry type="inheritedJdk" />
+    <orderEntry type="jdk" jdkName="Python 3.10 (Zamark)" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
 </module>

.idea/misc.xml

@@ -1,4 +1,7 @@
 <?xml version="1.0" encoding="UTF-8"?>
 <project version="4">
-  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.12 (Team8)" project-jdk-type="Python SDK" />
+  <component name="Black">
+    <option name="sdkName" value="Python 3.10 (Zamark)" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10 (Zamark)" project-jdk-type="Python SDK" />
 </project>

app.py

@@ -6,6 +6,8 @@ import uuid
 import time
 import json
 
+from blockchain import Blockchain, print_blockchain_details
+
 def generate_mock_hash():
     return hashlib.sha256(str(time.time()).encode()).hexdigest()
 
@@ -216,27 +218,27 @@ if 'block_data' not in st.session_state:
 
 # Button to update the blockchain
 if st.button("Update Blockchain"):
-    previous_hash = generate_mock_hash()
-    timestamp = int(time.time() * 1000)  # Current timestamp in milliseconds
-    watermarked_model_hash = generate_mock_hash()
-    trigger_set_hash = generate_mock_hash()
+    # Charger la blockchain depuis le fichier JSON
+    loaded_blockchain, data = Blockchain.load_from_file("blockchain.json")
+
+    # Vérifier que la blockchain chargée est valide
+    print(f"La blockchain chargée est valide : {loaded_blockchain.is_chain_valid()}")
+
+    # Ajouter un nouveau bloc à la blockchain chargée
+    loaded_blockchain.add_block("HF Trigger Set 4", "Client Trigger Set 4", "Encrypted Model 4")
+
+    print("\nLoaded Blockchain:")
+    print_blockchain_details(loaded_blockchain)
+
+    # Sauvegarder la blockchain mise à jour
+    loaded_blockchain.save_to_file("blockchain.json")
+
+
+    # watermarked_model_hash = generate_mock_hash()
+    # trigger_set_hash = generate_mock_hash()
 
     # Create the block data structure
-    st.session_state.block_data = {
-        "blockNumber": 42,
-        "previousHash": previous_hash,
-        "timestamp": timestamp,
-        "transactions": [
-            {
-                "type": "Watermarked Model Hash",
-                "hash": watermarked_model_hash
-            },
-            {
-                "type": "Trigger Set Hash",
-                "hash": trigger_set_hash
-            }
-        ]
-    }
+    st.session_state.block_data = data
 
     st.success("Blockchain updated successfully!")
 

blockchain.json

@@ -0,0 +1,65 @@
+{
+    "0": {
+        "timestamp": 1727520928.1020324,
+        "previous_hash": "0",
+        "counter": 0,
+        "trigger_set_huggingface": "Genesis HuggingFace",
+        "trigger_set_client": "Genesis Client",
+        "encrypted_watermarked_model": "Genesis Model",
+        "hash": "0fe4deddadab0f2d5884b955fe6761fc39a448d73efc96cdebb2673a14ed45dd"
+    },
+    "1": {
+        "timestamp": 1727520928.1021242,
+        "previous_hash": "0fe4deddadab0f2d5884b955fe6761fc39a448d73efc96cdebb2673a14ed45dd",
+        "counter": 1,
+        "trigger_set_huggingface": "HF Trigger Set 1",
+        "trigger_set_client": "Client Trigger Set 1",
+        "encrypted_watermarked_model": "Encrypted Model 1",
+        "hash": "4e95a394e97baa18ea44e17a32266c4c39ffd5da28bc45d79811d50bcc4c8e28"
+    },
+    "2": {
+        "timestamp": 1727520928.1021426,
+        "previous_hash": "4e95a394e97baa18ea44e17a32266c4c39ffd5da28bc45d79811d50bcc4c8e28",
+        "counter": 2,
+        "trigger_set_huggingface": "HF Trigger Set 2",
+        "trigger_set_client": "Client Trigger Set 2",
+        "encrypted_watermarked_model": "Encrypted Model 2",
+        "hash": "195c1701b1b019219182df1b7c7a524a9a9a3dcaaab3fa899ae29bdab7e8e44f"
+    },
+    "3": {
+        "timestamp": 1727520928.1021535,
+        "previous_hash": "195c1701b1b019219182df1b7c7a524a9a9a3dcaaab3fa899ae29bdab7e8e44f",
+        "counter": 3,
+        "trigger_set_huggingface": "HF Trigger Set 3",
+        "trigger_set_client": "Client Trigger Set 3",
+        "encrypted_watermarked_model": "Encrypted Model 3",
+        "hash": "e30ccc1af1af5241babc635ebf2f23ebcf7d97690eefb3d49f52f84e3f5ab64a"
+    },
+    "4": {
+        "timestamp": 1727520928.10693,
+        "previous_hash": "e30ccc1af1af5241babc635ebf2f23ebcf7d97690eefb3d49f52f84e3f5ab64a",
+        "counter": 4,
+        "trigger_set_huggingface": "HF Trigger Set 4",
+        "trigger_set_client": "Client Trigger Set 4",
+        "encrypted_watermarked_model": "Encrypted Model 4",
+        "hash": "6bcd4768b2b5d2a6d51ae12add4c6da6cb43a32ada36a247ea4734fa9a60e647"
+    },
+    "5": {
+        "timestamp": 1727520938.4695382,
+        "previous_hash": "6bcd4768b2b5d2a6d51ae12add4c6da6cb43a32ada36a247ea4734fa9a60e647",
+        "counter": 5,
+        "trigger_set_huggingface": "HF Trigger Set 4",
+        "trigger_set_client": "Client Trigger Set 4",
+        "encrypted_watermarked_model": "Encrypted Model 4",
+        "hash": "83d84d3bfc3f707b1987227187468aeb4cfca44c4bdf9e746c9334f6cb80d2a4"
+    },
+    "6": {
+        "timestamp": 1727520943.492908,
+        "previous_hash": "83d84d3bfc3f707b1987227187468aeb4cfca44c4bdf9e746c9334f6cb80d2a4",
+        "counter": 6,
+        "trigger_set_huggingface": "HF Trigger Set 4",
+        "trigger_set_client": "Client Trigger Set 4",
+        "encrypted_watermarked_model": "Encrypted Model 4",
+        "hash": "aaccd1f43b43d924619665155cfcb292cf2cf2597d9c4c32197e2155696fde47"
+    }
+}

blockchain.py

@@ -0,0 +1,138 @@
+import hashlib
+import time
+import json
+
+
+class Block:
+    def __init__(self, previous_hash, trigger_set_huggingface_hash, trigger_set_client_hash, encrypted_watermarked_model_hash, counter,
+                 timestamp=None):
+        self.timestamp = timestamp if timestamp else time.time()
+        self.previous_hash = previous_hash
+        self.counter = counter
+        self.trigger_set_huggingface = trigger_set_huggingface_hash
+        self.trigger_set_client = trigger_set_client_hash
+        self.encrypted_watermarked_model = encrypted_watermarked_model_hash
+        self.hash = self.calculate_hash()
+
+    def calculate_hash(self):
+        hash_string = (
+                f"{self.timestamp:.6f}" +
+                str(self.previous_hash) +
+                str(self.counter) +
+                str(self.trigger_set_huggingface) +
+                str(self.trigger_set_client) +
+                str(self.encrypted_watermarked_model)
+        )
+        return hashlib.sha256(hash_string.encode()).hexdigest()
+
+    @staticmethod
+    def hash_data(data):
+        return hashlib.sha256(str(data).encode()).hexdigest()
+
+    def to_dict(self):
+        return {
+            "timestamp": self.timestamp,
+            "previous_hash": self.previous_hash,
+            "counter": self.counter,
+            "trigger_set_huggingface": self.trigger_set_huggingface,
+            "trigger_set_client": self.trigger_set_client,
+            "encrypted_watermarked_model": self.encrypted_watermarked_model,
+            "hash": self.hash
+        }
+
+
+class Blockchain:
+    def __init__(self):
+        self.chain = {}
+        self.add_block("Genesis HuggingFace", "Genesis Client", "Genesis Model")
+
+    def add_block(self, trigger_set_huggingface, trigger_set_client, encrypted_watermarked_model):
+        counter = len(self.chain)
+        previous_hash = self.chain[counter - 1].hash if counter > 0 else "0"
+        new_block = Block(previous_hash, trigger_set_huggingface, trigger_set_client, encrypted_watermarked_model,
+                          counter)
+        self.chain[counter] = new_block
+
+    def is_chain_valid(self):
+        for i in range(1, len(self.chain)):
+            current_block = self.chain[i]
+            previous_block = self.chain[i - 1]
+
+            if current_block.hash != current_block.calculate_hash():
+                print(f"Invalid hash for block {i}")
+                return False
+
+            if current_block.previous_hash != previous_block.hash:
+                print(f"Invalid previous hash for block {i}")
+                return False
+
+        return True
+
+    def to_dict(self):
+        return {str(counter): block.to_dict() for counter, block in self.chain.items()}
+
+    def save_to_file(self, filename):
+        with open(filename, 'w') as file:
+            json.dump(self.to_dict(), file, indent=4)
+        print(f"Blockchain saved to {filename}")
+
+    @classmethod
+    def load_from_file(cls, filename):
+        with open(filename, 'r') as file:
+            data = json.load(file)
+
+        blockchain = cls()
+        blockchain.chain.clear()  # Clear the genesis block
+        for counter, block_data in data.items():
+            block = Block(
+                block_data["previous_hash"],
+                block_data["trigger_set_huggingface"],
+                block_data["trigger_set_client"],
+                block_data["encrypted_watermarked_model"],
+                int(counter),
+                block_data["timestamp"]
+            )
+            blockchain.chain[int(counter)] = block
+
+        print(f"Blockchain loaded from {filename}")
+        return blockchain, data
+
+
+def print_blockchain_details(blockchain):
+    for counter, block in blockchain.chain.items():
+        print(f"Block {counter}:")
+        print(f"  Timestamp: {block.timestamp:.6f}")
+        print(f"  Previous Hash: {block.previous_hash}")
+        print(f"  Hash: {block.hash}")
+        print(f"  Calculated Hash: {block.calculate_hash()}")
+        print()
+
+
+# Exemple d'utilisation
+blockchain = Blockchain()
+
+# Ajouter quelques blocs
+blockchain.add_block("HF Trigger Set 1", "Client Trigger Set 1", "Encrypted Model 1")
+blockchain.add_block("HF Trigger Set 2", "Client Trigger Set 2", "Encrypted Model 2")
+blockchain.add_block("HF Trigger Set 3", "Client Trigger Set 3", "Encrypted Model 3")
+
+print("Original Blockchain:")
+print_blockchain_details(blockchain)
+
+# Sauvegarder la blockchain dans un fichier JSON
+blockchain.save_to_file("blockchain.json")
+
+# Charger la blockchain depuis le fichier JSON
+loaded_blockchain, _ = Blockchain.load_from_file("blockchain.json")
+
+print("\nLoaded Blockchain:")
+print_blockchain_details(loaded_blockchain)
+
+# Vérifier que la blockchain chargée est valide
+print(f"La blockchain chargée est valide : {loaded_blockchain.is_chain_valid()}")
+
+# Ajouter un nouveau bloc à la blockchain chargée
+loaded_blockchain.add_block("HF Trigger Set 4", "Client Trigger Set 4", "Encrypted Model 4")
+
+# Sauvegarder la blockchain mise à jour
+loaded_blockchain.save_to_file("blockchain.json")