Update app.py

app.py

@@ -60,9 +60,27 @@ def check_flagged_address(address: str, hf_token: str):
     
     # Format the response
     if flagged:
-        return f"Address {address} is Flagged ✅\nChain: {chain}\nStatus: Present in Bybit hack database"
+        return f"Address {address} is Flagged ✅\nChain: {chain}\nStatus: Present in Bybit Blocked List"
     else:
-        return f"Address {address} is Not Flagged ❌\nChain: {chain}\nStatus: Not present in Bybit hack database"
+        return f"Address {address} is Not Flagged ❌\nChain: {chain}\nStatus: Not present in Bybit Blocked List"
+
+# Function to handle multiple addresses input
+def handle_multiple_addresses(user_input, hf_token):
+    """Process multiple addresses separated by commas."""
+    results = []
+    
+    # Split input into multiple addresses
+    addresses = [addr.strip() for addr in user_input.split(",")]
+    
+    # Process each address individually
+    for address in addresses:
+        if address:  # Skip empty inputs
+            result = check_flagged_address(address, hf_token)
+            results.append(result)
+        else:
+            results.append("Empty address entered, please try again.")
+    
+    return "\n\n".join(results)  # Combine results with two newlines
 
 # Gradio chat function
 def chat_with_agent(user_input, chat_history, hf_token):
@@ -81,8 +99,8 @@ def chat_with_agent(user_input, chat_history, hf_token):
         chat_history.append({"role": "assistant", "content": "Please enter a valid Hugging Face API token."})
         return chat_history, ""
     
-    # Run the address checker
-    response = check_flagged_address(user_input, hf_token)
+    # Handle multiple addresses
+    response = handle_multiple_addresses(user_input, hf_token)
     
     # Append as OpenAI-style messages
     chat_history.append({"role": "user", "content": user_input})
@@ -92,13 +110,13 @@ def chat_with_agent(user_input, chat_history, hf_token):
     return chat_history, ""
 
 # Create the Gradio interface with tabs
-with gr.Blocks(title="Bybit Hack Address Checker") as demo:
-    gr.Markdown("# Bybit Hack Address Checker")
+with gr.Blocks(title="Bybit Blocked Address Check") as demo:
+    gr.Markdown("# Bybit Blocked Address Check")
     
     with gr.Tabs():
         # Tab 1: Address Checker UI
         with gr.Tab(label="Check Address"):
-            gr.Markdown("Enter a cryptocurrency address to check if it's flagged in the Bybit hack database and identify its blockchain.")
+            gr.Markdown("Enter one or more cryptocurrency addresses (separated by commas) to check if they're flagged in the Bybit hack database.")
             gr.Markdown("Provide your Hugging Face API token below (get it from [huggingface.co/settings/tokens](https://huggingface.co/settings/tokens)).")
             
             # HF Token input
@@ -114,7 +132,7 @@ with gr.Blocks(title="Bybit Hack Address Checker") as demo:
             
             # Address input
             msg = gr.Textbox(
-                placeholder="Enter address here (e.g., 0x123..., bc1q...)",
+                placeholder="Enter address(es) here (separated by commas)",
                 label="Address",
                 lines=1
             )
@@ -142,18 +160,18 @@ with gr.Blocks(title="Bybit Hack Address Checker") as demo:
                 ## What This App Does
                 This application helps you verify if a cryptocurrency address is associated with the Bybit hack, as tracked by the Elliptic dataset. Here’s how it works:
 
-                1. **Input an Address**: Enter a wallet address (e.g., Ethereum `0x...`, Bitcoin `bc1q...`, Tron `T...`) in the 'Check Address' tab.
+                1. **Input one or more Addresses**: Enter one or more wallet addresses (separated by commas).
                 2. **Provide HF Token**: Supply a Hugging Face API token to access the AI model powering the agent.
-                3. **Agent Processing**: The app uses a **language model** to infer the blockchain of the address (Ethereum, Bitcoin, Tron, etc.).
-                4. **Flag Check**: It then checks the address against a list of flagged addresses from the Elliptic Bybit hack dataset (`https://dt074px2e9qbh.cloudfront.net/exploit-api.json`).
-                5. **Result**: You’ll see whether the address is flagged ✅ or not ❌, its chain type, and if it’s present in the database.
+                3. **Agent Processing**: The app uses a **language model** to infer the blockchain of each address (Ethereum, Bitcoin, Tron, etc.).
+                4. **Flag Check**: It then checks each address against a list of flagged addresses from the Elliptic Bybit hack dataset (`https://dt074px2e9qbh.cloudfront.net/exploit-api.json`).
+                5. **Result**: You’ll see whether each address is flagged ✅ or not ❌, its chain type, and if it’s present in the database.
 
                 ### Why It’s Useful
-                - **Security**: Quickly check if an address you’re dealing with is linked to a known exploit.
+                - **Security**: Quickly check if any address you're dealing with is linked to a known exploit.
                 - **Free & Open**: Uses free-tier AI and public data, making it accessible to anyone with an HF account.
                 - **Educational**: Demonstrates AI-driven blockchain analysis with a large language model.
 
-                Get started by entering your HF token and an address in the 'Check Address' tab!
+                Get started by entering your HF token and addresses in the 'Check Address' tab!
             """)
 
 # Launch the app