Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,6 +2,13 @@ import streamlit as st
 import requests
 import json
 import web3
+import os
+
+PINATA_JWT = os.environ.get("PinataJWT")
+INFURA_URL = os.environ.get("infura")
+CONTRACT_ADDRESS = os.environ.get("ContractAddress")
+ETH_WALLET = os.environ.get("EthWallet")
+PRIVATE_KEY = os.environ.get("pk")
 
 abi = [{"anonymous":False,"inputs":[{"indexed":False,"internalType":"string","name":"ipfsHash","type":"string"}],"name":"Store","type":"event"},{"inputs":[],"name":"getHashes","outputs":[{"internalType":"string[]","name":"","type":"string[]"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"string","name":"ipfsHash","type":"string"}],"name":"storeHash","outputs":[],"stateMutability":"nonpayable","type":"function"}]
 # function that uploads the results to ipfs
@@ -16,7 +23,7 @@ def upload_json_to_ipfs(data):
             st.write("No JWT secret found, please add your JWT in a secret titled \"PinataJWT\"")
             return
 
-        jwt_token = st.secrets["PinataJWT"].strip()
+        jwt_token = PINATA_JWT.strip()
         headers = {
             "Authorization": f"Bearer {jwt_token}",
             "Content-Type": "application/json"
@@ -39,18 +46,18 @@ def upload_json_to_ipfs(data):
 # function that uploads to blockchain
 def upload_to_blockchain(hash):
     print("starting blockchain upload")
-    w3 = web3.Web3(web3.HTTPProvider(st.secrets["infura"]))
+    w3 = web3.Web3(web3.HTTPProvider(INFURA_URL))
 
     # create an instance of our contract
-    contract = w3.eth.contract(address=st.secrets["ContractAddress"], abi = abi)
+    contract = w3.eth.contract(address=CONTRACT_ADDRESS, abi = abi)
 
     # Call your function: 11155111 is Sepolia's id
     call_function = contract.functions.storeHash(hash).build_transaction({"chainId": 11155111, 
-                                                                            "from": st.secrets["EthWallet"],
-                                                                            "nonce": w3.eth.get_transaction_count(st.secrets["EthWallet"])})
+                                                                            "from": ETH_WALLET,
+                                                                            "nonce": w3.eth.get_transaction_count(ETH_WALLET)})
 
     # Sign transaction
-    signed_tx = w3.eth.account.sign_transaction(call_function, private_key=st.secrets["pk"])
+    signed_tx = w3.eth.account.sign_transaction(call_function, private_key=PRIVATE_KEY)
 
     # Send transaction
     send_tx = w3.eth.send_raw_transaction(signed_tx.raw_transaction)
@@ -64,10 +71,10 @@ def upload_to_blockchain(hash):
     return tx_receipt.logs[0].transactionHash.hex()
 
 def get_ipfs_hashes():
-    w3 = web3.Web3(web3.HTTPProvider(st.secrets["infura"]))
+    w3 = web3.Web3(web3.HTTPProvider(INFURA_URL))
 
     # Create an instance of the contract
-    contract = w3.eth.contract(address=st.secrets["ContractAddress"], abi=abi)
+    contract = w3.eth.contract(address=CONTRACT_ADDRESS, abi=abi)
 
     # Call the getHashes function
     try: