BlockchainCoinSafe/app.py at main · Shiv-D-Coder/BlockchainCoinSafe · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
import streamlit as st
from token_wallet import TokenWallet
from visualizations import visualize_blockchain, visualize_merkle_tree, visualize_transaction
import networkx as nx
import matplotlib.pyplot as plt

# Initialize session state
if 'wallet' not in st.session_state:
    st.session_state.wallet = TokenWallet()

def main():
    st.set_page_config(page_title="TokenWallet App", page_icon="💰", layout="wide")

    st.title("🏦 TokenWallet App with Blockchain")

    # Sidebar for wallet creation and selection
    with st.sidebar:
        st.header("Wallet Management")
        new_wallet = st.text_input("Create a new wallet", key="new_wallet")
        if st.button("Create Wallet"):
            result = st.session_state.wallet.create_wallet(new_wallet)
            st.success(result)

        st.subheader("Select Wallet")
        wallet_options = list(st.session_state.wallet.wallets.keys())
        if wallet_options:
            selected_wallet = st.selectbox("Choose a wallet", options=wallet_options)
        else:
            st.info("No wallets created yet. Create a wallet to get started.")
            return

    # Main area
    st.subheader(f"💼 Wallet: {selected_wallet}")
    balance = st.session_state.wallet.get_balance(selected_wallet)
    if balance is not None:
        st.metric("Balance", f"{balance} tokens")
    else:
        st.error("Wallet not found!")

    col1, col2 = st.columns(2)

    with col1:
        st.subheader("📥 Receive Tokens")
        st.info("The 'Receive Tokens' feature simulates receiving tokens from an external source.")
        receive_amount = st.number_input("Amount to receive", min_value=0, step=1, key="receive")
        receive_label = st.text_input("Label for transaction (optional)", key="receive_label")
        if st.button("Receive"):
            result = st.session_state.wallet.receive_token(selected_wallet, receive_amount, receive_label)
            st.success(result)

    with col2:
        st.subheader("📤 Send Tokens")
        receiver_options = [w for w in wallet_options if w != selected_wallet]
        if receiver_options:
            receiver = st.selectbox("Select receiver", options=receiver_options)
            send_amount = st.number_input("Amount to send", min_value=0, step=1, key="send")
            send_label = st.text_input("Label for transaction (optional)", key="send_label")
            if st.button("Send"):
                result = st.session_state.wallet.send_token(selected_wallet, receiver, send_amount, send_label)
                if "Sent" in result:
                    st.success(result)
                else:
                    st.error(result)
        else:
            st.warning("No other wallets available to send tokens to.", icon="⚠️")

    # Transaction History Section
    st.subheader("📝 Transaction History")
    if st.button("Download Transaction History"):
        transaction_history_text = st.session_state.wallet.get_transaction_history(selected_wallet)
        st.download_button(
            label="Download Transaction History",
            data=transaction_history_text,
            file_name=f"{selected_wallet}_transaction_history.txt",
            mime="text/plain"
        )

    # Proof of Work (PoW) Section
    st.subheader("🔨 Proof of Work Simulation")
    st.info("Adjust the difficulty level of the mining process.")

    difficulty = st.slider("Set Mining Difficulty (Number of leading zeros)", 1, 5, value=2)
    if st.button("Set Difficulty"):
        result = st.session_state.wallet.set_difficulty(difficulty)
        st.info(result)

    # Blockchain Visualization
    st.subheader("🔗 Blockchain Visualization")
    blockchain = st.session_state.wallet.get_blockchain()
    if blockchain:
        visualize_blockchain(blockchain)

        # Merkle Tree Visualization for the entire blockchain
        st.subheader("🌳 Merkle Tree Visualization (All Transactions)")
        merkle_tree = st.session_state.wallet.get_merkle_tree()
        if merkle_tree:
            visualize_merkle_tree(merkle_tree)
        else:
            st.info("No transactions in the blockchain yet.")

        # Transaction Visualization
        st.subheader("💸 Latest Transaction Visualization")
        if blockchain[-1]['transactions']:
            transaction = blockchain[-1]['transactions'][0]  # Visualize the latest transaction
            visualize_transaction(transaction)
        else:
            st.info("No transactions in the latest block.")

        # Download button for blockchain data


# Download button for blockchain data
        blockchain_text = "\n".join([f"Block {block['index']} - Timestamp: {block['timestamp']}, Transactions: {block['transactions']}, "
            f"Previous Hash: {block['previous_hash']}, Nonce: {block['nonce']}, Hash: {block['hash']}"
            for block in blockchain
        ])
        st.download_button(
            label="Download Blockchain Data",
            data=blockchain_text,
            file_name="blockchain_data.txt",
            mime="text/plain",
        )


    else:
        st.warning("Blockchain is empty.", icon="⚠️")

if __name__ == "__main__":
    main()