Update app.py

app.py

@@ -7,6 +7,9 @@ from pathlib import Path
 import re
 import requests
 import time
+import base64
+import io
+from PIL import Image
 
 from gemini_client import AsyncChatbot, Model, load_cookies
 
@@ -559,6 +562,127 @@ TRADUZA TUDO DE IMPORTANTE NO {media_desc_final}, que tenha dialogo... Nunca dei
                 pass
 
 
+def flip_image_both_axes(image_path: str) -> str:
+    """
+    Inverte uma imagem horizontalmente e verticalmente.
+    Retorna o caminho para um novo arquivo temporário com a imagem invertida.
+    """
+    with Image.open(image_path) as img:
+        # Inverter horizontalmente e verticalmente
+        flipped = img.transpose(Image.FLIP_LEFT_RIGHT).transpose(Image.FLIP_TOP_BOTTOM)
+        
+        # Salvar em arquivo temporário
+        temp_flipped = tempfile.NamedTemporaryFile(delete=False, suffix=Path(image_path).suffix)
+        flipped.save(temp_flipped.name)
+        temp_flipped.close()
+        
+        return temp_flipped.name
+
+
+def flip_base64_image_both_axes(img_base64: str, content_type: str) -> str:
+    """
+    Inverte uma imagem base64 horizontalmente e verticalmente.
+    Retorna o base64 da imagem invertida.
+    """
+    # Decodificar base64 para bytes
+    img_bytes = base64.b64decode(img_base64)
+    
+    # Abrir imagem dos bytes
+    with Image.open(io.BytesIO(img_bytes)) as img:
+        # Inverter horizontalmente e verticalmente
+        flipped = img.transpose(Image.FLIP_LEFT_RIGHT).transpose(Image.FLIP_TOP_BOTTOM)
+        
+        # Salvar em buffer
+        buffer = io.BytesIO()
+        # Determinar formato baseado no content_type
+        img_format = 'PNG'
+        if 'jpeg' in content_type or 'jpg' in content_type:
+            img_format = 'JPEG'
+        elif 'webp' in content_type:
+            img_format = 'WEBP'
+        
+        flipped.save(buffer, format=img_format)
+        buffer.seek(0)
+        
+        # Converter de volta para base64
+        return base64.b64encode(buffer.read()).decode('utf-8')
+
+
+async def _try_upscale(chatbot, image_path: str, prompt: str):
+    """
+    Tenta fazer upscale de uma imagem.
+    Retorna (result, upscaled_url) ou (None, None) em caso de erro.
+    """
+    result = await chatbot.ask(prompt, image=image_path)
+    
+    if result.get("error"):
+        return None, None
+    
+    upscaled_url = None
+    if result.get("images"):
+        # Preferir imagens geradas
+        for img in result["images"]:
+            if "[Generated Image" in img.get("title", ""):
+                upscaled_url = img["url"]
+                break
+        
+        # Se não achou 'Generated Image', pega a última
+        if not upscaled_url and len(result["images"]) > 0:
+            upscaled_url = result["images"][-1]["url"]
+    
+    return result, upscaled_url
+
+
+async def _download_upscaled_image(upscaled_url: str, cookies_dict: dict) -> tuple:
+    """
+    Baixa a imagem upscaled e retorna (img_base64, content_type, download_url).
+    """
+    print(f"📥 Resolvendo redirect da URL...")
+    
+    # Primeira requisição: seguir redirect para obter URL final COM cookies
+    redirect_response = requests.get(upscaled_url, timeout=60, allow_redirects=True, 
+        cookies=cookies_dict,
+        headers={
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
+            'Referer': 'https://gemini.google.com/',
+            'Accept': 'image/*,*/*'
+        }
+    )
+    
+    # A URL final após redirect
+    final_url = redirect_response.url
+    print(f"📍 URL após redirect: {final_url[:80]}...")
+    
+    # Remover parâmetros existentes e adicionar =s0-d-I para full resolution
+    if "?" in final_url:
+        final_url = final_url.split("?")[0]
+    # Remover qualquer =sXXX existente
+    if "=s" in final_url:
+        final_url = final_url.rsplit("=s", 1)[0]
+    
+    download_url = final_url + "=s0-d-I"
+    print(f"📥 Baixando imagem full res de: {download_url[:80]}...")
+    
+    # Segunda requisição: baixar a imagem em full resolution COM cookies
+    img_response = requests.get(download_url, timeout=60, 
+        cookies=cookies_dict,
+        headers={
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
+            'Referer': 'https://gemini.google.com/',
+            'Accept': 'image/*,*/*'
+        }
+    )
+    img_response.raise_for_status()
+    
+    # Converter para base64
+    img_base64 = base64.b64encode(img_response.content).decode('utf-8')
+    content_type = img_response.headers.get('content-type', 'image/png')
+    
+    print(f"✅ Imagem baixada e convertida para base64 ({len(img_base64)} chars)")
+    
+    return img_base64, content_type, download_url
+
+
 @app.get("/upscale")
 async def upscale_image(
     file: str
@@ -566,11 +690,15 @@ async def upscale_image(
     """
     Endpoint para fazer upscale 4x de uma imagem usando o Nano Banana Pro.
     
+    Se a primeira tentativa falhar (ex: erro de figuras públicas), 
+    tenta novamente invertendo a imagem horizontalmente e verticalmente,
+    e depois inverte o resultado de volta.
+    
     Parâmetros:
     - file: URL da imagem
     
     Retorna:
-    - JSON com a URL da imagem gerada (upscaled)
+    - JSON com a imagem gerada em base64 (upscaled)
     """
     if upscale_chatbot is None:
         raise HTTPException(status_code=500, detail="Upscale Chatbot não inicializado")
@@ -579,6 +707,8 @@ async def upscale_image(
         raise HTTPException(status_code=400, detail="Parâmetro 'file' é obrigatório")
     
     temp_file = None
+    temp_flipped_file = None
+    
     try:
         # Baixar arquivo da URL com retry
         response = download_file_with_retry(file, max_retries=3, timeout=300)
@@ -603,46 +733,7 @@ async def upscale_image(
         
         prompt = "Upscale this image by 4x. Keep everything exactly as it is, including text, colors, texture, aspect ratio, zoom and proportions. Just increase the quality and sharpness. Do not add any new elements or modify the composition"
         
-        # Enviar para o Gemini (Nano Banana)
-        print(f"🧠 Enviando imagem para upscale...")
-        result = await upscale_chatbot.ask(prompt, image=temp_file.name)
-        
-        if result.get("error"):
-            raise HTTPException(
-                status_code=500,
-                detail=f"Erro ao gerar upscale: {result.get('content', 'Erro desconhecido')}"
-            )
-        
-        # Tentar extrair imagem gerada (geralmente formato 2 no core.py)
-        upscaled_url = None
-        
-        # Debug do resultado
-        # print("Resultado completo:", result)
-
-        if result.get("images"):
-            # Preferir imagens geradas
-            for img in result["images"]:
-                if "[Generated Image" in img.get("title", ""):
-                    upscaled_url = img["url"]
-                    break
-            
-            # Se não achou 'Generated Image', pega a última (comportamento padrão)
-            if not upscaled_url and len(result["images"]) > 0:
-                upscaled_url = result["images"][-1]["url"]
-
-        if not upscaled_url:
-             raise HTTPException(
-                status_code=500,
-                detail="Nenhuma imagem de upscale foi retornada pelo modelo."
-            )
-        # Primeiro, acessar a URL original para obter o redirect
-        # A URL gg-dl redireciona para rd-gg-dl
-        print(f"📥 Resolvendo redirect da URL...")
-        
-        import base64
-        
-        # Carregar cookies do arquivo para usar nas requisições autenticadas
-        # Usa load_cookies que suporta JSON e Netscape
+        # Carregar cookies para download
         cookies_dict = {}
         cookie_path = os.getenv("COOKIE_PATH", "cookies.json")
         try:
@@ -654,62 +745,58 @@ async def upscale_image(
         except Exception as cookie_err:
             print(f"⚠️ Aviso: Não foi possível carregar cookies: {cookie_err}")
         
-        try:
-            # Primeira requisição: seguir redirect para obter URL final COM cookies
-            redirect_response = requests.get(upscaled_url, timeout=60, allow_redirects=True, 
-                cookies=cookies_dict,
-                headers={
-                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
-                    'Referer': 'https://gemini.google.com/',
-                    'Accept': 'image/*,*/*'
-                }
-            )
-            
-            # A URL final após redirect
-            final_url = redirect_response.url
-            print(f"📍 URL após redirect: {final_url[:80]}...")
-            
-            # Remover parâmetros existentes e adicionar =s0-d-I para full resolution
-            if "?" in final_url:
-                final_url = final_url.split("?")[0]
-            # Remover qualquer =sXXX existente
-            if "=s" in final_url:
-                final_url = final_url.rsplit("=s", 1)[0]
-            
-            download_url = final_url + "=s0-d-I"
-            print(f"📥 Baixando imagem full res de: {download_url[:80]}...")
+        # ========== PRIMEIRA TENTATIVA: Normal ==========
+        print(f"🧠 [Tentativa 1] Enviando imagem para upscale...")
+        result, upscaled_url = await _try_upscale(upscale_chatbot, temp_file.name, prompt)
+        
+        used_flip_workaround = False
+        
+        if result is None or upscaled_url is None:
+            # ========== SEGUNDA TENTATIVA: Inverter imagem ==========
+            print(f"⚠️ Primeira tentativa falhou. Tentando workaround de inversão...")
             
-            # Segunda requisição: baixar a imagem em full resolution COM cookies
-            img_response = requests.get(download_url, timeout=60, 
-                cookies=cookies_dict,
-                headers={
-                    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36',
-                    'Referer': 'https://gemini.google.com/',
-                    'Accept': 'image/*,*/*'
-                }
-            )
-            img_response.raise_for_status()
+            # Inverter a imagem horizontalmente e verticalmente
+            print(f"🔄 Invertendo imagem (horizontal + vertical)...")
+            temp_flipped_file = flip_image_both_axes(temp_file.name)
+            print(f"✅ Imagem invertida salva em: {temp_flipped_file}")
             
-            # Converter para base64
-            img_base64 = base64.b64encode(img_response.content).decode('utf-8')
-            content_type = img_response.headers.get('content-type', 'image/png')
+            # Tentar novamente com a imagem invertida
+            print(f"🧠 [Tentativa 2] Enviando imagem INVERTIDA para upscale...")
+            result, upscaled_url = await _try_upscale(upscale_chatbot, temp_flipped_file, prompt)
             
-            print(f"✅ Imagem baixada e convertida para base64 ({len(img_base64)} chars)")
+            if result is None or upscaled_url is None:
+                raise HTTPException(
+                    status_code=500,
+                    detail="Nenhuma imagem de upscale foi retornada pelo modelo após múltiplas tentativas."
+                )
             
+            used_flip_workaround = True
+            print(f"✅ Segunda tentativa (com inversão) funcionou!")
+        
+        # Baixar imagem upscaled
+        try:
+            img_base64, img_content_type, download_url = await _download_upscaled_image(upscaled_url, cookies_dict)
         except Exception as download_error:
             print(f"❌ Erro ao baixar imagem: {download_error}")
             raise HTTPException(
                 status_code=500,
                 detail=f"Erro ao baixar imagem upscaled: {str(download_error)}"
             )
+        
+        # Se usamos o workaround de inversão, precisamos inverter o resultado de volta
+        if used_flip_workaround:
+            print(f"🔄 Invertendo resultado de volta (restaurando orientação original)...")
+            img_base64 = flip_base64_image_both_axes(img_base64, img_content_type)
+            print(f"✅ Imagem restaurada para orientação original")
             
         return JSONResponse(
             content={
                 "image_base64": img_base64,
-                "content_type": content_type,
+                "content_type": img_content_type,
                 "success": True,
                 "original_url": file,
-                "upscaled_url": download_url
+                "upscaled_url": download_url,
+                "used_flip_workaround": used_flip_workaround
             }
         )
 
@@ -720,9 +807,14 @@ async def upscale_image(
         traceback.print_exc()
         raise HTTPException(status_code=500, detail=f"Erro interno no upscale: {str(e)}")
     finally:
-        # Limpar arquivo temporário
+        # Limpar arquivos temporários
         if temp_file and os.path.exists(temp_file.name):
             try:
                 os.unlink(temp_file.name)
+            except:
+                pass
+        if temp_flipped_file and os.path.exists(temp_flipped_file):
+            try:
+                os.unlink(temp_flipped_file)
             except:
                 pass