import json
import sys

with open('/root/lelyon_prospecting/data/enrichment_progress.json', 'r', encoding='utf-8') as f:
    d = json.load(f)

enriched = d['enriched']

# Filtrar empresas com algum sucesso
with_website = [(k,v) for k,v in enriched.items() if v.get('website') and v.get('website_status') == 'active']
with_contact = [(k,v) for k,v in enriched.items() if v.get('email') or v.get('telefone')]

output = []
output.append('=' * 80)
output.append(f'EMPRESAS COM WEBSITE ENCONTRADO ({len(with_website)}):')
output.append('=' * 80)
for cnpj, c in with_website:
    output.append(f'• {c.get("nome_fantasia", c.get("razao_social", ""))}')
    output.append(f'  CNPJ: {cnpj} | Cidade: {c.get("cidade", "")} | Site: {c.get("website", "")}')
    output.append(f'  Status: {c.get("website_status", "")} | Obs: {c.get("observacoes", "")}')
    output.append('')

output.append('=' * 80)
output.append(f'EMPRESAS COM CONTATO EXTRAÍDO (email/telefone) ({len(with_contact)}):')
output.append('=' * 80)
for cnpj, c in with_contact:
    output.append(f'• {c.get("nome_fantasia", c.get("razao_social", ""))}')
    output.append(f'  CNPJ: {cnpj} | Cidade: {c.get("cidade", "")}')
    output.append(f'  Email: {c.get("email", "-")}')
    output.append(f'  Telefone: {c.get("telefone", "-")}')
    output.append(f'  WhatsApp: {c.get("whatsapp", "-")}')
    output.append(f'  Site: {c.get("website", "-")}')
    output.append('')

# Também mostrar as que têm LinkedIn/redes sociais
with_social = [(k,v) for k,v in enriched.items() if any(v.get(k2) for k2 in ["linkedin", "instagram", "facebook"])]
if with_social:
    output.append('=' * 80)
    output.append(f'EMPRESAS COM REDES SOCIAIS ({len(with_social)}):')
    output.append('=' * 80)
    for cnpj, c in with_social:
        output.append(f'• {c.get("nome_fantasia", c.get("razao_social", ""))}')
        output.append(f'  LinkedIn: {c.get("linkedin", "-")}')
        output.append(f'  Instagram: {c.get("instagram", "-")}')
        output.append(f'  Facebook: {c.get("facebook", "-")}')
        output.append('')

with open('/root/lelyon_prospecting/data/firecrawl_report.txt', 'w', encoding='utf-8') as f:
    f.write('\n'.join(output))

print(f"Relatório salvo. Websites: {len(with_website)}, Contatos: {len(with_contact)}, Sociais: {len(with_social)}")