Update gemini_client/core.py

gemini_client/core.py

@@ -346,30 +346,39 @@ class AsyncChatbot:
             snlm0e_match = re.search(r'''["']SNlM0e["']\s*:\s*["'](.*?)["']''', resp.text)
             if not snlm0e_match:
                 # Se não encontrou SNlM0e, pode ser cookie expirado - tentar atualizar
-                console.log("[yellow]SNlM0e não encontrado na resposta. Possível cookie expirado. Tentando atualizar cookie...[/yellow]")
+                console.log("[yellow]SNlM0e não encontrado na resposta. Tentando atualizar cookie...[/yellow]")
                 try:
-                    await self.__rotate_cookies()
-                    # Retentar buscar SNlM0e após atualizar cookie
-                    resp = await self.session.get(
-                        Endpoint.INIT.value,
-                        timeout=self.timeout
-                    )
-                    resp.raise_for_status()
+                    # Tentar atualizar cookie
+                    new_cookie = await self.__rotate_cookies()
                     
-                    if "Sign in to continue" in resp.text or "accounts.google.com" in str(resp.url):
-                        raise PermissionError("Authentication failed. Cookies might be invalid or expired. Please update them.")
+                    # Se obteve novo cookie, atualizar na sessão e tentar novamente
+                    if new_cookie and new_cookie != self.secure_1psidts:
+                        # Retentar buscar SNlM0e após atualizar cookie
+                        resp = await self.session.get(
+                            Endpoint.INIT.value,
+                            timeout=self.timeout
+                        )
+                        resp.raise_for_status()
+                        
+                        if "Sign in to continue" in resp.text or "accounts.google.com" in str(resp.url):
+                            raise PermissionError("Authentication failed. Cookies might be invalid or expired. Please update them.")
+                        
+                        snlm0e_match = re.search(r'''["']SNlM0e["']\s*:\s*["'](.*?)["']''', resp.text)
+                        if snlm0e_match:
+                            console.log("[green]✓ SNlM0e obtido com sucesso após atualização do cookie![/green]")
+                            return snlm0e_match.group(1)
                     
-                    snlm0e_match = re.search(r'''["']SNlM0e["']\s*:\s*["'](.*?)["']''', resp.text)
-                    if snlm0e_match:
-                        console.log("[green]✓ SNlM0e obtido com sucesso após atualização do cookie![/green]")
-                        return snlm0e_match.group(1)
+                    # Se não conseguiu atualizar ou ainda não encontrou SNlM0e
+                    error_message = "SNlM0e value not found in response."
+                    if resp.status_code == 429:
+                        error_message += " Rate limit likely exceeded."
                     else:
-                        error_message = "SNlM0e value not found in response after cookie update."
-                        if resp.status_code == 429:
-                            error_message += " Rate limit likely exceeded."
-                        else:
-                            error_message += f" Response status: {resp.status_code}. Check cookie validity and network."
-                        raise ValueError(error_message)
+                        error_message += (
+                            f" Response status: {resp.status_code}. "
+                            f"Por favor, acesse https://gemini.google.com/app e copie os novos cookies "
+                            f"__Secure-1PSID e __Secure-1PSIDTS manualmente."
+                        )
+                    raise ValueError(error_message)
                 except ValueError as rotate_error:
                     # Re-raise ValueError (já tem mensagem clara sobre atualização manual)
                     raise
@@ -434,24 +443,55 @@ class AsyncChatbot:
         """
         Rotates the __Secure-1PSIDTS cookie and saves it automatically.
         Returns the new cookie value if successful.
+        Usa todos os cookies disponíveis para autenticação.
         """
         try:
             console.log("[yellow]Atualizando cookie __Secure-1PSIDTS automaticamente...[/yellow]")
             
