Update app.py

app.py

@@ -45,12 +45,15 @@ def generate_keys():
     add_label = str(address)
     add_label = add_label.strip("b").strip("'")
     address_im = overlay.textover(address_im, "Wallet",add_label)
+    address_im.save("address_im.png")
     #qr_link="test"
-
+    address_im = "address_im.png"
     private_key_im = overlay.textover("private_key.png", "Key","")
-    priv_key = stegan.conv_im(private_key_im,data=private_key)
+    private_key_im.save("private_key_im.png")
+    priv_key = stegan.conv_im("private_key_im.png",data=private_key)
+    
+    pub_key = stegan.conv_im("address_im.png",data=public_key)   
     
-    pub_key = stegan.conv_im(address_im,data=public_key)   
     return public_key,private_key,address_im,address,priv_key,pub_key
 
 def encrypt_text(data,in2,address):