Update app.py

app.py

@@ -5,8 +5,8 @@ import requests
 API_URL = "https://dt074px2e9qbh.cloudfront.net/exploit-api.json"
 try:
     response = requests.get(API_URL)
-    flagged_addresses = response.json()  # Load the JSON data into a list of dictionaries
-    # Ensure the data is a list of dictionaries (addresses)
+    flagged_addresses = response.json()  # Load the JSON data into a list of strings or dictionaries
+    # Ensure the data is a list of strings or dictionaries (addresses)
     if not isinstance(flagged_addresses, list):
         flagged_addresses = []
 except Exception as e:
@@ -47,8 +47,13 @@ def check_flagged_address(address: str, hf_token: str):
     # Check if the address is in the flagged addresses
     flagged = False
     for entry in flagged_addresses:
-        if isinstance(entry, dict) and "address" in entry:
-            # Normalize the flagged address (lowercase) and check for match
+        # Check if entry is a string (address)
+        if isinstance(entry, str):
+            if entry.strip().lower() == normalized_address:
+                flagged = True
+                break
+        # Check if entry is a dictionary (with "address" key)
+        elif isinstance(entry, dict) and "address" in entry:
             if entry["address"].strip().lower() == normalized_address:
                 flagged = True
                 break