-            # Tentar método 1: Endpoint de rotação direto
+            # Método mais confiável: Acessar a página do Gemini para obter novos cookies
+            # Isso funciona porque o Google atualiza o cookie automaticamente quando você acessa a página
+            console.log("[cyan]Acessando página do Gemini para obter novo cookie...[/cyan]")
+            init_response = await self.session.get(
+                Endpoint.INIT.value,
+                timeout=self.timeout
+            )
+            init_response.raise_for_status()
+            
+            # Verificar se obteve novo cookie na resposta
+            new_1psidts = init_response.cookies.get("__Secure-1PSIDTS")
+            if new_1psidts and new_1psidts != self.secure_1psidts:
+                self.secure_1psidts = new_1psidts
+                self.session.cookies.set("__Secure-1PSIDTS", new_1psidts)
+                
+                if self.cookie_path:
+                    try:
+                        save_cookies(
+                            self.cookie_path,
+                            self.secure_1psid,
+                            self.secure_1psidts,
+                            self.additional_cookies
+                        )
+                        console.log("[green]✓ Cookie atualizado e salvo automaticamente via acesso à página![/green]")
+                    except Exception as save_error:
+                        console.log(f"[yellow]Aviso: Cookie atualizado mas não foi possível salvar: {save_error}[/yellow]")
+                
+                return new_1psidts
+            
+            # Se não obteve novo cookie, tentar endpoint de rotação como fallback
+            console.log("[yellow]Não obteve novo cookie via página, tentando endpoint de rotação...[/yellow]")
             try:
                 response = await self.session.post(
                     Endpoint.ROTATE_COOKIES.value,
                     headers={
                         "Content-Type": "application/json",
                         "Origin": "https://accounts.google.com",
-                        "Referer": "https://accounts.google.com/"
+                        "Referer": "https://accounts.google.com/",
+                        "X-Requested-With": "XMLHttpRequest"
                     },
                     json=[0, "-0000000000000000000"],
                     timeout=self.timeout
                 )
                 
-                # Se retornou 200, verificar se tem novo cookie
                 if response.status_code == 200:
                     new_1psidts = response.cookies.get("__Secure-1PSIDTS")
                     if new_1psidts:
@@ -466,107 +506,29 @@ class AsyncChatbot:
                                     self.secure_1psidts,
                                     self.additional_cookies
                                 )
-                                console.log("[green]✓ Cookie atualizado e salvo automaticamente![/green]")
+                                console.log("[green]✓ Cookie atualizado via endpoint de rotação![/green]")
                             except Exception as save_error:
                                 console.log(f"[yellow]Aviso: Cookie atualizado mas não foi possível salvar: {save_error}[/yellow]")
                         
                         return new_1psidts
                 
-                # Se não funcionou, tentar método alternativo: acessar Gemini para obter novo cookie
-                if response.status_code in (401, 403, 404):
-                    console.log("[yellow]Endpoint de rotação não disponível, tentando método alternativo...[/yellow]")
-                    # Acessar a página do Gemini para obter novos cookies
-                    init_response = await self.session.get(
-                        Endpoint.INIT.value,
-                        timeout=self.timeout
-                    )
-                    
-                    # Verificar se obteve novo cookie na resposta
-                    new_1psidts = init_response.cookies.get("__Secure-1PSIDTS")
-                    if new_1psidts and new_1psidts != self.secure_1psidts:
-                        self.secure_1psidts = new_1psidts
-                        self.session.cookies.set("__Secure-1PSIDTS", new_1psidts)
-                        
-                        if self.cookie_path:
-                            try:
-                                save_cookies(
-                                    self.cookie_path,
-                                    self.secure_1psid,
-                                    self.secure_1psidts,
-                                    self.additional_cookies
-                                )
-                                console.log("[green]✓ Cookie atualizado via método alternativo![/green]")
-                            except Exception as save_error:
-                                console.log(f"[yellow]Aviso: Cookie atualizado mas não foi possível salvar: {save_error}[/yellow]")
-                        
-                        return new_1psidts
-                    else:
-                        raise ValueError("Não foi possível obter novo cookie - cookies podem estar completamente expirados")
-                
-                response.raise_for_status()
+                # Se endpoint retornou 404, pode ser que não exista mais ou mudeu
+                if response.status_code == 404:
+                    console.log("[yellow]Endpoint de rotação retornou 404 - pode ter mudado ou não estar disponível[/yellow]")
                 
             except Exception as rotate_error:
