How to Track GEO Visibility with a Search API

import os, requests, json, sqlite3
from datetime import datetime

API_KEY = os.environ['SCAVIO_API_KEY']
SH = {'x-api-key': API_KEY, 'Content-Type': 'application/json'}

BRAND_TERMS = ['scavio', 'scavio.dev', 'scavio api']
KEYWORDS = ['best serp api 2026', 'search api for agents', 'google search api python',
            'web scraping alternative', 'ai agent search tool']

db = sqlite3.connect('geo_visibility.db')
db.execute('''CREATE TABLE IF NOT EXISTS checks (
    keyword TEXT, checked_at TEXT, has_ai_overview INTEGER,
    brand_cited INTEGER, citation_text TEXT, position INTEGER
)''')
db.commit()
print(f'Tracking {len(KEYWORDS)} keywords for {len(BRAND_TERMS)} brand terms')

def check_keyword(keyword, brand_terms):
    data = requests.post('https://api.scavio.dev/api/v1/search',
        headers=SH, json={'query': keyword, 'country_code': 'us',
                          'include_ai_overview': True}).json()
    ao = data.get('ai_overview', {})
    ao_text = json.dumps(ao).lower() if ao else ''
    cited = any(b in ao_text for b in brand_terms)
    org = data.get('organic_results', [])
    pos = next((r['position'] for r in org
                if any(b in r.get('link', '').lower() for b in brand_terms)), None)
    now = datetime.now().isoformat()
    db.execute('INSERT INTO checks VALUES (?,?,?,?,?,?)',
        (keyword, now, 1 if ao else 0, 1 if cited else 0,
         ao_text[:500] if cited else '', pos))
    db.commit()
    return {'keyword': keyword, 'has_ao': bool(ao), 'cited': cited, 'position': pos}

for kw in KEYWORDS:
    r = check_keyword(kw, BRAND_TERMS)
    status = 'CITED' if r['cited'] else ('AO present' if r['has_ao'] else 'No AO')
    print(f'  {kw:35} | {status:12} | Organic: #{r["position"] or "-"}')

def daily_report():
    today = datetime.now().strftime('%Y-%m-%d')
    rows = db.execute(
        "SELECT keyword, has_ai_overview, brand_cited, position FROM checks WHERE checked_at LIKE ?",
        (f'{today}%',)).fetchall()
    total = len(rows)
    ao_count = sum(1 for r in rows if r[1])
    cited_count = sum(1 for r in rows if r[2])
    avg_pos = [r[3] for r in rows if r[3]]
    print(f'\nGEO Visibility Report - {today}')
    print(f'  Keywords checked: {total}')
    print(f'  AI Overviews present: {ao_count}/{total} ({ao_count/total*100:.0f}%)')
    print(f'  Brand cited in AO: {cited_count}/{total} ({cited_count/total*100:.0f}%)')
    if avg_pos:
        print(f'  Avg organic position: {sum(avg_pos)/len(avg_pos):.1f}')
    print(f'  Cost: ${total * 0.005:.3f}')
    return {'date': today, 'total': total, 'ao_rate': ao_count/total, 'citation_rate': cited_count/total}

daily_report()

def trend_report(days=7):
    rows = db.execute(
        'SELECT DATE(checked_at) as d, AVG(has_ai_overview), AVG(brand_cited) FROM checks GROUP BY d ORDER BY d DESC LIMIT ?',
        (days,)).fetchall()
    print(f'\nGEO Visibility Trend ({len(rows)} days):')
    for date, ao_rate, cite_rate in rows:
        bar_ao = '#' * int(ao_rate * 20)
        bar_cite = '#' * int(cite_rate * 20)
        print(f'  {date} | AO: {ao_rate*100:5.1f}% {bar_ao:20} | Cited: {cite_rate*100:5.1f}% {bar_cite:20}')
    if len(rows) >= 2:
        change = (rows[0][2] - rows[-1][2]) * 100
        direction = 'UP' if change > 0 else 'DOWN' if change < 0 else 'FLAT'
        print(f'  Citation rate {direction} {abs(change):.1f}pp over {len(rows)} days')

trend_report()