habulaj commited on
Commit
7c43e5e
·
verified ·
1 Parent(s): fff5342

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -70
app.py CHANGED
@@ -22,33 +22,8 @@ async def update_cookie_if_needed(cookie_path: str, secure_1psid: str, secure_1p
22
  Tenta atualizar o cookie __Secure-1PSIDTS se necessário.
23
  Retorna o novo cookie ou o original se não precisar atualizar.
24
  """
25
- try:
26
- # Criar sessão temporária para verificar/atualizar cookie
27
- async with AsyncSession() as temp_session:
28
- # Tentar fazer requisição de rotação de cookies
29
- rotate_resp = await temp_session.post(
30
- Endpoint.ROTATE_COOKIES.value,
31
- headers=Headers.ROTATE_COOKIES.value,
32
- data='[000,"-0000000000000000000"]',
33
- cookies={
34
- "__Secure-1PSID": secure_1psid,
35
- "__Secure-1PSIDTS": secure_1psidts
36
- },
37
- timeout=20
38
- )
39
- rotate_resp.raise_for_status()
40
-
41
- # Obter novo cookie se disponível
42
- new_1psidts = rotate_resp.cookies.get("__Secure-1PSIDTS")
43
- if new_1psidts and new_1psidts != secure_1psidts:
44
- # Salvar novo cookie
45
- save_cookies(cookie_path, secure_1psid, new_1psidts, additional_cookies)
46
- print(f"✓ Cookie atualizado automaticamente antes da inicialização")
47
- return new_1psidts
48
- except Exception as e:
49
- print(f"Aviso: Não foi possível atualizar cookie proativamente: {e}")
50
- # Continuar com o cookie original
51
-
52
  return secure_1psidts
53
 
54
  async def init_chatbot(retry_count=0, max_retries=2):
@@ -81,48 +56,12 @@ async def init_chatbot(retry_count=0, max_retries=2):
81
  print(f"Chatbot inicializado com sucesso usando {cookie_path}")
82
  except (ValueError, PermissionError) as e:
83
  error_str = str(e).lower()
84
- # Verificar se é erro relacionado a cookie/autenticação
85
- if retry_count < max_retries and ("snlm0e" in error_str or "authentication" in error_str or "cookie" in error_str or "not found" in error_str):
86
- print(f"Erro ao inicializar (possível cookie expirado): {e}")
87
- print(f"Tentando atualizar cookie e reinicializar (tentativa {retry_count + 1}/{max_retries})...")
88
-
89
- try:
90
- # Carregar cookies novamente
91
- secure_1psid, secure_1psidts, additional_cookies = load_cookies(cookie_path)
92
-
93
- # Forçar atualização do cookie
94
- async with AsyncSession() as temp_session:
95
- rotate_resp = await temp_session.post(
96
- Endpoint.ROTATE_COOKIES.value,
97
- headers=Headers.ROTATE_COOKIES.value,
98
- data='[000,"-0000000000000000000"]',
99
- cookies={
100
- "__Secure-1PSID": secure_1psid,
101
- "__Secure-1PSIDTS": secure_1psidts
102
- },
103
- timeout=20
104
- )
105
- rotate_resp.raise_for_status()
106
-
107
- new_1psidts = rotate_resp.cookies.get("__Secure-1PSIDTS")
108
- if new_1psidts:
109
- # Salvar novo cookie
110
- save_cookies(cookie_path, secure_1psid, new_1psidts, additional_cookies)
111
- print(f"✓ Cookie atualizado! Reinicializando...")
112
-
113
- # Retentar inicialização
114
- return await init_chatbot(retry_count=retry_count + 1, max_retries=max_retries)
115
- else:
116
- raise ValueError("Novo cookie não encontrado na resposta de rotação")
117
- except Exception as update_error:
118
- print(f"Erro ao atualizar cookie: {update_error}")
119
- if retry_count < max_retries:
120
- # Tentar mais uma vez
121
- return await init_chatbot(retry_count=retry_count + 1, max_retries=max_retries)
122
- raise Exception(f"Falha ao inicializar chatbot após {max_retries} tentativas: {update_error}")
123
- else:
124
- print(f"Erro ao inicializar chatbot: {e}")
125
- raise
126
  except Exception as e:
127
  print(f"Erro ao inicializar chatbot: {e}")
128
  raise
@@ -619,4 +558,4 @@ async def post_response(
619
  try:
620
  os.unlink(temp_file.name)
621
  except:
622
- pass
 
22
  Tenta atualizar o cookie __Secure-1PSIDTS se necessário.
23
  Retorna o novo cookie ou o original se não precisar atualizar.
24
  """
25
+ # Não tentar atualizar proativamente - deixar o sistema fazer isso quando necessário
26
+ # Isso evita erros 401/404 quando o cookie já expirou
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
27
  return secure_1psidts
28
 
29
  async def init_chatbot(retry_count=0, max_retries=2):
 
56
  print(f"Chatbot inicializado com sucesso usando {cookie_path}")
57
  except (ValueError, PermissionError) as e:
58
  error_str = str(e).lower()
59
+ # Se o erro é relacionado a cookie expirado, não tentar atualizar novamente
60
+ # O sistema tentou atualizar automaticamente e falhou
61
+ print(f"Erro ao inicializar chatbot: {e}")
62
+ print(f"AVISO: Cookies podem estar expirados. Por favor, atualize manualmente os cookies em {cookie_path}")
63
+ print(f"Para atualizar: acesse https://gemini.google.com/app e copie os novos cookies __Secure-1PSID e __Secure-1PSIDTS")
64
+ raise
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
65
  except Exception as e:
66
  print(f"Erro ao inicializar chatbot: {e}")
67
  raise
 
558
  try:
559
  os.unlink(temp_file.name)
560
  except:
561
+ pass