Upload 58 files

app/services/investigation.py

@@ -9,6 +9,7 @@ import asyncio
 
 from app.services.brazil_apis import consultar_cnpj, CompanyData
 from app.services.transparencia_api import verificar_sancoes
+from app.services.tse_api import buscar_politico
 from app.services import lancer
 from app.services.nlp import entity_extractor
 from app.core.database import get_db
@@ -38,6 +39,7 @@ class Dossier:
     dados_cadastrais: Optional[DossierSection] = None
     socios: Optional[DossierSection] = None
     sancoes: Optional[DossierSection] = None
+    dados_politicos: Optional[DossierSection] = None  # TSE data
     noticias: Optional[DossierSection] = None
     entidades_relacionadas: Optional[DossierSection] = None
     
@@ -226,7 +228,42 @@ async def investigar_pessoa(nome: str, cpf: Optional[str] = None) -> Dossier:
                 conteudo=sancoes
             )
     
-    # 2. Web search for information
+    # 2. Check TSE for political data
+    try:
+        tse_data = await buscar_politico(nome)
+        
+        if tse_data.get("encontrado"):
+            dossier.fonte_dados.append("TSE (DivulgaCand)")
+            
+            # Build political section
+            candidaturas = tse_data.get("candidaturas", [])
+            patrimonio = tse_data.get("total_patrimonio", 0)
+            partidos = tse_data.get("partidos", [])
+            
+            dossier.dados_politicos = DossierSection(
+                titulo=f"Dados Políticos ({len(candidaturas)} candidatura(s))",
+                icone="🗳️",
+                conteudo={
+                    "ultimo_cargo": tse_data.get("ultimo_cargo"),
+                    "partidos": partidos,
+                    "patrimonio_declarado": patrimonio,
+                    "ufs": tse_data.get("ufs", []),
+                    "candidaturas": candidaturas
+                }
+            )
+            
+            # Red flags for politicians
+            if patrimonio > 10000000:  # > 10 milhões
+                dossier.red_flags.append(f"💰 Patrimônio declarado alto: R$ {patrimonio:,.2f}")
+            
+            # Check for multiple parties (can indicate opportunism)
+            if len(partidos) >= 3:
+                dossier.red_flags.append(f"🔄 Passou por {len(partidos)} partidos: {', '.join(partidos)}")
+                dossier.dados_politicos.status = "warning"
+    except Exception as e:
+        print(f"TSE search error: {e}")
+    
+    # 3. Web search for information
     try:
         web_result = await lancer.search(f'"{nome}" biografia cargo empresa', max_results=10)
         
@@ -296,7 +333,7 @@ def dossier_to_dict(dossier: Dossier) -> Dict[str, Any]:
         "secoes": {}
     }
     
-    for field_name in ["dados_cadastrais", "socios", "sancoes", "noticias", "entidades_relacionadas"]:
+    for field_name in ["dados_cadastrais", "socios", "sancoes", "dados_politicos", "noticias", "entidades_relacionadas"]:
         section = getattr(dossier, field_name)
         if section:
             result["secoes"][field_name] = {