Update app.py

app.py

@@ -1,15 +1,13 @@
-# app.py
-
 import streamlit as st
 import requests
 
 st.title("Verify 👻")
 
-# Fixed Token Mint Address
-TOKEN_MINT_ADDRESS = "PR9SDkf1SxrJqSXzQyoR17VdGZ1ZgWMiRcS5VtYmis1"
-   
-     else:
-         TOKEN_MINT_ADDRESS = "9yT1p7grHJh3uKw3sp9XrQwNtCtAKDa8mqpMW2Y9BLz5"
+# Fixed Token Mint Addresses
+TOKEN_MINT_ADDRESSES = [
+    "PR9SDkf1SxrJqSXzQyoR17VdGZ1ZgWMiRcS5VtYmis1",  # Original mint address
+    "9yT1p7grHJh3uKw3sp9XrQwNtCtAKDa8mqpMW2Y9BLz5"  # Additional mint address
+]
 
 # Input field for the wallet address
 wallet_address = st.text_input("Enter your Solana wallet address:")
@@ -20,52 +18,57 @@ if st.button("Verify"):
             # Solana RPC endpoint
             rpc_endpoint = "https://api.mainnet-beta.solana.com"
 
-            # Prepare the request payload
-            payload = {
-                "jsonrpc": "2.0",
-                "id": 1,
-                "method": "getTokenAccountsByOwner",
-                "params": [
-                    wallet_address,
-                    {"mint": TOKEN_MINT_ADDRESS},
-                    {"encoding": "jsonParsed"}
-                ]
-            }
-
-            # Send the request
-            response = requests.post(rpc_endpoint, json=payload)
-            result = response.json()
-
-            # Check if any token accounts were found
-            if (
-                'result' in result and
-                'value' in result['result'] and
-                len(result['result']['value']) > 0
-            ):
-                # Get the balance (assuming only one account for simplicity)
-                token_account = result['result']['value'][0]
-                balance_info = token_account['account']['data']['parsed']['info']['tokenAmount']
-                amount = int(balance_info['amount'])  # amount in the smallest unit
-
-                # Adjust for token decimals
-                decimals = balance_info['decimals']
-                adjusted_amount = amount / (10 ** decimals)
-
-                # Display the amount the user holds
-                st.write(f"You hold {adjusted_amount:,.0f} tokens.")
-
-                # Compare the adjusted amount to thresholds
-                if adjusted_amount >= 3333333:
-                    st.success("✅ Guaranteed OG Status")
-                elif adjusted_amount >= 333333:
-                    st.success("✅ Guaranteed WLT Status")
-                elif adjusted_amount >= 33333:
-                    st.success("✅ Guaranteed WLR Status")
+            total_adjusted_amount = 0
+
+            for token_mint_address in TOKEN_MINT_ADDRESSES:
+                # Prepare the request payload
+                payload = {
+                    "jsonrpc": "2.0",
+                    "id": 1,
+                    "method": "getTokenAccountsByOwner",
+                    "params": [
+                        wallet_address,
+                        {"mint": token_mint_address},
+                        {"encoding": "jsonParsed"}
+                    ]
+                }
+
+                # Send the request
+                response = requests.post(rpc_endpoint, json=payload)
+                result = response.json()
+
+                # Check if any token accounts were found
+                if (
+                    'result' in result and
+                    'value' in result['result'] and
+                    len(result['result']['value']) > 0
+                ):
+                    # Get the balance (assuming only one account for simplicity)
+                    token_account = result['result']['value'][0]
+                    balance_info = token_account['account']['data']['parsed']['info']['tokenAmount']
+                    amount = int(balance_info['amount'])  # amount in the smallest unit
+
+                    # Adjust for token decimals
+                    decimals = balance_info['decimals']
+                    adjusted_amount = amount / (10 ** decimals)
+
+                    # 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}.")
                 else:
-                    st.error("You don't have enough tokens yet 👻")
+                    st.write(f"No tokens found for mint address {token_mint_address}.")
+
+            # Compare the total adjusted amount to thresholds
+            if total_adjusted_amount >= 333333:
+                st.success("✅ Guaranteed OG Status")
+            elif total_adjusted_amount >= 33333:
+                st.success("✅ Guaranteed WLT Status")
+            elif total_adjusted_amount >= 3333:
+                st.success("✅ Guaranteed WLR Status")
             else:
-                adjusted_amount = 0
-                st.error("No tokens found, try another address.")
+                st.error("You don't have enough tokens yet 👻")
 
             # Log the query to Google Form
             try:
@@ -73,7 +76,7 @@ if st.button("Verify"):
 
                 form_data = {
                     'entry.1721628210': wallet_address,          # Replace with your actual Wallet Address field name
-                    'entry.1357327531': f"{adjusted_amount:,.0f}"  # Replace with your actual Token Amount field name
+                    'entry.1357327531': f"{total_adjusted_amount:,.0f}"  # Replace with your actual Token Amount field name
                 }
 
                 response = requests.post(form_url, data=form_data)