oraculo / app.py
salomonsky's picture
Update app.py
c91d9d5
raw
history blame
2.37 kB
import os
import openai
import gradio as gr
import subprocess
from gtts import gTTS
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",
prompt=prompt,
max_tokens=180,
temperature=0.6,
n=1,
stop=None,
)
gpt3_output = response.choices[0].text.strip()
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."
try:
tts = gTTS(personalized_response, lang='es')
audio_path = "audio.mp3"
tts.save(audio_path)
except Exception as e:
return None, f"No se pudo generar el audio: {str(e)}"
video_path = "video.mp4"
command = f"python3 inference.py --checkpoint_path checkpoints/wav2lip_gan.pth --face face.jpg --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
return None, f"No se pudo generar el video: {error_message}"
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_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")
def generate_and_display_output(name, birth_date):
video_path, error_message = generate_output(name, birth_date)
if error_message:
print(f"Error: {error_message}")
return None, error_message
else:
return video_path, None
outputs = [output, error_output]
iface = gr.Interface(
fn=generate_and_display_output,
inputs=[name_input, birth_date_input],
outputs=outputs,
layout="vertical",
theme="dark"
)
iface.launch()