from trotta import TrottaClient

trotta = TrottaClient(api_key=TROTTA_API_KEY)

@app.post('/check')
async def check_threat(request):
    data = request.json
    
    result = await trotta.analyze(
        content=data['content'],
        sender=data.get('sender')
    )
    
    return {
        'action': 'BLOCK' if result.is_threat else 'ALLOW',
        'confidence': result.confidence
    }