main.py

from fastapi import FastAPI, HTTPException
from fastapi.staticfiles import StaticFiles
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
from dotenv import load_dotenv
import re
import os

load_dotenv()

from solana_utils import fetch_wallet_data
from analyzer import analyze_wallet

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_methods=["*"],
    allow_headers=["*"],
)

class WalletRequest(BaseModel):
    address: str

def is_valid_solana_address(addr: str) -> bool:
    return bool(re.match(r'^[1-9A-HJ-NP-Za-km-z]{32,44}$', addr))

@app.post("/analyze")
async def analyze(req: WalletRequest):
    if not is_valid_solana_address(req.address):
        raise HTTPException(status_code=400, detail="Invalid Solana wallet address")
    try:
        wallet_data = fetch_wallet_data(req.address)
        analysis = analyze_wallet(wallet_data)
        return {"analysis": analysis, "wallet_data": wallet_data}
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))

app.mount("/", StaticFiles(directory="static", html=True), name="static")