api-engine.py

import os
import time
import requests
import json
import ollama
from dotenv import load_dotenv
from web3 import Web3
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware

app = FastAPI(title="GHOSTFUND")
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"], # In production, change this to your Vercel domain
    allow_methods=["*"],
    allow_headers=["*"],
)

@app.get("/")
def home():
    return {"status": "GHOSTFUND AI is Online", "version": "1.0.0"}

load_dotenv()


print("🤖 INITIALIZING TRIDENT HEDGE FUND ENGINE...")

# A. CREDENTIALS
PRIVATE_KEY = os.getenv("PRIVATE_KEY")
BASESCAN_KEY = os.getenv("BASE_SCAN")
CRYPTOPANIC_KEY = os.getenv("CRYPTOPANIC_API_KEY")
print("✅ CREDENTIALS LOADED.")
print(f"      - BaseScan Key: {BASESCAN_KEY[:4]}...{BASESCAN_KEY[-4:]}")
print(f"      - CryptoPanic Key: {CRYPTOPANIC_KEY[:4]}...{CRYPTOPANIC_KEY[-4:]}")

# B. BLOCKCHAIN CONNECTION (Base Sepolia for Trading)
RPC_URL = os.getenv("RPC_URL")
w3 = Web3(Web3.HTTPProvider(RPC_URL))
account = w3.eth.account.from_key(PRIVATE_KEY)
my_address = account.address
print(f"🔌 WALLET CONNECTED: {my_address}")




# C. ASSETS
USDC_MAINNET = "0x833589fCD6eDb6E08f4c7C32D4f71b54bdA02913" # Watching Real Whales
WETH_TESTNET = "0x4200000000000000000000000000000000000006" # Trading Fake ETH
USDC_TESTNET = "0x036CbD53842c5426634e7929541eC2318f3dCF7e" # Trading Fake USDC
ROUTER_ADDRESS = "0x94cC0AaC535CCDB3C01d6787D6413C739ae12bc4" # Uniswap Router


router_abi = [
    {
        "inputs": [
            {
                "components": [
                    {"internalType": "address", "name": "tokenIn", "type": "address"},
                    {"internalType": "address", "name": "tokenOut", "type": "address"},
                    {"internalType": "uint24", "name": "fee", "type": "uint24"},
                    {"internalType": "address", "name": "recipient", "type": "address"},
                    {"internalType": "uint256", "name": "amountIn", "type": "uint256"},
                    {"internalType": "uint256", "name": "amountOutMinimum", "type": "uint256"},
                    {"internalType": "uint160", "name": "sqrtPriceLimitX96", "type": "uint160"}
                ],
                "internalType": "struct IV3SwapRouter.ExactInputSingleParams",
                "name": "params",
                "type": "tuple"
            }
        ],
        "name": "exactInputSingle",
        "outputs": [{"internalType": "uint256", "name": "amountOut", "type": "uint256"}],
        "stateMutability": "payable",
        "type": "function"
    }
]
print(f"🔍 Loaded Router ABI. {len(router_abi)} items")

# Create the Contract Object
router_contract = w3.eth.contract(address=ROUTER_ADDRESS, abi=router_abi)


def get_ai_sentiment():
    print("\n[1/3] 🧠 AI ANALYST (Fundamental Analysis)")
    
    # Using the V2 Developer URL structure you confirmed earlier
    url = f"https://cryptopanic.com/api/developer/v2/posts/?auth_token={CRYPTOPANIC_KEY}&kind=news&filter=rising&public=true"
    
    try:
        response = requests.get(url, headers={"User-Agent": "TridentBot/1.0"})
        
      
        if response.status_code != 200:
            print(f"      ❌ API HTTP ERROR: {response.status_code}")
            print(f"      ❌ MSG: {response.text[:100]}...") # Print first 100 chars
            return 0

        data = response.json()
        
        if 'results' not in data:
            print(f"      ⚠️ UNEXPECTED DATA FORMAT: {data.keys()}")
            return 0
            
        headlines = [f"- {p['title']}" for p in data['results'][:5]]
        
        if not headlines:
            print("      ⚠️ No rising news found.")
            return 0

        news_block = "\n".join(headlines)
        print(f"      📖 Reading {len(headlines)} headlines...")

        # Analyze with Llama
        prompt = f"""
        Analyze these crypto headlines and output a JSON with a 'score' (-100 to 100).
        Headlines: {news_block}
        Output ONLY JSON: {{"score": 0, "reason": "..."}}
        """
        
        ai_res = ollama.chat(model='llama3.2', messages=[{'role': 'user', 'content': prompt}])
        
        # Clean AI Output (Find JSON)
        content = ai_res['message']['content']
        start, end = content.find('{'), content.rfind('}') + 1
        result = json.loads(content[start:end])
        
        score = int(result['score'])
        print(f"      ✅ AI VERDICT: {score}/100 ({result.get('reason', 'No reason')})")
        return score

    except Exception as e:
        print(f"      ❌ AI PROCESSING FAILED: {e}")
        return 0 

