Update app.py

app.py

@@ -23,12 +23,26 @@ def generate_audio(text, output_path):
     tts = gTTS(text, lang='es')
     tts.save(output_path)
 
+def generate_audio_coqui(text, output_path, api_key):
+    url = "https://api.coqui.ai/tts"
+    payload = {
+        "input_text": text,
+        "output_path": output_path,
+        "use_model": "coqui/tts/coqui-tts-ljspeech",
+        "api_key": api_key
+    }
+    response = requests.post(url, data=payload)
+
+    if response.status_code == 200:
+        print("Audio generado exitosamente.")
+    else:
+        print(f"Error al generar el audio: {response.text}")
+
 def generate_video(audio_path, face_image_path, video_path):
     command = f"python3 inference.py --checkpoint_path checkpoints/wav2lip_gan.pth --face {face_image_path} --audio {audio_path} --outfile {video_path} --nosmooth"
     process = subprocess.run(command, shell=True, text=True)
     return process.returncode
 
-# Descargar el archivo wav2lip_gan.pth si no existe
 checkpoint_path = "checkpoints/wav2lip_gan.pth"
 if not os.path.exists(checkpoint_path):
     url = "https://huggingface.co/spaces/salomonsky/tarot/resolve/main/checkpoints/wav2lip_gan.pth"
@@ -42,7 +56,6 @@ if not os.path.exists(checkpoint_path):
     else:
         print("Error al descargar el archivo.")
 
-# Crear la carpeta 'temp' si no existe
 os.makedirs('temp', exist_ok=True)
 
 def generate_output(name):
@@ -78,21 +91,17 @@ if st.button("Tirar Cartas de Tarot"):
         st.success("Cartas tiradas exitosamente.")
         st.write(f"Cartas seleccionadas: {', '.join(cartas)}")
         st.write(f"Interpretación de GPT-3: {gpt_output}")
-        st.write("Generando video...")
+        st.write("Generando audio...")
 
-        # Directorio donde se guardará el video
+        audio_path = "temp/temp.wav"
         video_dir = "videos/"
         os.makedirs(video_dir, exist_ok=True)
-
-        # Rutas de los archivos
-        audio_path = "temp/temp.wav"  # Cambiado el nombre del archivo de audio
         video_path = f"{video_dir}video.mp4"
         face_image_path = "face.jpg"
-        
-        # Aquí es donde generas el audio, por ejemplo:
-        generate_audio(gpt_output, audio_path)
+        coqui_api_key = "yx9GhszGxbJVZ11kO6BrxAPS5OvZLhQdM01F5BwzLMFjhgApXbeJfFauzBrJIoB6"
+
+        generate_audio_coqui(gpt_output, audio_path, coqui_api_key)
 
-        # Luego, genera el video usando el código que proporcionaste
         return_code = generate_video(audio_path, face_image_path, video_path)
         if return_code != 0:
             st.error(f"No se pudo generar el video.")