Update app.py

- Fix API issues

app.py

@@ -35,12 +35,21 @@ def check_flagged_address(address: str) -> str:
         String indicating if the address is flagged and its blockchain.
     """
     chain = infer_chain(address)
+    flagged = False
+    
+    # Check if the address is in the API data
     for entry in flagged_addresses:
         if isinstance(entry, dict) and "address" in entry:
             if entry["address"].lower() == address.lower():
-                chain = entry.get("chain", chain)  # Use API chain if available
-                return f"Address {address} is flagged as part of the Bybit hack on {chain}."
-    return f"Address {address} is not flagged as part of the Bybit hack (inferred chain: {chain})."
+                flagged = True
+                chain = entry.get("chain", chain)  # Override with API-provided chain if available
+                break
+    
+    # Format the response with emojis and clear details
+    if flagged:
+        return f"Address {address} is Flagged ✅\nChain: {chain}\nStatus: Present in Bybit hack database"
+    else:
+        return f"Address {address} is Not Flagged ❌\nChain: {chain}\nStatus: Not present in Bybit hack database"
 
 # Gradio chat function
 def chat_with_agent(user_input, chat_history, hf_token):
@@ -128,7 +137,7 @@ with gr.Blocks(title="Bybit Hack Address Checker") as demo:
                 3. **Agent Processing**: The app uses a `CodeAgent` from the `smolagents` library, powered by the free `Mixtral-8x7B-Instruct-v0.1` model via the Hugging Face Inference API.
                 4. **Tool Usage**: The agent calls a custom tool (`check_flagged_address`) to compare your input against a list of flagged addresses from the Elliptic Bybit hack dataset (`https://dt074px2e9qbh.cloudfront.net/exploit-api.json`).
                 5. **Chain Identification**: If the address isn’t explicitly tagged with a blockchain in the dataset, the app infers it (Ethereum, Bitcoin, Tron, or Unknown) based on common address formats.
-                6. **Result**: You’ll see whether the address is flagged and its inferred or confirmed blockchain in the chat interface.
+                6. **Result**: You’ll see whether the 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.