Update app.py

app.py

@@ -8,6 +8,7 @@ from pydub import AudioSegment
 openai.api_key = os.environ.get("openai_api_key")
 
 def generate_output(name, horoscope_type):
+    image_filename = f"{horoscope_type.lower()}.jpg"  # Construye el nombre de la imagen
     prompt = f"{name}, tu horóscopo {horoscope_type} y oráculo de hoy es:"
     response = openai.Completion.create(
         engine="text-davinci-003",
@@ -33,7 +34,7 @@ def generate_output(name, horoscope_type):
     except Exception as e:
         return None, f"No se pudo generar el audio: {str(e)}"
 
-    command = f"python3 inference.py --checkpoint_path checkpoints/wav2lip_gan.pth --face oraculo.jpg --audio audio.wav --outfile video.mp4 --nosmooth"
+    command = f"python3 inference.py --checkpoint_path checkpoints/wav2lip_gan.pth --face {image_filename} --audio audio.wav --outfile video.mp4 --nosmooth"
     process = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     if process.returncode != 0:
         error_message = process.stderr.decode("utf-8")
@@ -47,7 +48,7 @@ def generate_output(name, horoscope_type):
     return None, "No se pudo generar el video"
 
 name_input = gr.inputs.Textbox(lines=1, placeholder="Escribe tu Nombre Completo")
-horoscope_type_radio = gr.inputs.Radio(["Egipcio", "Griego", "Maya", "Chino", "Celta", "Hindú"], label="Tipo de Horóscopo")
+horoscope_type_radio = gr.inputs.Radio(["Griego", "Egipcio", "Maya", "Chino", "Celta", "Hindú"], label="Tipo de Horóscopo")
 output = gr.outputs.Video(label=None)
 
 def generate_and_display_output(name, horoscope_type):