Update conver.py

conver.py

@@ -56,21 +56,29 @@ class URLToAudioConverter:
         if not text:
             raise ValueError("Input text cannot be empty")
         try:
-            prompt = f"{text}\nConvierte esto en un diálogo de podcast en español entre Anfitrión1 y Anfitrión2. Genera un diálogo extenso y detallado, con respuestas completas y naturales, como una conversación real de podcast. Devuelve SOLO:\nAnfitrión1: [texto]\nAnfitrión2: [texto]\n..."
+            prompt = (
+                f"{text}\nConvierte el texto proporcionado en un diálogo de podcast en español "
+                f"entre Anfitrión1 y Anfitrión2. Genera una conversación extensa, detallada y natural, "
+                f"como en un podcast real, con al menos 5 intercambios por hablante. "
+                f"Devuelve SOLO un objeto JSON con la siguiente estructura:\n"
+                '{"conversation": [{"speaker": "Anfitrión1", "text": "..."}, {"speaker": "Anfitrión2", "text": "..."}]}'
+            )
+            print(f"Texto de entrada: {text[:200]}...")  # Depuración
             response = self.llm_client.chat.completions.create(
                 messages=[{"role": "user", "content": prompt}],
-                model=self.config.model_name
+                model=self.config.model_name,
+                response_format={"type": "json_object"}
             )
-            raw_text = response.choices[0].message.content
-            dialogue = {"conversation": []}
-            for line in raw_text.split('\n'):
-                if ':' in line:
-                    speaker, _, content = line.partition(':')
-                    if speaker.strip() in ("Anfitrión1", "Anfitrión2"):
-                        dialogue["conversation"].append({
-                            "speaker": speaker.strip(),
-                            "text": content.strip()
-                        })
+            response_content = response.choices[0].message.content
+            print(f"Respuesta cruda del modelo: {response_content[:500]}...")  # Depuración
+            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]
+            dialogue = json.loads(json_str)
+            if not dialogue.get("conversation"):
+                print("Error: No se generó diálogo válido.")
             return dialogue
         except Exception as e:
             raise RuntimeError(f"Failed to parse dialogue: {str(e)}")
@@ -186,4 +194,4 @@ class URLToAudioConverter:
             f"{turn['speaker']}: {turn['text']}" 
             for turn in conversation["conversation"]
         )
-        return output_path, text_output
+        return output_path, text_output