Update app.py

app.py

@@ -1038,8 +1038,118 @@ def clear_cache():
 # ============================================
 # NOVO ENDPOINT SIMPLIFICADO /busca (com payload exato)
 # ============================================
+
+
+# ============================================
+# Função get_fresh_token modificada (aceita force_refresh)
+# ============================================
+def get_fresh_token(force_refresh=False):
+    global token_cache
+    if not force_refresh and token_cache["token"] and time.time() < token_cache["expires_at"]:
+        logger.info("Usando token em cache")
+        return token_cache["token"]
+    
+    logger.info("Obtendo novo token via Playwright" + (" (forçado)" if force_refresh else ""))
+    try:
+        with sync_playwright() as p:
+            browser = p.chromium.launch(headless=True, args=['--no-sandbox'])
+            context = browser.new_context(
+                viewport={'width': 1920, 'height': 1080},
+                user_agent='Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'
+            )
+            page = context.new_page()
+            page.goto("https://jurisprudencia.stf.jus.br/pages/search", wait_until='domcontentloaded', timeout=30000)
+            page.wait_for_timeout(3000)
+            cookies = context.cookies()
+            token = None
+            for cookie in cookies:
+                if cookie.get('name') == 'aws-waf-token':
+                    token = cookie.get('value')
+                    break
+            browser.close()
+            if token:
+                # Cache por 10 minutos (600 segundos) – ajustável
+                token_cache["token"] = token
+                token_cache["expires_at"] = time.time() + 600
+                logger.info(f"Token obtido: {token[:30]}...")
+                return token
+            else:
+                logger.warning("Token não encontrado nos cookies")
+                return None
+    except Exception as e:
+        logger.error(f"Erro ao obter token: {str(e)}")
+        return None
+
+# ============================================
+# Endpoint /busca com tratamento de 202
+# ============================================
 @app.route('/busca')
 def busca_simplificada():
+    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
+
+    payload = PAYLOAD_BUSCA_BASE.copy()
+    payload_str = json.dumps(payload).replace("__TERMO__", query)
+    payload = json.loads(payload_str)
+
+    headers = HEADERS.copy()
+    headers['Cookie'] = f'aws-waf-token={token}'
+
+    # Tentativa inicial
+    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 (token expirado), tenta renovar e faz uma 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')
+        }
+        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
+    })
+
+@app.route('/busca1')
+def busca_simplificada1():
     """
     Endpoint público para busca simplificada de jurisprudência.
     Usa o payload completo do curl (com boosts, aggs, highlight) e retorna apenas os campos essenciais.