-                # Se falhou, tentar método alternativo apenas se não for 401/403/404
-                error_str = str(rotate_error).lower()
-                if "401" in error_str or "403" in error_str or "404" in error_str:
-                    # Se for erro de autenticação, cookies estão completamente expirados
-                    raise ValueError(
-                        "Cookies completamente expirados. Não é possível atualizar automaticamente. "
-                        "Por favor, acesse https://gemini.google.com/app e copie os novos cookies "
-                        "__Secure-1PSID e __Secure-1PSIDTS manualmente."
-                    )
-                
-                # Tentar método alternativo apenas para outros erros
-                console.log(f"[yellow]Método de rotação falhou ({rotate_error}), tentando método alternativo...[/yellow]")
-                try:
-                    # Acessar a página do Gemini para obter novos cookies
-                    init_response = await self.session.get(
-                        Endpoint.INIT.value,
-                        timeout=self.timeout
-                    )
-                    
-                    new_1psidts = init_response.cookies.get("__Secure-1PSIDTS")
-                    if new_1psidts and new_1psidts != self.secure_1psidts:
-                        self.secure_1psidts = new_1psidts
-                        self.session.cookies.set("__Secure-1PSIDTS", new_1psidts)
-                        
-                        if self.cookie_path:
-                            try:
-                                save_cookies(
-                                    self.cookie_path,
-                                    self.secure_1psid,
-                                    self.secure_1psidts,
-                                    self.additional_cookies
-                                )
-                                console.log("[green]✓ Cookie atualizado via método alternativo![/green]")
-                            except Exception as save_error:
-                                console.log(f"[yellow]Aviso: Cookie atualizado mas não foi possível salvar: {save_error}[/yellow]")
-                        
-                        return new_1psidts
-                    else:
-                        raise ValueError(
-                            "Não foi possível obter novo cookie. Cookies podem estar completamente expirados. "
-                            "Por favor, acesse https://gemini.google.com/app e copie os novos cookies manualmente."
-                        )
-                except Exception as alt_error:
-                    console.log(f"[red]Método alternativo também falhou: {alt_error}[/red]")
-                    raise ValueError(
-                        f"Não foi possível atualizar cookie automaticamente. "
-                        f"Por favor, acesse https://gemini.google.com/app e copie os novos cookies "
-                        f"__Secure-1PSID e __Secure-1PSIDTS manualmente. Erro: {rotate_error}"
-                    )
+                console.log(f"[yellow]Endpoint de rotação falhou: {rotate_error}[/yellow]")
+            
+            # Se não conseguiu atualizar, retornar o cookie atual (pode ainda ser válido)
+            console.log("[yellow]Não foi possível obter novo cookie - usando cookie atual[/yellow]")
+            return self.secure_1psidts
             
-            raise ValueError("Novo cookie __Secure-1PSIDTS não encontrado na resposta")
-        except ValueError:
-            # Re-raise ValueError sem modificar (já tem mensagem clara)
-            raise
         except Exception as e:
             console.log(f"[red]Falha ao atualizar cookie: {e}[/red]")
-            raise ValueError(
-                f"Não foi possível atualizar cookie automaticamente. "
-                f"Por favor, acesse https://gemini.google.com/app e copie os novos cookies "
-                f"__Secure-1PSID e __Secure-1PSIDTS manualmente. Erro: {e}"
-            )
+            # Se falhou completamente, retornar o cookie atual em vez de falhar
+            # Isso permite que o sistema continue funcionando mesmo se não conseguir atualizar
+            console.log("[yellow]Retornando cookie atual - sistema continuará funcionando[/yellow]")
+            return self.secure_1psidts
 
 
     async def ask(