# ==============================================================================
# 3. SIGNAL 2: WHALE WATCHER (On-Chain Data)
# ==============================================================================
def get_whale_score():
    print("[2/3] 🐳 WHALE WATCHER (Omnichain V2 Analysis)")
    BASE_CHAIN_ID = 8453
    
    url = f"https://api.etherscan.io/v2/api?chainid={BASE_CHAIN_ID}&module=account&action=tokentx&contractaddress={USDC_MAINNET}&page=1&offset=20&sort=desc&apikey={BASESCAN_KEY}"
    
    # RETRY LOGIC: Try 3 times before failing
    max_retries = 3
    for attempt in range(max_retries):
        try:
            res = requests.get(url).json()
            
            # Check for Success
            if res['status'] == '1':
                # SUCCESS! Process the data
                transfers = res.get('result', [])
                whale_moves = 0
                volume = 0
                for tx in transfers:
                    amount = int(tx['value']) / 1_000_000
                    if amount > 100_000: 
                        whale_moves += 1
                        volume += amount
                
                print(f"      found {whale_moves} Whales moving ${volume:,.0f} USDC.")
                
                if whale_moves > 3: return -50
                if whale_moves > 0: return 25
                return 50 
            
            else:
                # FAILED (NOTOK). Print why and wait.
                print(f"      ⚠️ Attempt {attempt+1}/{max_retries} Failed: {res['message']}")
                if attempt < max_retries - 1:
                    time.sleep(2) # Wait 2 seconds and try again
                    continue

        except Exception as e:
            print(f"      ❌ Connection Error: {e}")
            time.sleep(2)
            
    # If we get here, all 3 attempts failed
    print("      ❌ WHALE WATCH FAILED (Using Neutral Score)")
    return 0
# ==============================================================================
# 4. SIGNAL 3: MARKET MOOD (Fear & Greed API)
# ==============================================================================
def get_market_mood():
    print("[3/3] 📊 MARKET MOOD (Technical Analysis)")
    try:
        res = requests.get("https://api.alternative.me/fng/")
        data = res.json()['data'][0]
        value = int(data['value'])
        print(f"      Global Sentiment: {value}/100 ({data['value_classification']})")
        return (value - 50) * 2
    except:
        return 0
# ==============================================================================
# 5. EXECUTION ENGINE (BIDIRECTIONAL)
# ==============================================================================

@app.get("/signal")
def get_trading_signal():
    print("\n📡 INCOMING REQUEST: Calculating Signal...")
    
    # 1. Gather Signals (Re-using your existing functions)
    s1_ai = get_ai_sentiment()
    s2_whale = get_whale_score()
    s3_mood = get_market_mood()
    
    # 2. Calculate Weighted Score
    final_score = (s1_ai * 0.4) + (s2_whale * 0.3) + (s3_mood * 0.3)
    
    # 3. Determine Recommendation
    decision = "HOLD"
    if final_score > 20: decision = "BUY" # Lower threshold for demo
    if final_score < -20: decision = "SELL"
    
    response = {
        "score": round(final_score, 2),
        "decision": decision,
        "breakdown": {
            "ai_sentiment": s1_ai,
            "whale_activity": s2_whale,
            "market_mood": s3_mood
        }
    }
    
    print(f"✅ RESPONSE SENT: {decision} (Score: {final_score})")
    return response


# ==============================================================================