Update conver.py

conver.py

@@ -10,6 +10,7 @@ import tempfile
 from pydub import AudioSegment
 import base64
 from pathlib import Path
+import shutil  # Importamos shutil para manejo de directorios
 
 @dataclass
 class ConversationConfig:
@@ -81,7 +82,7 @@ class URLToAudioConverter:
         
         try:
             for i, turn in enumerate(conversation_json["conversation"]):
-                filename = output_dir / f"output_{i}.mp3"  # Cambiado a MP3
+                filename = output_dir / f"output_{i}.mp3"
                 voice = voice_1 if i % 2 == 0 else voice_2
 
                 tmp_path, error = await self._generate_audio(turn["text"], voice)
@@ -106,7 +107,7 @@ class URLToAudioConverter:
         pitch_str = f"{pitch:+d}Hz"
         communicate = edge_tts.Communicate(text, voice_short_name, rate=rate_str, pitch=pitch_str)
 
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:  # Cambiado a MP3
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp3") as tmp_file:
             tmp_path = tmp_file.name
             await communicate.save(tmp_path)
 
@@ -125,17 +126,29 @@ class URLToAudioConverter:
         try:
             combined = AudioSegment.empty()
             for filename in filenames:
-                # Leer como MP3
                 audio_segment = AudioSegment.from_file(filename, format="mp3")
                 combined += audio_segment
 
-            # Exportar como MP3
             combined.export(output_file, format="mp3")
 
-            # Limpieza de archivos temporales
-            for filename in filenames:
-                os.remove(filename)
-            os.rmdir(os.path.dirname(filenames[0]))
+            # Limpieza mejorada y robusta
+            dir_path = os.path.dirname(filenames[0])
+            
+            # Eliminar todos los archivos en el directorio
+            for file in os.listdir(dir_path):
+                file_path = os.path.join(dir_path, file)
+                if os.path.isfile(file_path):
+                    try:
+                        os.remove(file_path)
+                    except Exception as e:
+                        print(f"Warning: Could not remove file {file_path}: {str(e)}")
+            
+            # Intentar eliminar el directorio (no crítico si falla)
+            try:
+                os.rmdir(dir_path)
+            except OSError as e:
+                print(f"Info: Could not remove directory {dir_path}: {str(e)}")
+                # No es crítico, el espacio puede continuar
                 
         except Exception as e:
             raise RuntimeError(f"Failed to combine audio files: {e}")
@@ -156,7 +169,7 @@ class URLToAudioConverter:
             conversation_json, voice_1, voice_2
         )
 
-        final_output = os.path.join(folder_name, "combined_output.mp3")  # Cambiado a MP3
+        final_output = os.path.join(folder_name, "combined_output.mp3")
         self.combine_audio_files(audio_files, final_output)
         return final_output, conversation_text
 
@@ -169,6 +182,6 @@ class URLToAudioConverter:
         audio_files, folder_name = await self.text_to_speech(
             conversation_json, voice_1, voice_2
         )
-        final_output = os.path.join(folder_name, "combined_output.mp3")  # Cambiado a MP3
+        final_output = os.path.join(folder_name, "combined_output.mp3")
         self.combine_audio_files(audio_files, final_output)
         return final_output, conversation_text