Update gemini_client/core.py

gemini_client/core.py

@@ -370,13 +370,21 @@ class AsyncChatbot:
                         else:
                             error_message += f" Response status: {resp.status_code}. Check cookie validity and network."
                         raise ValueError(error_message)
+                except ValueError as rotate_error:
+                    # Re-raise ValueError (já tem mensagem clara sobre atualização manual)
+                    raise
                 except Exception as rotate_error:
                     console.log(f"[red]Falha ao atualizar cookie: {rotate_error}[/red]")
                     error_message = "SNlM0e value not found in response."
                     if resp.status_code == 429:
                         error_message += " Rate limit likely exceeded."
                     else:
-                        error_message += f" Response status: {resp.status_code}. Cookie update failed. Check cookie validity and network."
+                        error_message += (
+                            f" Response status: {resp.status_code}. "
+                            f"Falha ao atualizar cookie automaticamente. "
+                            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)
 
             # Try to refresh PSIDTS if needed
@@ -429,37 +437,136 @@ class AsyncChatbot:
         """
         try:
             console.log("[yellow]Atualizando cookie __Secure-1PSIDTS automaticamente...[/yellow]")
-            response = await self.session.post(
-                Endpoint.ROTATE_COOKIES.value,
-                headers=Headers.ROTATE_COOKIES.value,
-                data='[000,"-0000000000000000000"]',
-                timeout=self.timeout
-            )
-            response.raise_for_status()
-
-            if new_1psidts := response.cookies.get("__Secure-1PSIDTS"):
-                self.secure_1psidts = new_1psidts
-                self.session.cookies.set("__Secure-1PSIDTS", new_1psidts)
+            
+            # Tentar método 1: Endpoint de rotação direto
+            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/"
+                    },
+                    json=[0, "-0000000000000000000"],
+                    timeout=self.timeout
+                )
                 
-                # Salvar automaticamente no arquivo se cookie_path estiver definido
-                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![/green]")
-                    except Exception as save_error:
-                        console.log(f"[yellow]Aviso: Cookie atualizado mas não foi possível salvar no arquivo: {save_error}[/yellow]")
+                # Se retornou 200, verificar se tem novo cookie
+                if response.status_code == 200:
+                    new_1psidts = response.cookies.get("__Secure-1PSIDTS")
+                    if new_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![/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
                 
-                return new_1psidts
-            else:
-                raise ValueError("Novo cookie __Secure-1PSIDTS não encontrado na resposta")
+                # 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()
+                
+            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}"
+                    )
+            
+            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
+            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}"
+            )
 
 
     async def ask(