Update app.py

app.py

@@ -46,10 +46,6 @@ p2p_list['id'] = '11111111'
 
 gpu_add_list = []
 
-def chat(id, npc, prompt):
-    
-    return "AI 응답입니다."
-
 
 def register(id, pw):
     if id in account_list:
@@ -104,11 +100,12 @@ def get_id_from_p2p(i):
 # Blockchain code
 
 
-def get_peers(name):
+model_name = "petals-team/StableBeluga2"
+def get_peers(model_name):
     data = requests.get("https://health.petals.dev/api/v1/state").json()
     out = []
     for d in data['model_reports']:
-        if d['name'] == "petals-team/StableBeluga2":
+        if d['name'] == model_name:
             for r in d['server_rows']:
                 out.append(r['peer_id'])
     return out
@@ -116,13 +113,17 @@ def get_peers(name):
 blockchain = Blockchain()
 
 def add_transaction(id, kind, data):
-    blockchain.new_transaction(id, kind, data)
+    if kind == "add" or kind == "inference" or kind == "out":
+        blockchain.new_transaction(id, kind, data)
+        return "ok"
+    else:
+        return "fail"
 
-def proof(name):
-    peers = get_peers(name)
+def proof(model_name):
+    peers = get_peers(model_name)
     for p in gpu_add_list:
         if not p in peers:
-            add_transaction(get_id_from_p2p(peer), "out", 0)
+            add_transaction(get_id_from_p2p(peer), "out", 0)    
 
 def get_coin(id):
     c = blockchain.get_user_balance(id)
@@ -141,6 +142,21 @@ def get_data():
 
     return output
 
+
+def chat(id, npc, prompt):
+    if blockchain.get_coin(id) == 0:
+        return "no coin"
+    
+    # model inference
+    output = "AI 응답입니다."
+    add_transaction(id, "inference", {"prompt":prompt, "output":output})
+
+    if len(blockchain.current_transactions)>=10:
+        proof(model_name)        
+        new_block = blockchain.new_block()      
+    return output
+
+
 with gr.Blocks() as demo:
     count = 0
     aa = gr.Interface(
@@ -185,18 +201,25 @@ with gr.Blocks() as demo:
       description="add_p, id로 p2p id 추가. ok 반환. \n /run/predict_5",
     )
 
+    addp = gr.Interface(
+      fn=add_transaction,
+      inputs=["text", "text", "text"],
+      outputs="text",
+      description="add_transaction  \n /run/predict_6",
+    )
+
     gc = gr.Interface(
       fn=get_coin,
       inputs=["text"],
       outputs="text",
-      description="get_coin, id로 잔여 코인(행동력) 반환. \n /run/predict_6",
+      description="get_coin, id로 잔여 코인(행동력) 반환. \n /run/predict_7",
     )
 
     gd = gr.Interface(
       fn=get_data,
       inputs=[],
       outputs="text",
-      description="get_data, 시각화용 모든 데이터 반환. gpu기여 목록, total/used, chain \n /run/predict_7",
+      description="get_data, 시각화용 모든 데이터 반환. gpu기여 목록, total/used, chain \n /run/predict_8",
     )
     
     demo.queue(max_size=32).launch(enable_queue=True)