Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import os
 import sys
 import json
@@ -8,8 +7,6 @@ import requests
 import urllib3
 from flask import Flask, request, jsonify, render_template_string
 from playwright.sync_api import sync_playwright
-import traceback
-import re
 
 # Suprimir warnings de SSL
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
@@ -20,438 +17,15 @@ logger = logging.getLogger(__name__)
 
 app = Flask(__name__)
 
-# ============================================
-# PAYLOAD EXATO DA BUSCA (conforme curl enviado)
-# ============================================
-PAYLOAD_BUSCA_BASE = {
-    "query": {
-        "function_score": {
-            
-            "query": {
-                "bool": {
-                    "filter": [
-                        {"query_string": {
-                            "default_operator": "AND",
-                            "fields": [
-                                "processo_codigo_completo",
-                                "acordao_ata^3",
-                                "documental_acordao_mesmo_sentido_lista_texto",
-                                "documental_doutrina_texto",
-                                "documental_indexacao_texto",
-                                "documental_jurisprudencia_citada_texto",
-                                "documental_legislacao_citada_texto",
-                                "documental_observacao_texto",
-                                "documental_publicacao_lista_texto",
-                                "documental_tese_tema_texto^3",
-                                "documental_tese_texto^3",
-                                "ementa_texto^3",
-                                "ministro_facet",
-                                "revisor_processo_nome",
-                                "orgao_julgador",
-                                "partes_lista_texto",
-                                "procedencia_geografica_completo",
-                                "processo_classe_processual_unificada_extenso",
-                                "titulo^6",
-                                "colac_numero",
-                                "colac_pagina",
-                                "decisao_texto^2",
-                                "documental_decisao_mesmo_sentido_lista_texto",
-                                "processo_precedente_texto",
-                                "sumula_texto^3",
-                                "ramo_direito^1",
-                                "situacao_sumula^1",
-                                "materia_noticia^1",
-                                "titulo_noticia^3",
-                                "resumo_noticia^3",
-                                "conteudo_noticia^1",
-                                "ramo_noticia^1",
-                                "inteiro_teor_texto"
-                            ],
-                            "query": "__TERMO__",
-                            "type": "cross_fields",
-                            "fuzziness": "AUTO:4,7",
-                            "analyzer": "radical_search_analyzer",
-                            "quote_analyzer": "standard_analyzer",
-                            "quote_field_suffix": ".standard"
-                        }}
-                    ],
-                    "must": [],
-                    "should": [
-                        {"query_string": {
-                            "default_operator": "AND",
-                            "fields": [
-                                "acordao_ata^3",
-                                "documental_doutrina_texto",
-                                "documental_indexacao_texto",
-                                "documental_jurisprudencia_citada_texto",
-                                "documental_observacao_texto",
-                                "documental_tese_tema_texto^3",
-                                "documental_tese_texto^3",
-                                "ementa_texto^3",
-                                "titulo^6",
-                                "decisao_texto^2",
-                                "sumula_texto^3",
-                                "ramo_direito^1",
-                                "situacao_sumula^1",
-                                "materia_noticia^1",
-                                "titulo_noticia^3",
-                                "resumo_noticia^3",
-                                "conteudo_noticia^1",
-                                "ramo_noticia^1",
-                                "inteiro_teor_texto"
-                            ],
-                            "query": "__TERMO__",
-                            "tie_breaker": 1,
-                            "fuzziness": "AUTO:4,7",
-                            "analyzer": "radical_search_analyzer",
-                            "quote_analyzer": "standard_analyzer",
-                            "quote_field_suffix": ".standard"
-                        }},
-                        {"query_string": {
-                            "default_operator": "and",
-                            "type": "phrase",
-                            "tie_breaker": 1,
-                            "phrase_slop": 20,
-                            "fields": [
-                                "acordao_ata^3",
-                                "documental_tese_tema_texto^3",
-                                "documental_tese_texto^3",
-                                "ementa_texto^3",
-                                "decisao_texto^2",
-                                "situacao_sumula^1",
-                                "titulo_noticia^3",
-                                "resumo_noticia^3",
-                                "conteudo_noticia^1",
-                                "inteiro_teor_texto^0.5"
-                            ],
-                            "query": "__TERMO__",
-                            "fuzziness": "AUTO:4,7",
-                            "analyzer": "radical_search_analyzer",
-                            "quote_analyzer": "standard_analyzer",
-                            "quote_field_suffix": ".standard"
-                        }},
-                        {"query_string": {
-                            "default_operator": "and",
-                            "type": "phrase",
-                            "tie_breaker": 1,
-                            "phrase_slop": 5,
-                            "fields": [
-                                "documental_acordao_mesmo_sentido_lista_texto",
-                                "documental_doutrina_texto",
-                                "documental_indexacao_texto",
-                                "documental_jurisprudencia_citada_texto",
-                                "documental_legislacao_citada_texto",
-                                "documental_observacao_texto",
-                                "partes_lista_texto",
-                                "processo_precedente_texto",
-                                "documental_decisao_mesmo_sentido_lista_texto"
-                            ],
-                            "query": "__TERMO__",
-                            "fuzziness": "AUTO:4,7",
-                            "analyzer": "radical_search_analyzer",
-                            "quote_analyzer": "standard_analyzer",
-                            "quote_field_suffix": ".standard"
-                        }},
-                        {"query_string": {
-                            "default_operator": "and",
-                            "type": "phrase",
-                            "phrase_slop": 1,
-                            "fields": [
-                                "documental_publicacao_lista_texto",
-                                "ministro_facet",
-                                "revisor_processo_nome",
-                                "orgao_julgador",
-                                "ramo_direito^1",
-                                "ramo_noticia^1",
-                                "procedencia_geografica_completo",
-                                "processo_classe_processual_unificada_extenso",
-                                "titulo^6",
-                                "colac_numero",
-                                "colac_pagina",
-                                "sumula_texto^3"
-                            ],
-                            "query": "__TERMO__",
-                            "fuzziness": "AUTO:4,7",
-                            "boost": 0,
-                            "analyzer": "radical_search_analyzer",
-                            "quote_analyzer": "standard_analyzer",
-                            "quote_field_suffix": ".standard"
-                        }}
-                    ]
-                }
-            }
-        }
-    },
-    "_source": [
-        "base", "_id", "id", "dg_unique", "titulo", "ministro_facet",
-        "procedencia_geografica_completo", "procedencia_geografica_pais_sigla",
-        "procedencia_geografica_uf_sigla", "procedencia_geografica_uf_extenso",
-        "processo_codigo_completo", "processo_classe_processual_unificada_extenso",
-        "processo_classe_processual_unificada_classe_sigla",
-        "processo_classe_processual_unificada_incidente_sigla", "processo_numero",
-        "julgamento_data", "publicacao_data", "is_decisao_presidencia",
-        "relator_processo_nome", "presidente_nome", "relator_decisao_nome",
-        "acordao_ata", "decisao_texto", "partes_lista_texto",
-        "acompanhamento_processual_url", "dje_url",
-        "documental_publicacao_lista_texto", "documental_decisao_mesmo_sentido_lista_texto",
-        "documental_decisao_mesmo_sentido_lista_html",
-        "documental_decisao_mesmo_sentido_is_secundario",
-        "documental_legislacao_citada_texto", "documental_indexacao_texto",
-        "documental_observacao_texto", "documental_observacao_html",
-        "documental_doutrina_texto", "externo_seq_objeto_incidente",
-        "dg_atualizado_em", "dg_unique", "volume_informativo", "ramo_noticia",
-        "materia_noticia", "titulo_noticia", "resumo_noticia", "conteudo_noticia",
-        "is_covid", "tipo_julgamento", "acompanhamento_processual_url",
-        "informativo_url", "pesquisa_url", "audio_url", "video_url",
-        "numero_noticias_url", "numero_noticias_processo", "informativo_url",
-        "periodo_inicio_data", "periodo_fim_data", "processo_lista_texto",
-        "sumula_numero", "orgao_julgador", "is_vinculante", "sumula_texto",
-        "situacao_sumula", "ramo_direito", "aprovacao_url",
-        "processo_precedente_texto", "processo_precedente_html",
-        "processo_classe_processual_unificada_sigla", "is_questao_ordem",
-        "is_repercussao_geral_admissibilidade", "is_repercussao_geral_merito",
-        "is_repercussao_geral_recurso_interno", "is_repercussao_geral",
-        "is_processo_antigo", "is_colac", "colac_numero", "colac_pagina",
-        "revisor_processo_nome", "relator_acordao_nome",
-        "julgamento_is_sessao_virtual", "republicacao_data", "ementa_texto",
-        "inteiro_teor_url", "documental_acordao_mesmo_sentido_lista_texto",
-        "documental_acordao_mesmo_sentido_lista_html",
-        "documental_acordao_mesmo_sentido_is_secundario",
-        "documental_jurisprudencia_citada_texto", "documental_assunto_texto",
-        "documental_tese_tipo", "documental_tese_texto", "documental_tese_tema_texto",
-        "old_seq_colac", "old_seq_repercussao_geral", "old_seq_sjur", "ods_onu"
-    ],
-    "aggs": {
-        "base_agg": {
-            "filters": {
-                "filters": {
-                    "acordaos": {"match": {"base": "acordaos"}},
-                    "sumulas": {"match": {"base": "sumulas"}},
-                    "decisoes": {"match": {"base": "decisoes"}},
-                    "informativos": {"match": {"base": "novo_informativo"}}
-                }
-            }
-        },
+# Constantes da API STF (endpoint simplificado)
+STF_API_URL = "https://jurisprudencia.stf.jus.br/api/search/search"  # ou use o /busca se preferir
+# Mas vamos usar o mesmo endpoint /busca que já implementamos no próprio app
+# Melhor: vamos usar o próprio endpoint /busca do nosso app, que por sua vez chama a API STF.
+# Então não precisamos chamar a API STF diretamente aqui; o frontend fará fetch para /api/busca.
 
-        
-        "processo_classe_processual_unificada_classe_sigla_agg": {
-            "aggs": {
-                "processo_classe_processual_unificada_classe_sigla_agg": {
-                    "terms": {"field": "processo_classe_processual_unificada_classe_sigla.keyword", "size": 200, "execution_hint": "map"}
-                }
-            },
-            "filter": {"bool": {"must": [{"term": {"base": "acordaos"}}]}}
-        },
-        "procedencia_geografica_uf_sigla_agg": {
-            "aggs": {
-                "procedencia_geografica_uf_sigla_agg": {
-                    "terms": {"field": "procedencia_geografica_uf_sigla", "size": 200, "execution_hint": "map"}
-                }
-            },
-            "filter": {"bool": {"must": [{"term": {"base": "acordaos"}}]}}
-        }
-    },
-    "size": 250,
-    "from": 0,
-    "post_filter": {
-        "bool": {
-            "must": [{"term": {"base": "acordaos"}}],
-            "should": []
-        }
-    },
-    "sort": [{"_score": "desc"}],
-    "highlight": {
-        "highlight_query": {
-            "bool": {
-                "filter": [
-                    {"query_string": {
-                        "default_operator": "AND",
-                        "fields": [
-                            "processo_codigo_completo",
-                            "acordao_ata^3",
-                            "documental_acordao_mesmo_sentido_lista_texto",
-                            "documental_doutrina_texto",
-                            "documental_indexacao_texto",
-                            "documental_jurisprudencia_citada_texto",
-                            "documental_legislacao_citada_texto",
-                            "documental_observacao_texto",
-                            "documental_publicacao_lista_texto",
-                            "documental_tese_tema_texto^3",
-                            "documental_tese_texto^3",
-                            "ementa_texto^3",
-                            "ministro_facet",
-                            "revisor_processo_nome",
-                            "orgao_julgador",
-                            "partes_lista_texto",
-                            "procedencia_geografica_completo",
-                            "processo_classe_processual_unificada_extenso",
-                            "titulo^6",
-                            "colac_numero",
-                            "colac_pagina",
-                            "decisao_texto^2",
-                            "documental_decisao_mesmo_sentido_lista_texto",
-                            "processo_precedente_texto",
-                            "sumula_texto^3",
-                            "ramo_direito^1",
-                            "situacao_sumula^1",
-                            "materia_noticia^1",
-                            "titulo_noticia^3",
-                            "resumo_noticia^3",
-                            "conteudo_noticia^1",
-                            "ramo_noticia^1",
-                            "inteiro_teor_texto"
-                        ],
-                        "query": "__TERMO__",
-                        "type": "cross_fields",
-                        "fuzziness": "AUTO:4,7",
-                        "analyzer": "radical_search_analyzer",
-                        "quote_analyzer": "standard_analyzer",
-                        "quote_field_suffix": ".standard"
-                    }}
-                ],
-                "must": [],
-                "should": [
-                    {"query_string": {
-                        "default_operator": "AND",
-                        "fields": [
-                            "acordao_ata^3",
-                            "documental_doutrina_texto",
-                            "documental_indexacao_texto",
-                            "documental_jurisprudencia_citada_texto",
-                            "documental_observacao_texto",
-                            "documental_tese_tema_texto^3",
-                            "documental_tese_texto^3",
-                            "ementa_texto^3",
-                            "titulo^6",
-                            "decisao_texto^2",
-                            "sumula_texto^3",
-                            "ramo_direito^1",
-                            "situacao_sumula^1",
-                            "materia_noticia^1",
-                            "titulo_noticia^3",
-                            "resumo_noticia^3",
-                            "conteudo_noticia^1",
-                            "ramo_noticia^1",
-                            "inteiro_teor_texto"
-                        ],
-                        "query": "__TERMO__",
-                        "tie_breaker": 1,
-                        "fuzziness": "AUTO:4,7",
-                        "analyzer": "radical_search_analyzer",
-                        "quote_analyzer": "standard_analyzer",
-                        "quote_field_suffix": ".standard"
-                    }},
-                    {"query_string": {
-                        "default_operator": "and",
-                        "type": "phrase",
-                        "tie_breaker": 1,
-                        "phrase_slop": 20,
-                        "fields": [
-                            "acordao_ata^3",
-                            "documental_tese_tema_texto^3",
-                            "documental_tese_texto^3",
-                            "ementa_texto^3",
-                            "decisao_texto^2",
-                            "situacao_sumula^1",
-                            "titulo_noticia^3",
-                            "resumo_noticia^3",
-                            "conteudo_noticia^1",
-                            "inteiro_teor_texto^0.5"
-                        ],
-                        "query": "__TERMO__",
-                        "fuzziness": "AUTO:4,7",
-                        "analyzer": "radical_search_analyzer",
-                        "quote_analyzer": "standard_analyzer",
-                        "quote_field_suffix": ".standard"
-                    }},
-                    {"query_string": {
-                        "default_operator": "and",
-                        "type": "phrase",
-                        "tie_breaker": 1,
-                        "phrase_slop": 5,
-                        "fields": [
-                            "documental_acordao_mesmo_sentido_lista_texto",
-                            "documental_doutrina_texto",
-                            "documental_indexacao_texto",
-                            "documental_jurisprudencia_citada_texto",
-                            "documental_legislacao_citada_texto",
-                            "documental_observacao_texto",
-                            "partes_lista_texto",
-                            "processo_precedente_texto",
-                            "documental_decisao_mesmo_sentido_lista_texto"
-                        ],
-                        "query": "__TERMO__",
-                        "fuzziness": "AUTO:4,7",
-                        "analyzer": "radical_search_analyzer",
-                        "quote_analyzer": "standard_analyzer",
-                        "quote_field_suffix": ".standard"
-                    }},
-                    {"query_string": {
-                        "default_operator": "and",
-                        "type": "phrase",
-                        "phrase_slop": 1,
-                        "fields": [
-                            "documental_publicacao_lista_texto",
-                            "ministro_facet",
-                            "revisor_processo_nome",
-                            "orgao_julgador",
-                            "ramo_direito^1",
-                            "ramo_noticia^1",
-                            "procedencia_geografica_completo",
-                            "processo_classe_processual_unificada_extenso",
-                            "titulo^6",
-                            "colac_numero",
-                            "colac_pagina",
-                            "sumula_texto^3"
-                        ],
-                        "query": "__TERMO__",
-                        "fuzziness": "AUTO:4,7",
-                        "boost": 0,
-                        "analyzer": "radical_search_analyzer",
-                        "quote_analyzer": "standard_analyzer",
-                        "quote_field_suffix": ".standard"
-                    }}
-                ]
-            }
-        },
-        "number_of_fragments": 64,
-        "fragment_size": 300,
-        "order": "score",
-        "pre_tags": ["<em>"],
-        "post_tags": ["</em>"],
-        "fields": {
-            "ementa_texto": {"fragment_size": 24000, "matched_fields": ["ementa_texto", "ementa_texto.standard"], "type": "fvh"},
-            "sumula_texto": {"number_of_fragments": 0, "matched_fields": ["sumula_texto", "sumula_texto.standard"], "type": "fvh"},
-            "materia_noticia": {"matched_fields": ["materia_noticia", "materia_noticia.standard"], "type": "fvh"},
-            "titulo_noticia": {"matched_fields": ["titulo_noticia", "titulo_noticia.standard"], "type": "fvh"},
-            "resumo_noticia": {"fragment_size": 5000, "matched_fields": ["resumo_noticia", "resumo_noticia.standard"], "type": "fvh"},
-            "conteudo_noticia": {"fragment_size": 50000, "matched_fields": ["conteudo_noticia", "conteudo_noticia.standard"], "type": "fvh"},
-            "acordao_ata": {"fragment_size": 600, "matched_fields": ["acordao_ata", "acordao_ata.standard"], "type": "fvh"},
-            "decisao_texto": {"fragment_size": 1200, "matched_fields": ["decisao_texto", "decisao_texto.standard"], "type": "fvh"},
-            "documental_tese_texto": {"fragment_size": 2000, "matched_fields": ["documental_tese_texto", "documental_tese_texto.standard"], "type": "fvh"},
-            "documental_tese_tema_texto": {"fragment_size": 2000, "matched_fields": ["documental_tese_tema_texto", "documental_tese_tema_texto.standard"], "type": "fvh"},
-            "documental_observacao_texto": {"matched_fields": ["documental_observacao_texto", "documental_observacao_texto.standard"], "type": "fvh"},
-            "documental_indexacao_texto": {"matched_fields": ["documental_indexacao_texto", "documental_indexacao_texto.standard"], "type": "fvh"},
-            "documental_legislacao_citada_texto": {"matched_fields": ["documental_legislacao_citada_texto", "documental_legislacao_citada_texto.standard"], "type": "fvh"},
-            "documental_jurisprudencia_citada_texto": {"matched_fields": ["documental_jurisprudencia_citada_texto", "documental_jurisprudencia_citada_texto.standard"], "type": "fvh"},
-            "documental_doutrina_texto": {"matched_fields": ["documental_doutrina_texto", "documental_doutrina_texto.standard"], "type": "fvh"},
-            "partes_lista_texto": {"matched_fields": ["partes_lista_texto", "partes_lista_texto.standard"], "type": "fvh"},
-            "documental_publicacao_lista_texto": {"matched_fields": ["documental_publicacao_lista_texto", "documental_publicacao_lista_texto.standard"], "type": "fvh"},
-            "documental_acordao_mesmo_sentido_lista_texto": {"matched_fields": ["documental_acordao_mesmo_sentido_lista_texto", "documental_acordao_mesmo_sentido_lista_texto.standard"], "type": "fvh"},
-            "documental_decisao_mesmo_sentido_lista_texto": {"matched_fields": ["documental_decisao_mesmo_sentido_lista_texto", "documental_decisao_mesmo_sentido_lista_texto.standard"], "type": "fvh"},
-            "processo_precedente_texto": {"matched_fields": ["processo_precedente_texto", "processo_precedente_texto.standard"], "type": "fvh"},
-            "procedencia_geografica_completo": {"matched_fields": ["procedencia_geografica_completo", "procedencia_geografica_completo.standard"], "type": "fvh"},
-            "inteiro_teor_texto": {"fragment_size": 600, "matched_fields": ["inteiro_teor_texto", "inteiro_teor_texto.standard"], "type": "fvh"}
-        }
-    },
-    "track_total_hits": True
-}
+# Cache de token (ainda necessário para as chamadas à API STF)
+token_cache = {"token": None, "expires_at": 0}
 
