Update app.py

app.py

@@ -12,7 +12,8 @@ import hashlib
 import BlockChain
 
 loop = asyncio.get_event_loop()
-# Monkey patch
+
+# init code
 def get_types(cls_set: List[Type], component: str):
     docset = []
     types = []
@@ -43,6 +44,8 @@ name_list['id'] = 'name'
 p2p_list = dict()
 p2p_list['id'] = '11111111'
 
+gpu_add_list = []
+
 def chat(id, prompt):
     
     return "AI 응답입니다."
@@ -100,6 +103,7 @@ def get_id_from_p2p(i):
 
 # Blockchain code
 
+
 def get_peers(name):
     data = requests.get("https://health.petals.dev/api/v1/state").json()
     out = []
@@ -109,13 +113,24 @@ def get_peers(name):
                 out.append(r['peer_id'])
     return out
 
+blockchain = Blockchain()
+
+def add_transaction(id, kind, data):
+    blockchain.new_transaction(id, kind, data)
+
+def proof(name):
+    peers = get_peers(name)
+    for p in gpu_add_list:
+        if not p in peers:
+            add_transaction(get_id_from_p2p(peer), "out", 0)
+
 with gr.Blocks() as demo:
     count = 0
     aa = gr.Interface(
       fn=chat,
       inputs=["text","text"],
       outputs="text",
-      description="chat, ai 응답을 반환합니다. \n /run/predict",
+      description="chat, ai 응답을 반환합니다. 내부적으로 트랜잭션 생성. \n /run/predict",
     )
 
     rr = gr.Interface(
@@ -145,40 +160,14 @@ with gr.Blocks() as demo:
       outputs="text",
       description="get_name, id로 닉네임 반환(없으면 no id)\n /run/predict_4",
     )
-
-    nnn = gr.Interface(
-      fn=get_id,
-      inputs=["text"],
-      outputs="text",
-      description="get_name, 닉네임으로 id 반환(없으면 no name)\n /run/predict_5",
-    )
     
     adp = gr.Interface(
       fn=add_p,
       inputs=["text", "text"],
       outputs="text",
-      description="add_p, id로 p2p id 추가. ok 반환. \n /run/predict_6",
-    )
-    
-    nnp = gr.Interface(
-      fn=get_p,
-      inputs=["text"],
-      outputs="text",
-      description="get_p, id로 p2p id 반환. 없으면 no id. \n /run/predict_7",
+      description="add_p, id로 p2p id 추가. ok 반환. \n /run/predict_5",
     )
 
-    nnp = gr.Interface(
-      fn=get_id_from_p2p,
-      inputs=["text"],
-      outputs="text",
-      description="get_p, p2p id로 일반 id 반환. 없으면 no id. \n /run/predict_8",
-    )
 
-    gpeer = gr.Interface(
-      fn=get_peers,
-      inputs=["text"],
-      outputs="text",
-      description="get_peers, 해당 모델을 분산처리중인 peer들의 p2p id list 반환\n /run/predict_8",
-    )
     
     demo.queue(max_size=32).launch(enable_queue=True)