Update app.py

- Fix API Errors

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import requests
-from smolagents import CodeAgent, tool, HfApiModel
+from transformers import pipeline
 
 # Fetch the API data once at startup
 API_URL = "https://dt074px2e9qbh.cloudfront.net/exploit-api.json"
@@ -11,41 +11,26 @@ except Exception as e:
     flagged_addresses = []
     print(f"Failed to load API data: {e}")
 
-# Function to infer chain from address (basic heuristics)
+# Load LLM for blockchain inference
+llm_pipeline = pipeline("text-generation", model="gpt-3.5-turbo")  # You can use a larger or more appropriate model
+
+# Function to infer chain using LLM
 def infer_chain(address):
-    address = address.lower()
-    if address.startswith("0x") and len(address) == 42:
-        return "Ethereum"
-    elif address.startswith("bc1") or address.startswith("1") or address.startswith("3"):
-        return "Bitcoin"
-    elif address.startswith("T") and len(address) == 34:
-        return "Tron"
-    else:
-        return "Unknown"
+    prompt = f"Identify the blockchain of the following cryptocurrency address: {address}"
+    result = llm_pipeline(prompt, max_length=50)
+    return result[0]["generated_text"].strip()
 
-# Define the tool with a strictly formatted docstring
-@tool
-def check_flagged_address(address: str) -> str:
-    """Check if a cryptocurrency address is flagged in the Bybit hack database.
+# Function to check if the address is flagged
+def check_flagged_address(address: str):
+    """Check if a cryptocurrency address is flagged in the Bybit hack database."""
     
-    Args:
-        address: String representing the cryptocurrency address to check.
-    
-    Returns:
-        String indicating if the address is flagged and its blockchain.
-    """
+    # Infer the chain of the address
     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():
-                flagged = True
-                chain = entry.get("chain", chain)  # Override with API-provided chain if available
-                break
+    flagged = any(entry.get("address", "").lower() == address.lower() for entry in flagged_addresses)
     
-    # Format the response with emojis and clear details
+    # Format the response
     if flagged:
         return f"Address {address} is Flagged ✅\nChain: {chain}\nStatus: Present in Bybit hack database"
     else:
@@ -62,17 +47,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, ""
     
-    # Initialize the CodeAgent with the user-provided HF token
-    agent = CodeAgent(
-        model=HfApiModel(model_id="mistralai/Mixtral-8x7B-Instruct-v0.1", token=hf_token),
-        tools=[check_flagged_address]
-    )
-    
-    # Run the agent with the user's input
-    try:
-        response = agent.run(user_input)
-    except Exception as e:
-        response = f"Error: {str(e)}. Check your HF token or try again."
+    # Run the address checker
+    response = check_flagged_address(user_input)
     
     # Append as OpenAI-style messages
     chat_history.append({"role": "user", "content": user_input})
@@ -134,18 +110,17 @@ with gr.Blocks(title="Bybit Hack Address Checker") as demo:
 
                 1. **Input an Address**: Enter a wallet address (e.g., Ethereum `0x...`, Bitcoin `bc1q...`, Tron `T...`) in the 'Check Address' tab.
                 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 `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 ✅ 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 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.
 
                 ### Why It’s Useful
                 - **Security**: Quickly check if an 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 `smolagents`.
+                - **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!
             """)
 
 # Launch the app
-demo.launch()
+demo.launch()