Update Blockchain.py

Blockchain.py

@@ -53,6 +53,7 @@ class Blockchain:
  if id not in self.user_wallets:
  # 새로운 유저인 경우 초기 행동력 설정
  self.user_wallets[id] = 10 # 초기 행동력을 10으로 설정
+ self.user_wallets[id] -= 1
  else:
  # inference 요청 시 차감
  self.user_wallets[id] -= 1
@@ -69,7 +70,7 @@ class Blockchain:
 
  def get_user_balance(self, id):
  # 특정 유저의 행동력을 조회
- return self.user_wallets.get(id, 0)
+ return self.user_wallets.get(id, 10)
 
  def get_user_gpu_mem(self, id):
  # 특정 유저의 기여 gpu를 조회
@@ -81,6 +82,13 @@ class Blockchain:
  for mem in self.user_gpus.values():
  result += int(mem)
  return result
+
+ def get_total_coin(self):
+ # 전체 코인(행동력) 수 반환
+ result = 0
+ for coin in self.user_wallets.values():
+ result += int(coin)
+ return result
  @property
  def last_block(self):
  """