Update app.py

app.py

@@ -382,69 +382,88 @@ def issue_tokens(send,rec,amount):
     #is_valid='True'
     mes = None
     try:
-        r = requests.get(f'{main_balance}/{send}') 
+        r = requests.get(f'{main_balance}/{send}.json') 
         lod = json.loads(r.text)
+        p=True
     except:
         lod=[]
+        p=False
+        mes = "Sender has no wallet"
         pass
-    block = {'index': len(lod) + 1,
-            'timestamp': str(datetime.datetime.now()),
-            'sender': f'{send}',
-            'recipient':f'{rec}',
-             
-            'amount': f'{amount}'}
-    lod.append(block)    
     
-    json_object = json.dumps(lod, indent=4)
-    with open("tmp1.json", "w") as outfile:
-        outfile.write(json_object)
-    try: 
-        api.upload_file(
-        path_or_fileobj="tmp1.json",
-        path_in_repo=f'{main_balance.split("main/",1)[1]}/{send}.json',
-        repo_id=main_balance.split('datasets/',1)[1].split('/raw',1)[0],
-        token=token_self,
-        repo_type="dataset",
-        )
-        os.remove("tmp1.json")
-    except Exception as e:
-        mes=e
-        pass 
-    lod.clear()
 
-    try:
-        r = requests.get(f'{main_balance}/{rec}') 
-        lod = json.loads(r.text)
-    except:
-        lod=[]
-        pass
-    block = {'index': len(lod) + 1,
-            'timestamp': str(datetime.datetime.now()),
-            'sender': f'{send}',
-            'recipient':f'{rec}',
-             
-            'amount': f'{amount}'}
-    lod.append(block)    
     
-    json_object = json.dumps(lod, indent=4)
-    with open("tmp1.json", "w") as outfile:
-        outfile.write(json_object)
-    try: 
-        api.upload_file(
-        path_or_fileobj="tmp1.json",
-        path_in_repo=f'{main_balance.split("main/",1)[1]}/{rec}.json',
-        repo_id=main_balance.split('datasets/',1)[1].split('/raw',1)[0],
-        token=token_self,
-        repo_type="dataset",
-        )
-        os.remove("tmp1.json")
-
-    except Exception as e:
-        mes = e
-        pass   
-    if mes == None:
-        mes = f'{amount} sent from {send} to {rec}'
-    return mes
+    
+    if p==True:
+    
+        balance = lod[-1]["balance"]
+        print (balance)
+        balance =int(balance)-int(amount)
+        if balance >=0:
+            block = {'index': len(lod) + 1,
+                    'timestamp': str(datetime.datetime.now()),
+                    'sender': f'{send}',
+                    'recipient':f'{rec}',
+                    'amount': f'{amount}'},
+                    'balance': f'{balance}'
+            lod.append(block)    
+            
+            json_object = json.dumps(lod, indent=4)
+            with open("tmp1.json", "w") as outfile:
+                outfile.write(json_object)
+            try: 
+                api.upload_file(
+                path_or_fileobj="tmp1.json",
+                path_in_repo=f'{main_balance.split("main/",1)[1]}/{send}.json',
+                repo_id=main_balance.split('datasets/',1)[1].split('/raw',1)[0],
+                token=token_self,
+                repo_type="dataset",
+                )
+                os.remove("tmp1.json")
+            except Exception as e:
+                mes=e
+                pass 
+    
+            try:
+                r = requests.get(f'{main_balance}/{rec}.json') 
+                lod = json.loads(r.text)
+            except:
+                lod=[]
+                pass
+            balance = lod[-1]["balance"]
+            print (balance)
+            balance =int(balance)+int(amount)                
+            block = {'index': len(lod) + 1,
+                    'timestamp': str(datetime.datetime.now()),
+                    'sender': f'{send}',
+                    'recipient':f'{rec}',
+                    'amount': f'{amount}',
+                    'balance': f'{balance}'}
+            lod.append(block)    
+            
+            json_object = json.dumps(lod, indent=4)
+            with open("tmp1.json", "w") as outfile:
+                outfile.write(json_object)
+            try: 
+                api.upload_file(
+                path_or_fileobj="tmp1.json",
+                path_in_repo=f'{main_balance.split("main/",1)[1]}/{rec}.json',
+                repo_id=main_balance.split('datasets/',1)[1].split('/raw',1)[0],
+                token=token_self,
+                repo_type="dataset",
+                )
+                os.remove("tmp1.json")
+        
+            except Exception as e:
+                mes = e
+                pass   
+            if mes == None:
+                mes = f'{amount} sent from {send} to {rec}'
+        if balance < 0:
+            mes ="Not enough tokens"
+            p = False
+            
+    return mes,p
     
 with gr.Blocks() as bc:
     with gr.Row(visible=True) as invalid:
@@ -486,7 +505,7 @@ with gr.Blocks() as bc:
         with gr.Accordion("Tokens", open=False):
             issue_btn=gr.Button()
 
-    issue_btn.click(issue_tokens,[send,rec,am],block_text)
+    issue_btn.click(issue_tokens,[send,rec,am],[block_text,None])
     
     node_add.click(add_node,[this_space,node_repo,node_space,node_file],block_text)
     pass_btn.click(checkp,pass_box,[invalid,valida])