Update app.py

app.py

@@ -3,11 +3,11 @@ import requests
 
 st.title("Verify 👻")
 
-# Fixed Token Mint Addresses
-TOKEN_MINT_ADDRESSES = [
-    "PR9SDkf1SxrJqSXzQyoR17VdGZ1ZgWMiRcS5VtYmis1",  # Original mint address
-    "9yT1p7grHJh3uKw3sp9XrQwNtCtAKDa8mqpMW2Y9BLz5"  # Additional mint address
-]
+# Fixed Token Mint Addresses (hidden from user)
+TOKEN_MINT_ADDRESSES = {
+    "TRICK": "PR9SDkf1SxrJqSXzQyoR17VdGZ1ZgWMiRcS5VtYmis1",  # TRICK token mint address
+    "TREAT": "9yT1p7grHJh3uKw3sp9XrQwNtCtAKDa8mqpMW2Y9BLz5"  # TREAT token mint address
+}
 
 # Input field for the wallet address
 wallet_address = st.text_input("Enter your Solana wallet address:")
@@ -20,7 +20,8 @@ if st.button("Verify"):
 
             total_adjusted_amount = 0
 
-            for token_mint_address in TOKEN_MINT_ADDRESSES:
+            # Loop through token types (TRICK and TREAT)
+            for token_name, token_mint_address in TOKEN_MINT_ADDRESSES.items():
                 # Prepare the request payload
                 payload = {
                     "jsonrpc": "2.0",
@@ -55,10 +56,10 @@ if st.button("Verify"):
                     # Add to total amount
                     total_adjusted_amount += adjusted_amount
 
-                    # Display the amount the user holds for each token
-                    st.write(f"You hold {adjusted_amount:,.0f} tokens for mint address {token_mint_address}.")
+                    # Display the amount the user holds for each token (TRICK or TREAT)
+                    st.write(f"You hold {adjusted_amount:,.0f} {token_name} tokens.")
                 else:
-                    st.write(f"No tokens found for mint address {token_mint_address}.")
+                    st.write(f"No {token_name} tokens found.")
 
             # Compare the total adjusted amount to thresholds
             if total_adjusted_amount >= 333333: