Update main.py

main.py

@@ -217,6 +217,7 @@ class ProcessUrlRequest(BaseModel):
     version: str  # "recurvepop" ou "girlsmoodaily"
     raw_url: Optional[bool] = True
     text_cut: Optional[bool] = True
+    translate: Optional[bool] = True
 
 @app.post("/process-url")
 async def process_url_endpoint(request: ProcessUrlRequest, background_tasks: BackgroundTasks):
@@ -634,7 +635,7 @@ Objetivo: O foco é detectar se o vídeo foi "re-editado" com textos informativo
 
                             # Legendas
                             needs_legenda = result_data.get("legenda", False)
-                            if needs_legenda:
+                            if needs_legenda and request.translate:
                                 print(f"🎙️ [{time.time()-t_start:.1f}s] Resolvendo legendas (Groq task + Gemini translate)...")
                                 try:
                                     if groq_task:
@@ -1215,7 +1216,7 @@ async def process_account_endpoint(account: str):
                         # 5. Gerar legendas se necessário
                         needs_legenda = result_data.get("legenda", False)
                         srt_for_export = None
-                        if needs_legenda:
+                        if needs_legenda and record.get("translate", True):
                             print(f"🎙️ [{time.time()-t_start:.1f}s] Resolvendo legendas (Groq task + Gemini translate)...")
                             try:
                                 if groq_task:
@@ -2362,6 +2363,7 @@ async def generate_subtitle_groq(request: GroqRequest):
 class GeminiSubtitleRequest(BaseModel):
     url: str
     has_bg_music: Optional[bool] = False
+    translate: Optional[bool] = True
     context: Optional[str] = "N/A"
     model: Optional[str] = "flash"
     time_start: Optional[float] = None
@@ -2372,9 +2374,23 @@ async def generate_subtitle(request: GeminiSubtitleRequest):
     if not client: raise HTTPException(status_code=500, detail="Gemini client is not initialized")
     try:
         srt_filtered, _, audio_url, word_level_text = await get_groq_srt_base(
-            request.url, "en", 0.4, request.has_bg_music, request.time_start, request.time_end
+            request.url, None, 0.4, request.has_bg_music, request.time_start, request.time_end
         )
         
+        # Se não for solicitado tradução/limpeza via Gemini, retorna o original direto
+        if not request.translate:
+            final_srt = srt_filtered
+            if request.time_start and request.time_start > 0:
+                final_srt = shift_srt_timestamps(final_srt, request.time_start)
+            
+            return JSONResponse(content={
+                "srt": final_srt,
+                "original_srt": final_srt,
+                "srt_word_level": word_level_text,
+                "used_audio_processed": True,
+                "translated": False
+            })
+        
         filename = audio_url.split("/")[-1]
         processed_audio_path = os.path.join("static", filename)
         if not os.path.exists(processed_audio_path): processed_audio_path = os.path.join("static", "processed", filename)