Update conver.py

conver.py

@@ -11,6 +11,7 @@ import base64
 from pathlib import Path
 import time
 from threading import Thread
+import re
 
 @dataclass
 class ConversationConfig:
@@ -25,6 +26,11 @@ class URLToAudioConverter:
         self.llm_client = OpenAI(api_key=llm_api_key, base_url="https://api.together.xyz/v1")
         self.llm_out = None
         self._start_cleaner()
+        # Define audio file paths relative to the script directory
+        self.ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
+        self.MUSICA_FONDO = os.path.join(self.ROOT_DIR, "musica.mp3")
+        self.TAG1 = os.path.join(self.ROOT_DIR, "tag.mp3")
+        self.TAG2 = os.path.join(self.ROOT_DIR, "tag2.mp3")
 
     def _start_cleaner(self, max_age_hours: int = 24):
         def cleaner():
@@ -71,14 +77,17 @@ class URLToAudioConverter:
             )
             response_content = response.choices[0].message.content
             print(f"Respuesta cruda del modelo: {response_content[:500]}...")
-            json_str = response_content.strip()
-            if not json_str.startswith('{'):
-                json_str = json_str[json_str.find('{'):]
-            if not json_str.endswith('}'):
-                json_str = json_str[:json_str.rfind('}')+1]
+            
+            # Extract valid JSON using regex
+            json_match = re.search(r'\{.*\}', response_content, re.DOTALL)
+            if json_match:
+                json_str = json_match.group(0)
+            else:
+                raise ValueError("No valid JSON object found in response")
+            
             dialogue = json.loads(json_str)
             if not dialogue.get("conversation"):
-                print("Error: No se generó diálogo válido.")
+                print("Error: No valid conversation generated.")
             return dialogue
         except Exception as e:
             raise RuntimeError(f"Failed to parse dialogue: {str(e)}")
@@ -135,14 +144,26 @@ class URLToAudioConverter:
         tags_paths: List[str],
         custom_music_path: str = None
     ) -> AudioSegment:
-        music_file = custom_music_path if custom_music_path and os.path.exists(custom_music_path) else music_path
+        # Use predefined paths if no custom path is provided
+        music_file = custom_music_path if custom_music_path and os.path.exists(custom_music_path) else self.MUSICA_FONDO
+        tag_outro_file = self.TAG1
+        tag_trans_file = self.TAG2
+
+        # Check if files exist
+        if not os.path.exists(music_file):
+            raise FileNotFoundError(f"Music file not found: {music_file}")
+        if not os.path.exists(tag_outro_file):
+            raise FileNotFoundError(f"Tag file not found: {tag_outro_file}")
+        if not os.path.exists(tag_trans_file):
+            raise FileNotFoundError(f"Tag file not found: {tag_trans_file}")
+
         music = AudioSegment.from_file(music_file).fade_out(2000) - 25
         if len(music) < len(speech_audio):
             music = music * ((len(speech_audio) // len(music)) + 1)
         music = music[:len(speech_audio)]
         mixed = speech_audio.overlay(music)
-        tag_outro = AudioSegment.from_file(tags_paths[0]) - 10
-        tag_trans = AudioSegment.from_file(tags_paths[1]) - 10
+        tag_outro = AudioSegment.from_file(tag_outro_file) - 10
+        tag_trans = AudioSegment.from_file(tag_trans_file) - 10
         final_audio = mixed + tag_outro
         silent_ranges = []
         for i in range(0, len(speech_audio) - 500, 100):
@@ -180,8 +201,8 @@ class URLToAudioConverter:
         combined = self.combine_audio_files(audio_files)
         final_audio = self.add_background_music_and_tags(
             combined,
-            "musica.mp3",
-            ["tag.mp3", "tag2.mp3"],
+            self.MUSICA_FONDO,  # Use instance variable
+            [self.TAG1, self.TAG2],  # Use instance variables
             custom_music_path
         )
         output_path = os.path.join(folder_name, "podcast_final.mp3")
@@ -192,5 +213,4 @@ class URLToAudioConverter:
             f"{turn['speaker']}: {turn['text']}" 
             for turn in conversation["conversation"]
         )
-        return output_path, text_output
-
+        return output_path, text_output