Update Blockchain.py

Blockchain.py

@@ -18,6 +18,9 @@ class Blockchain:
         :param previous_hash: 이전 블록의 해시 값
         :return: 새로 생성된 블록
         """
+        for id, mem in self.user_gpus.items():
+            self.user_wallets[id] += mem//2
+        
         block = {
             'index': len(self.chain) + 1,
             'timestamp': time(),
@@ -41,6 +44,7 @@ class Blockchain:
         
         return: 해당 트랜잭션을 포함한 블록의 인덱스
         """
+        
         transaction = {
             'id': id,
             'kind': kind,
@@ -57,6 +61,8 @@ class Blockchain:
             else:
                 # inference 요청 시 차감
                 self.user_wallets[id] -= 1
+                if self.user_wallets[id]<0:
+                    self.user_wallets[id] = 0
         elif kind == "add":
             if id not in self.user_gpus:
                 self.user_gpus[id] = int(data)