Update srt_utils.py

srt_utils.py

@@ -349,39 +349,39 @@ def process_audio_for_transcription(input_file: str) -> str:
             # Fallback will trigger below
         else:
             # Demucs output structure: output_dir / model_name / input_filename_no_ext / vocals.wav
-        input_filename = os.path.basename(input_file)
-        input_stem = os.path.splitext(input_filename)[0]
-        
-        vocals_path = os.path.join(output_dir, model, input_stem, "vocals.wav")
-        
-        if os.path.exists(vocals_path):
-            print(f"✅ Demucs sucesso: {vocals_path}")
-            
-            # Convert Wav to MP3 to save space/bandwidth if needed, 
-            # OR just return the wav if Groq supports it (Groq supports wav).
-            # Let's convert to MP3 16kHz mono to optimize upload to Groq
+            input_filename = os.path.basename(input_file)
+            input_stem = os.path.splitext(input_filename)[0]
             
-            final_output = input_file + ".vocals.mp3"
-            
-            ffmpeg_cmd = shutil.which("ffmpeg")
-            if ffmpeg_cmd:
-                 # Compress to mono mp3
-                 cmd_convert = [
-                    ffmpeg_cmd, "-y",
-                    "-i", vocals_path,
-                    "-ac", "1", "-ar", "16000",
-                    "-c:a", "libmp3lame", "-q:a", "2",
-                    final_output
-                 ]
-                 subprocess.run(cmd_convert, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
-                 
-                 # Cleanup demucs folder? Maybe keep for cache but better safe space
-                 try: shutil.rmtree(os.path.join(output_dir, model, input_stem))
-                 except: pass
-                 
-                 return final_output
+            vocals_path = os.path.join(output_dir, model, input_stem, "vocals.wav")
             
-            return vocals_path
+            if os.path.exists(vocals_path):
+                print(f"✅ Demucs sucesso: {vocals_path}")
+                
+                # Convert Wav to MP3 to save space/bandwidth if needed, 
+                # OR just return the wav if Groq supports it (Groq supports wav).
+                # Let's convert to MP3 16kHz mono to optimize upload to Groq
+                
+                final_output = input_file + ".vocals.mp3"
+                
+                ffmpeg_cmd = shutil.which("ffmpeg")
+                if ffmpeg_cmd:
+                     # Compress to mono mp3
+                     cmd_convert = [
+                        ffmpeg_cmd, "-y",
+                        "-i", vocals_path,
+                        "-ac", "1", "-ar", "16000",
+                        "-c:a", "libmp3lame", "-q:a", "2",
+                        final_output
+                     ]
+                     subprocess.run(cmd_convert, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+                     
+                     # Cleanup demucs folder? Maybe keep for cache but better safe space
+                     try: shutil.rmtree(os.path.join(output_dir, model, input_stem))
+                     except: pass
+                     
+                     return final_output
+                
+                return vocals_path
             
     except Exception as e:
         print(f"⚠️ Falha no Demucs: {e}")