Update app.py

app.py

@@ -6,6 +6,7 @@ import tempfile
 from pathlib import Path
 import re
 import requests
+import time
 
 from gemini_client import AsyncChatbot, Model, load_cookies
 
@@ -210,6 +211,82 @@ def clean_and_validate_srt(srt_content):
     
     return srt_corrigido.strip()
 
+def download_file_with_retry(url: str, max_retries: int = 3, timeout: int = 300):
+    """
+    Baixa arquivo com retry logic e tratamento de rate limiting.
+    
+    Parâmetros:
+    - url: URL do arquivo
+    - max_retries: Número máximo de tentativas
+    - timeout: Timeout em segundos
+    
+    Retorna: Response object do requests
+    """
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
+        'Accept': '*/*',
+        'Accept-Language': 'en-US,en;q=0.9',
+        'Accept-Encoding': 'gzip, deflate, br',
+        'Connection': 'keep-alive',
+        'Upgrade-Insecure-Requests': '1'
+    }
+    
+    for attempt in range(max_retries):
+        try:
+            if attempt > 0:
+                # Backoff exponencial: 2^attempt segundos
+                wait_time = 2 ** attempt
+                print(f"⏳ Aguardando {wait_time}s antes de tentar novamente (tentativa {attempt + 1}/{max_retries})...")
+                time.sleep(wait_time)
+            
+            print(f"📥 Tentativa {attempt + 1}/{max_retries} - Baixando arquivo de: {url}")
+            response = requests.get(url, headers=headers, timeout=timeout, stream=True)
+            
+            # Tratar erro 429 (Too Many Requests)
+            if response.status_code == 429:
+                retry_after = response.headers.get('Retry-After')
+                if retry_after:
+                    wait_time = int(retry_after)
+                    print(f"⚠️ Rate limit atingido. Aguardando {wait_time}s conforme Retry-After header...")
+                    time.sleep(wait_time)
+                elif attempt < max_retries - 1:
+                    # Se não houver Retry-After, usar backoff exponencial
+                    wait_time = (2 ** attempt) * 5  # 5s, 10s, 20s...
+                    print(f"⚠️ Rate limit atingido. Aguardando {wait_time}s antes de tentar novamente...")
+                    time.sleep(wait_time)
+                    continue
+                else:
+                    raise HTTPException(
+                        status_code=429,
+                        detail=f"Rate limit atingido após {max_retries} tentativas. Tente novamente mais tarde."
+                    )
+            
+            response.raise_for_status()
+            return response
+            
+        except requests.exceptions.HTTPError as e:
+            if e.response.status_code == 429 and attempt < max_retries - 1:
+                continue
+            elif attempt == max_retries - 1:
+                raise HTTPException(
+                    status_code=400,
+                    detail=f"Erro ao baixar arquivo após {max_retries} tentativas: {str(e)}"
+                )
+            else:
+                raise
+        except requests.exceptions.RequestException as e:
+            if attempt == max_retries - 1:
+                raise HTTPException(
+                    status_code=400,
+                    detail=f"Erro ao baixar arquivo após {max_retries} tentativas: {str(e)}"
+                )
+            continue
+    
+    raise HTTPException(
+        status_code=400,
+        detail=f"Falha ao baixar arquivo após {max_retries} tentativas"
+    )
+
 @app.get("/subtitle")
 async def generate_subtitle(
     file: str,
@@ -237,10 +314,8 @@ async def generate_subtitle(
     
     temp_file = None
     try:
-        # Baixar arquivo da URL
-        print(f"📥 Baixando arquivo de: {file}")
-        response = requests.get(file, timeout=300, stream=True)
-        response.raise_for_status()
+        # Baixar arquivo da URL com retry
+        response = download_file_with_retry(file, max_retries=3, timeout=300)
         
         # Determinar tipo de mídia e extensão
         content_type = response.headers.get('content-type', '').lower()
@@ -467,4 +542,4 @@ TRADUZA TUDO DE IMPORTANTE NO {media_desc_final}, que tenha dialogo... Nunca dei
             try:
                 os.unlink(temp_file.name)
             except:
-                pass
+                pass