Update app1.py

app1.py

@@ -4,9 +4,12 @@ import gradio as gr
 import subprocess
 from gtts import gTTS
 
-openai.api_key = os.environ.get("openai_api_key")
+openai.api_key = os.environ.get("OPENAI_API_KEY")
 
 def generate_output(name, birth_date):
+    if not birth_date:
+        return None, "El campo de fecha de nacimiento es obligatorio."
+
     prompt = f"Tú horóscopo de hoy, si naciste el {birth_date}, es:"
     response = openai.Completion.create(
         engine="text-davinci-003",
@@ -17,7 +20,7 @@ def generate_output(name, birth_date):
         stop=None,
     )
     gpt3_output = response.choices[0].text.strip()
-    personalized_response = f"Tu horóscopo {name} es: {gpt3_output}"
+    personalized_response = f"Tu horóscopo {name} nacido el {birth_date} es: {gpt3_output}"
     
     if len(response.choices) == 0 or 'text' not in response.choices[0]:
         return None, "No se pudo generar el texto."
@@ -29,18 +32,19 @@ def generate_output(name, birth_date):
     except Exception as e:
         return None, f"No se pudo generar el audio: {str(e)}"
 
-    command = f"python3 inference.py --checkpoint_path wav2lip_gan.pth --face face2.png --audio audio.mp3 --outfile video.mp4 --nosmooth"
-    process = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+    video_path = "video.mp4"
+    command = f"python3 inference.py --checkpoint_path wav2lip_gan.pth --face face2.png --audio {audio_path} --outfile {video_path} --nosmooth"
+    process = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
     if process.returncode != 0:
-        error_message = process.stderr.decode("utf-8")
+        error_message = process.stderr
         return None, f"No se pudo generar el video: {error_message}"
 
-    if os.path.isfile("video.mp4"):
-        return "video.mp4", None
+    if os.path.isfile(video_path):
+        return video_path, None
     return None, "No se pudo generar el video"
 
 name_input = gr.inputs.Textbox(lines=1, placeholder="Escribe tu Nombre Completo", label="Nombre")
-birth_date = gr.inputs.Textbox(lines=1, placeholder="Fecha Nacimiento - DD/MM/AAAA", label="Cumpleaños")
+birth_date_input = gr.inputs.Textbox(lines=1, placeholder="Fecha Nacimiento - DD/MM/AAAA", label="Cumpleaños")
 output = gr.outputs.Video(label="Resultado", type="mp4").style(width=350)
 error_output = gr.outputs.Textbox(label="Errores")
 
@@ -56,10 +60,10 @@ outputs = [output, error_output]
 
 iface = gr.Interface(
     fn=generate_and_display_output,
-    inputs=[name_input, birth_date],
+    inputs=[name_input, birth_date_input],
     outputs=outputs,
     layout="vertical",
     theme="darkdefault"
 )
 
-iface.launch(share=True)
+iface.launch(share=True)