-# ============================================
-# Constantes da API e cache (mantidos)
-# ============================================
-URL_API = "https://jurisprudencia.stf.jus.br/api/search/search"
-URL_API_GET = "https://jurisprudencia.stf.jus.br/api/search/get"
 HEADERS = {
     "Accept": "application/json, text/plain, */*",
     "Content-Type": "application/json",
@@ -460,8 +34,6 @@ HEADERS = {
     "Origin": "https://jurisprudencia.stf.jus.br"
 }
 
-token_cache = {"token": None, "expires_at": 0}
-
 # ============================================
 # Funções de token e busca
 # ============================================
@@ -501,286 +73,199 @@ def get_fresh_token(force_refresh=False):
         logger.error(f"Erro ao obter token: {str(e)}")
         return None
 
-def search_with_token(token, payload):
+def search_stf(query: str):
+    """Faz a busca na API do STF usando o payload completo"""
+    token = get_fresh_token()
     if not token:
-        return {"success": False, "error": "Token não fornecido"}
-    headers = HEADERS.copy()
-    headers['Cookie'] = f'aws-waf-token={token}'
-    try:
-        response = requests.post(URL_API, headers=headers, json=payload, verify=False, timeout=30)
-        logger.info(f"Resposta: status {response.status_code}")
-        if response.status_code == 200:
-            return {"success": True, "data": response.json()}
-        elif response.status_code in (403, 202):
-            token_cache["token"] = None
-            return {"success": False, "error": "Token expirado ou desafio", "status": response.status_code}
-        else:
-            return {"success": False, "error": f"HTTP {response.status_code}", "status": response.status_code, "text": response.text[:500]}
-    except Exception as e:
-        logger.error(f"Erro na requisição: {str(e)}")
-        return {"success": False, "error": str(e)}
+        return {"error": "Não foi possível obter token de acesso"}, 503
+
+    # Payload completo (pode ser simplificado se quiser, mas usaremos o mesmo)
+    # Vamos usar o payload básico que busca em todos os campos
+    payload = {
+        "query": {
+            "query_string": {
+                "query": query,
+                "default_operator": "AND",
+                "fields": ["ementa_texto", "acordao_ata", "decisao_texto", "titulo"],
+                "fuzziness": "AUTO:4,7"
+            }
+        },
+        "_source": ["id", "titulo", "ementa_texto", "processo_codigo_completo", "relator_processo_nome", "orgao_julgador", "julgamento_data", "inteiro_teor_url"],
+        "size": 20,
+        "sort": [{"julgamento_data": {"order": "desc"}}],
+        "track_total_hits": True
+    }
 
-def get_document_raw(token, doc_id):
-    if not token:
-        return {"success": False, "error": "Token não fornecido"}
     headers = HEADERS.copy()
     headers['Cookie'] = f'aws-waf-token={token}'
-    headers.pop("Content-Type", None)
-    url = f"{URL_API_GET}/{doc_id}"
+
     try:
-        response = requests.get(url, headers=headers, verify=False, timeout=30)
+        response = requests.post(STF_API_URL, headers=headers, json=payload, verify=False, timeout=30)
         if response.status_code == 200:
-            return {"success": True, "data": response.json()}
-        elif response.status_code in (403, 202):
-            token_cache["token"] = None
-            return {"success": False, "error": "Token expirado", "status": response.status_code}
+            return response.json()
+        elif response.status_code == 202:
+            # Tenta renovar o token uma vez
+            token = get_fresh_token(force_refresh=True)
+            if token:
+                headers['Cookie'] = f'aws-waf-token={token}'
+                response = requests.post(STF_API_URL, headers=headers, json=payload, verify=False, timeout=30)
+                if response.status_code == 200:
+                    return response.json()
+            return {"error": "Token expirado e renovação falhou"}, 503
         else:
-            return {"success": False, "error": f"HTTP {response.status_code}"}
+            return {"error": f"API retornou status {response.status_code}"}, response.status_code
     except Exception as e:
-        return {"success": False, "error": str(e)}
+        return {"error": str(e)}, 502
 
 # ============================================
-# HTML TEMPLATE (mantido, mas omitido para brevidade)
+# Rota principal (HTML)
 # ============================================
-HTML_TEMPLATE = """<!DOCTYPE html>
+@app.route('/')
+def index():
+    return render_template_string("""
+<!DOCTYPE html>
 <html>
 <head>
-    <title>⚖️ STF Jurisprudência</title>
+    <title>Busca STF</title>
     <meta charset="utf-8">
     <meta name="viewport" content="width=device-width, initial-scale=1">
     <style>
-        body { font-family: 'Segoe UI', Roboto, system-ui, sans-serif; max-width: 1600px; margin: 0 auto; padding: 20px; background: #f0f2f5; }
-        .container { background: white; border-radius: 12px; padding: 30px; box-shadow: 0 8px 20px rgba(0,0,0,0.1); }
-        h1 { color: #1a1a2e; border-bottom: 3px solid #4CAF50; padding-bottom: 15px; }
-        .info-box { background: #e8f0fe; border-left: 5px solid #2196F3; padding: 15px 20px; margin: 20px 0; border-radius: 8px; }
-        .stats-grid { display: grid; grid-template-columns: repeat(auto-fit, minmax(200px,1fr)); gap:15px; margin:25px 0; }
-        .stat-card { background: white; border:1px solid #e0e0e0; border-radius:10px; padding:20px; text-align:center; }
-        .stat-value { font-size:32px; font-weight:bold; color:#1a73e8; }
-        .button-group { display:flex; gap:15px; flex-wrap:wrap; margin:25px 0; }
-        button { background:#1a73e8; color:white; border:none; padding:14px 28px; border-radius:8px; cursor:pointer; }
-        button:hover { background:#1557b0; }
-        .loading { display:inline-block; width:20px; height:20px; border:3px solid rgba(255,255,255,0.3); border-top:3px solid white; border-radius:50%; animation:spin 1s linear infinite; }
-        @keyframes spin { 0% { transform:rotate(0deg); } 100% { transform:rotate(360deg); } }
-        pre { background:#f8f9fa; border:1px solid #e0e0e0; border-radius:10px; padding:20px; overflow:auto; max-height:600px; }
-        .success { background:#e6f4ea; border-left:5px solid #34a853; padding:15px; margin:15px 0; }
-        .error { background:#fce8e6; border-left:5px solid #ea4335; padding:15px; margin:15px 0; }
-        .token-box { background:#1a1a2e; color:#e0e0e0; padding:15px; border-radius:8px; font-family:monospace; margin:15px 0; }
-        .filters { background:#f8f9fa; border-radius:8px; padding:20px; margin:20px 0; border:1px solid #e0e0e0; }
-        .filter-group { display:grid; grid-template-columns:repeat(auto-fit, minmax(200px,1fr)); gap:15px; }
-        .filter-item label { display:block; margin-bottom:8px; color:#5f6368; font-size:14px; font-weight:500; }
-        .filter-item input, .filter-item select { width:100%; padding:10px; border:1px solid #ddd; border-radius:6px; }
+        body { font-family: 'Segoe UI', Roboto, system-ui, sans-serif; max-width: 1200px; margin: 0 auto; padding: 20px; background: #f5f5f5; }
+        .container { background: white; border-radius: 8px; padding: 30px; box-shadow: 0 2px 10px rgba(0,0,0,0.1); }
+        h1 { color: #333; border-bottom: 2px solid #4CAF50; padding-bottom: 10px; }
+        .search-box { display: flex; gap: 10px; margin: 20px 0; }
+        .search-box input { flex: 1; padding: 12px; font-size: 16px; border: 1px solid #ddd; border-radius: 4px; }
+        .search-box button { padding: 12px 24px; background: #4CAF50; color: white; border: none; border-radius: 4px; font-size: 16px; cursor: pointer; }
+        .search-box button:hover { background: #45a049; }
+        .search-box button:disabled { background: #ccc; cursor: not-allowed; }
+        .result { border: 1px solid #e0e0e0; border-radius: 8px; padding: 15px; margin: 10px 0; background: #fafafa; }
+        .result h3 { margin: 0 0 8px; color: #1a0dab; }
+        .result .meta { color: #666; font-size: 14px; margin-bottom: 10px; }
+        .result .ementa { font-size: 14px; line-height: 1.6; color: #333; white-space: pre-wrap; }
+        .result a { color: #1a73e8; text-decoration: none; }
+        .result a:hover { text-decoration: underline; }
+        .loading { display: inline-block; width: 20px; height: 20px; border: 3px solid rgba(0,0,0,0.1); border-top-color: #4CAF50; border-radius: 50%; animation: spin 1s linear infinite; margin-right: 10px; }
+        @keyframes spin { to { transform: rotate(360deg); } }
+        .error { color: #d93025; background: #fce8e6; border-left: 4px solid #ea4335; padding: 15px; margin: 10px 0; border-radius: 4px; }
     </style>
 </head>
 <body>
     <div class="container">
-        <h1>STF Jurisprudência - Visualizador</h1>
-        <div class="info-box">
-            <strong>📌 API de Jurisprudência do STF</strong><br>
-            • Documentos disponíveis: <span id="totalDocs">carregando...</span>
-        </div>
-        <div class="stats-grid">
-            <div class="stat-card"><div class="stat-value" id="requestsCount">0</div><div class="stat-label">Requisições</div></div>
-            <div class="stat-card"><div class="stat-value" id="successCount">0</div><div class="stat-label">Sucessos</div></div>
-            <div class="stat-card"><div class="stat-value" id="failCount">0</div><div class="stat-label">Falhas</div></div>
-            <div class="stat-card"><div class="stat-value" id="docsCount">0</div><div class="stat-label">Docs Obtidos</div></div>
+        <h1>⚖️ Busca na Jurisprudência do STF</h1>
+        <div class="search-box">
+            <input type="text" id="query" placeholder="Digite sua busca (ex: dano moral mordida cachorro)" value="">
+            <button id="searchBtn" onclick="doSearch()">Buscar</button>
         </div>
-        <div class="filters">
-            <h3>🔍 Filtros de Busca</h3>
-            <div class="filter-group">
-                <div class="filter-item"><label>Quantidade</label><select id="pageSize"><option value="5">5</option><option value="10">10</option><option value="25">25</option><option value="50">50</option><option value="100" selected>100</option></select></div>
-                <div class="filter-item"><label>Ordenar por</label><select id="sortOrder"><option value="desc">Mais recentes</option><option value="asc">Mais antigos</option></select></div>
-                <div class="filter-item"><label>Base</label><select id="base"><option value="acordaos">Acórdãos</option><option value="decisoes">Decisões</option><option value="sumulas">Súmulas</option><option value="informativos">Informativos</option><option value="">Todas</option></select></div>
-                <div class="filter-item"><label>Busca por ID</label><input type="text" id="docId" placeholder="Ex: sjur505215"></div>
-            </div>
-        </div>
-        <div class="button-group">
-            <button id="testBtn" onclick="runSearch()"><span class="loading" id="loading" style="display:none;"></span>🔍 Buscar Documentos</button>
-            <button id="getByIdBtn" class="secondary" onclick="getDocumentById()">📄 Buscar por ID</button>
-            <button id="downloadBtn" class="download" onclick="downloadJSON()" disabled>📥 Download JSON</button>
-            <button id="copyBtn" class="secondary" onclick="copyToken()">🔑 Copiar Token</button>
-        </div>
-        <div id="tokenDisplay" style="display:none;" class="token-box"><span class="token-label">🔐 Token AWS WAF</span><span id="tokenValue"></span></div>
-        <div id="result" style="margin-top:25px;"></div>
+        <div id="loading" style="display: none;" class="loading"></div>
+        <div id="results"></div>
     </div>
+
     <script>
-        let lastResult = null, lastToken = null, requestsCount = 0, successCount = 0, failCount = 0, docsCount = 0;
-        function displayResult(data) {
-            if (!data?.result?.hits?.hits) return document.getElementById('result').innerHTML = '<div class="error">Nenhum documento encontrado</div>';
-            const hits = data.result.hits.hits, total = data.result.hits.total?.value || 0;
-            let html = `<div class="success">✅ Encontrados ${total.toLocaleString()} documentos. Exibindo ${hits.length} resultados.</div>`;
-            hits.forEach((hit, idx) => {
-                const src = hit._source || {}, docId = src.id || hit._id || `doc_${idx}`;
-                html += `<div style="border:1px solid #ccc; margin-bottom:10px; padding:10px;"><h3>${src.titulo || src.processo_codigo_completo || docId}</h3>`;
-                html += `<p><strong>ID:</strong> ${docId}</p>`;
-                html += `<p><strong>Relator:</strong> ${src.relator_processo_nome || ''}</p>`;
-                html += `<p><strong>Órgão:</strong> ${src.orgao_julgador || ''}</p>`;
-                html += `<p><strong>Data:</strong> ${src.julgamento_data || ''}</p>`;
-                html += `<p><strong>Ementa:</strong> ${(src.ementa_texto || '').substring(0,500)}...</p>`;
-                if (src.inteiro_teor_url) html += `<p><a href="${src.inteiro_teor_url}" target="_blank">📄 Inteiro Teor</a></p>`;
-                html += `<button onclick="fetchFullDocument('${docId}')">📄 Obter completo</button>`;
-                html += `</div>`;
-            });
-            document.getElementById('result').innerHTML = html;
-        }
-        async function runSearch() {
-            const btn = document.getElementById('testBtn'), loading = document.getElementById('loading'), resultDiv = document.getElementById('result');
-            btn.disabled = true; loading.style.display = 'inline-block'; resultDiv.innerHTML = '<div class="info-box">⏳ Buscando...</div>';
-            try {
-                const resp = await fetch('/api/search-advanced', {
-                    method:'POST', headers:{'Content-Type':'application/json'},
-                    body:JSON.stringify({ pageSize: parseInt(document.getElementById('pageSize').value), sortOrder: document.getElementById('sortOrder').value, base: document.getElementById('base').value || undefined })
-                });
-                const data = await resp.json();
-                requestsCount++; document.getElementById('requestsCount').textContent = requestsCount;
-                if (data.success) {
-                    successCount++; document.getElementById('successCount').textContent = successCount;
-                    if (data.data?.result?.hits) {
-                        docsCount = data.data.result.hits.hits.length;
-                        document.getElementById('docsCount').textContent = docsCount;
-                        if (data.data.result.hits.total?.value) document.getElementById('totalDocs').textContent = data.data.result.hits.total.value.toLocaleString();
-                    }
-                    lastResult = data.data; lastToken = data.token;
-                    if (lastToken) { document.getElementById('tokenValue').textContent = lastToken; document.getElementById('tokenDisplay').style.display = 'block'; document.getElementById('downloadBtn').disabled = false; }
-                    displayResult(data.data);
-                } else {
-                    failCount++; document.getElementById('failCount').textContent = failCount;
-                    resultDiv.innerHTML = '<div class="error">❌ Falha</div><pre>'+JSON.stringify(data,null,2)+'</pre>';
-                }
-            } catch(e) {
-                failCount++; document.getElementById('failCount').textContent = failCount;
-                resultDiv.innerHTML = '<div class="error">❌ Erro: '+e.message+'</div>';
-            } finally {
-                btn.disabled = false; loading.style.display = 'none';
+        async function doSearch() {
+            const query = document.getElementById('query').value.trim();
+            if (!query) {
+                alert('Digite algo para buscar');
+                return;
             }
-        }
-        async function getDocumentById() {
-            const docId = document.getElementById('docId').value.trim();
-            if (!docId) return alert('Digite um ID');
-            const btn = document.getElementById('getByIdBtn'), loading = document.getElementById('loading'), resultDiv = document.getElementById('result');
-            btn.disabled = true; loading.style.display = 'inline-block'; resultDiv.innerHTML = '<div class="info-box">⏳ Buscando...</div>';
+            const btn = document.getElementById('searchBtn');
+            const loading = document.getElementById('loading');
+            const resultsDiv = document.getElementById('results');
+            
+            btn.disabled = true;
+            loading.style.display = 'inline-block';
+            resultsDiv.innerHTML = '';
+            
             try {
-                const resp = await fetch(`/api/document/${docId}`);
-                const data = await resp.json();
-                requestsCount++; document.getElementById('requestsCount').textContent = requestsCount;
-                if (data.success && data.document) {
-                    successCount++; document.getElementById('successCount').textContent = successCount; docsCount++; document.getElementById('docsCount').textContent = docsCount;
-                    lastResult = { result: { hits: { hits: [data.document] } } };
-                    displayResult(lastResult);
+                const response = await fetch(`/api/busca?q=${encodeURIComponent(query)}`);
+                const data = await response.json();
+                
+                if (data.error) {
+                    resultsDiv.innerHTML = `<div class="error">Erro: ${data.error}</div>`;
                 } else {
-                    failCount++; document.getElementById('failCount').textContent = failCount;
-                    resultDiv.innerHTML = '<div class="error">❌ Documento não encontrado</div>';
+                    const resultados = data.resultados || [];
+                    if (resultados.length === 0) {
+                        resultsDiv.innerHTML = '<p>Nenhum resultado encontrado.</p>';
+                    } else {
+                        let html = `<p>Encontrados ${data.total || resultados.length} resultados.</p>`;
+                        resultados.forEach(item => {
+                            html += `
+                                <div class="result">
+                                    <h3>${item.titulo || item.processo || 'Sem título'}</h3>
+                                    <div class="meta">
+                                        ${item.relator ? `Relator: ${item.relator} |` : ''}
+                                        ${item.orgao ? `${item.orgao} |` : ''}
+                                        ${item.data ? `Data: ${item.data} |` : ''}
+                                        ${item.id ? `ID: ${item.id}` : ''}
+                                    </div>
+                                    <div class="ementa">${(item.ementa || '').replace(/\\n/g, '<br>')}</div>
+                                    ${item.url_documento ? `<p><a href="${item.url_documento}" target="_blank">📄 Inteiro Teor</a></p>` : ''}
+                                </div>
+                            `;
+                        });
+                        resultsDiv.innerHTML = html;
+                    }
                 }
-            } catch(e) {
-                failCount++; document.getElementById('failCount').textContent = failCount;
-                resultDiv.innerHTML = '<div class="error">❌ Erro: '+e.message+'</div>';
+            } catch (err) {
+                resultsDiv.innerHTML = `<div class="error">Erro na requisição: ${err.message}</div>`;
             } finally {
-                btn.disabled = false; loading.style.display = 'none';
+                btn.disabled = false;
+                loading.style.display = 'none';
             }
         }
-        async function fetchFullDocument(docId) {
-            try {
-                const resp = await fetch(`/api/document-raw/${docId}`);
-                const data = await resp.json();
-                if (data.success) {
-                    alert('Documento completo obtido! Verifique o console.');
-                    console.log('Documento completo:', data.document);
-                } else alert('Erro: ' + (data.error || 'desconhecido'));
-            } catch(e) { alert('Erro: ' + e.message); }
-        }
-        function downloadJSON() {
-            if (!lastResult) return alert('Nenhum resultado');
-            const dataStr = JSON.stringify(lastResult, null, 2);
-            const blob = new Blob([dataStr], {type:'application/json'});
-            const url = URL.createObjectURL(blob);
-            const a = document.createElement('a');
-            a.href = url; a.download = `stf_${new Date().toISOString().slice(0,10)}.json`;
-            document.body.appendChild(a); a.click(); document.body.removeChild(a); URL.revokeObjectURL(url);
-        }
-        function copyToken() { if (lastToken) navigator.clipboard.writeText(lastToken).then(()=>alert('✅ Token copiado!')); else alert('Nenhum token'); }
-        window.onload = async () => {
-            try {
-                const resp = await fetch('/api/health');
-                const data = await resp.json();
-                if (data.total_docs) document.getElementById('totalDocs').textContent = data.total_docs.toLocaleString();
-            } catch(e) { document.getElementById('totalDocs').textContent = 'indisponível'; }
-        };
+
+        // Permitir busca ao pressionar Enter
+        document.getElementById('query').addEventListener('keypress', function(e) {
+            if (e.key === 'Enter') {
+                doSearch();
+            }
+        });
     </script>
 </body>
-</html>"""
+</html>
+""")
 
 # ============================================
-# Rotas
+# Endpoint /api/busca (retorna JSON)
 # ============================================
-@app.route('/')
-def index():
-    try:
-        return render_template_string(HTML_TEMPLATE)
-    except Exception as e:
-        return f"Erro no template: {str(e)}<br><pre>{traceback.format_exc()}</pre>", 500
-
-@app.route('/api/search-advanced', methods=['POST'])
-def search_advanced():
-    data = request.json or {}
-    page_size = data.get('pageSize', 100)
-    sort_order = data.get('sortOrder', 'desc')
-    base = data.get('base')
-
-    token = get_fresh_token()
-    if not token:
-        return jsonify({"success": False, "error": "Token não disponível"}), 500
-
-    payload = PAYLOAD_BUSCA_BASE.copy()
-    payload["size"] = min(page_size, 250)
-    if base:
-        payload["post_filter"]["bool"]["must"] = [{"term": {"base": base}}]
-    # Substituir placeholder se houver (não usado aqui)
-    result = search_with_token(token, payload)
-    if result.get("success"):
-        return jsonify({"success": True, "token": token, "data": result["data"], "timestamp": time.time()})
-    else:
-        if result.get("status") in (403, 202):
-            token = get_fresh_token(force_refresh=True)
-            if token:
-                result = search_with_token(token, payload)
-                if result.get("success"):
-                    return jsonify({"success": True, "token": token, "data": result["data"], "timestamp": time.time()})
-        return jsonify({"success": False, "error": result.get("error")}), 500
+@app.route('/api/busca')
+def busca_api():
+    query = request.args.get('q', '')
+    if not query:
+        return jsonify({"error": "Parâmetro 'q' obrigatório"}), 400
 
-@app.route('/api/document/<doc_id>', methods=['GET'])
-def get_document(doc_id):
-    token = get_fresh_token()
-    if not token:
-        return jsonify({"error": "Não foi possível obter token"}), 500
-    payload = {
-        "query": {"ids": {"values": [doc_id]}},
-        "_source": ["id", "titulo", "ementa_texto", "processo_codigo_completo", "relator_processo_nome", "orgao_julgador", "julgamento_data", "inteiro_teor_url"],
-        "size": 1
-    }
-    result = search_with_token(token, payload)
-    if result.get("success") and result["data"].get("hits", {}).get("hits"):
-        doc = result["data"]["hits"]["hits"][0]
-        return jsonify({"success": True, "document": doc})
-    else:
-        return jsonify({"success": False, "error": result.get("error", "Documento não encontrado")}), 404
+    result = search_stf(query)
+    if isinstance(result, tuple):
+        return jsonify(result[0]), result[1]
+    
+    # Converte para o formato simplificado de saída
+    hits = result.get('result', {}).get('hits', {}).get('hits', [])
+    resultados = []
+    for hit in hits:
+        src = hit.get('_source', {})
+        item = {
+            "id": src.get('id') or hit.get('_id'),
+            "titulo": src.get('titulo'),
+            "processo": src.get('processo_codigo_completo'),
+            "relator": src.get('relator_processo_nome'),
+            "orgao": src.get('orgao_julgador'),
+            "data": src.get('julgamento_data'),
+            "ementa": src.get('ementa_texto'),
+            "url_documento": src.get('inteiro_teor_url'),
+            "score": hit.get('_score')
+        }
+        # Remove campos vazios
+        item = {k: v for k, v in item.items() if v is not None}
+        resultados.append(item)
 
-@app.route('/api/document-raw/<doc_id>', methods=['GET'])
-def get_document_raw_endpoint(doc_id):
-    token = get_fresh_token()
-    if not token:
-        return jsonify({"error": "Não foi possível obter token"}), 500
-    result = get_document_raw(token, doc_id)
-    if result.get("success"):
-        return jsonify({"success": True, "document": result["data"]})
-    else:
-        if result.get("status") == 403:
-            token_cache["token"] = None
-            token = get_fresh_token(force_refresh=True)
-            if token:
-                result = get_document_raw(token, doc_id)
-                if result.get("success"):
-                    return jsonify({"success": True, "document": result["data"]})
-        return jsonify({"success": False, "error": result.get("error", "Falha")}), 500
+    return jsonify({
+        "q": query,
+        "total": result.get('result', {}).get('hits', {}).get('total', {}).get('value', len(resultados)),
+        "resultados": resultados
+    })
 
+# ============================================
+# Endpoint de saúde (opcional)
+# ============================================
 @app.route('/api/health', methods=['GET'])
 def health():
     playwright_status = False
@@ -790,110 +275,10 @@ def health():
             playwright_status = True
     except:
         pass
-    total_docs = None
-    token = get_fresh_token()
-    if token:
-        try:
-            res = search_with_token(token, {"size": 0, "track_total_hits": True})
-            if res.get("success") and res["data"].get("hits", {}).get("total"):
-                total_docs = res["data"]["hits"]["total"]["value"]
-        except:
-            pass
     return jsonify({
         "status": "healthy",
-        "timestamp": time.time(),
         "playwright_ready": playwright_status,
-        "token_cached": bool(token_cache["token"]),
-        "token_expires_in": max(0, token_cache["expires_at"] - time.time()) if token_cache["expires_at"] else 0,
-        "total_docs": total_docs,
-        "python_version": sys.version
-    })
-
-@app.route('/api/token', methods=['GET'])
-def get_token():
-    token = get_fresh_token()
-    if token:
-        return jsonify({"success": True, "token": token, "expires_in": max(0, token_cache["expires_at"] - time.time())})
-    else:
-        return jsonify({"success": False, "error": "Não foi possível obter token"}), 500
-
-@app.route('/api/clear-cache', methods=['POST'])
-def clear_cache():
-    global token_cache
-    token_cache = {"token": None, "expires_at": 0}
-    return jsonify({"success": True, "message": "Cache limpo"})
-
-# ============================================
-# ENDPOINT SIMPLIFICADO /busca (com payload exato)
-# ============================================
-@app.route('/busca')
-def busca_simplificada():
-    """
-    Endpoint público para busca simplificada de jurisprudência.
-    Usa o payload completo do STF e retorna apenas campos essenciais.
-    Exemplo: /busca?q=dano%20moral
-    """
-    query = request.args.get('q', '')
-    if not query:
-        return jsonify({"erro": "Parâmetro 'q' é obrigatório"}), 400
-
-    token = get_fresh_token()
-    if not token:
-        return jsonify({"erro": "Não foi possível obter token de acesso"}), 503
-
-    # Substitui placeholder no payload pelo termo
-    payload_str = json.dumps(PAYLOAD_BUSCA_BASE).replace("__TERMO__", query)
-    payload = json.loads(payload_str)
-
-    headers = HEADERS.copy()
-    headers['Cookie'] = f'aws-waf-token={token}'
-
-    try:
-        response = requests.post(URL_API, headers=headers, json=payload, verify=False, timeout=30)
-    except Exception as e:
-        return jsonify({"erro": f"Falha na comunicação com a API: {str(e)}"}), 502
-
-    # Se recebeu 202, tenta renovar e faz segunda tentativa
-    if response.status_code == 202:
-        logger.info("Recebido status 202, forçando renovação do token...")
-        token = get_fresh_token(force_refresh=True)
-        if token:
-            headers['Cookie'] = f'aws-waf-token={token}'
-            try:
-                response = requests.post(URL_API, headers=headers, json=payload, verify=False, timeout=30)
-            except Exception as e:
-                return jsonify({"erro": f"Falha na comunicação com a API após renovação: {str(e)}"}), 502
-        else:
-            return jsonify({"erro": "Não foi possível renovar o token"}), 503
-
-    if response.status_code != 200:
-        return jsonify({"erro": f"API retornou status {response.status_code}"}), response.status_code
-
-    data = response.json()
-    hits = data.get('result', {}).get('hits', {}).get('hits', [])
-
-    resultados = []
-    for hit in hits:
-        source = hit.get('_source', {})
-        item = {
-            "id": source.get('id') or hit.get('_id'),
-            "titulo": source.get('titulo'),
-            "processo": source.get('processo_codigo_completo'),
-            "relator": source.get('relator_processo_nome'),
-            "orgao": source.get('orgao_julgador'),
-            "data": source.get('julgamento_data'),
-            "ementa": source.get('ementa_texto'),
-            "url_documento": source.get('inteiro_teor_url'),
-            "score": hit.get('_score')
-        }
-        # Remove campos nulos
-        item = {k: v for k, v in item.items() if v is not None}
-        resultados.append(item)
-
-    return jsonify({
-        "q": query,
-        "total": data.get('result', {}).get('hits', {}).get('total', {}).get('value', 0),
-        "resultados": resultados
+        "token_cached": bool(token_cache["token"])
     })
 
 if __name__ == '__main__':
@@ -904,13 +289,7 @@ if __name__ == '__main__':
     except:
         pass
     logger.info("="*50)
-    logger.info("🚀 Iniciando aplicação STF Jurisprudência")
+    logger.info("🚀 Iniciando app simplificado STF")
     logger.info("="*50)
-    try:
-        with sync_playwright() as p:
-            p.chromium.launch(headless=True).close()
-        logger.info("✅ Playwright pronto para uso")
-    except Exception as e:
-        logger.warning(f"⚠️ Playwright pode não estar configurado: {e}")
     port = int(os.environ.get('PORT', 7860))
     app.run(host='0.0.0.0', port=port, debug=False)