Spaces:
Running
Running
File size: 3,059 Bytes
d64793e 972a12b 05d3cbf 7b6fb4e 05d3cbf f6f48d0 7b6fb4e 0c5d6b6 2619ab3 f6f48d0 a0eb8b5 972a12b a0eb8b5 972a12b 4b73d5c 05d3cbf bb6e368 b3d0e64 161fa06 bb6e368 b3d0e64 a5c316e 4caa659 55866b4 05d3cbf b43a3bd b3d0e64 05d3cbf 4caa659 05d3cbf bb6e368 4caa659 b43a3bd 05d3cbf f6f48d0 ae6fcec f6f48d0 b9469b5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
import os
import openai
import gradio as gr
import subprocess
from gtts import gTTS
from pydub import AudioSegment
import ipywidgets as widgets
from IPython.display import display
openai.api_key = os.environ.get("openai_api_key")
def generate_output(name, date_of_birth, progress=gr.Progress(), image_widget=None):
prompt = f"{name}, tu horóscopo y oráculo astrológico de hoy es:"
response = openai.Completion.create(
engine="text-davinci-003",
prompt=prompt,
max_tokens=120,
temperature=0.6,
n=1,
stop=None,
)
gpt3_output = response.choices[0].text.strip()
generated_text = gpt3_output.replace(prompt, "").strip()
if len(response.choices) == 0 or 'text' not in response.choices[0]:
return None, "No se pudo generar el texto."
def _progress(generated, to_generate):
progress((generated, to_generate))
_progress(0, 1)
try:
tts = gTTS(generated_text, lang='es')
temp_audio_path = "temp_audio.mp3"
tts.save(temp_audio_path)
audio_path = "audio.wav"
audio = AudioSegment.from_mp3(temp_audio_path)
audio.export(audio_path, format="wav")
print("Archivo de audio generado:", audio_path)
_progress(1, 2)
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"
process = subprocess.run(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
if process.returncode != 0:
error_message = process.stderr.decode("utf-8")
return None, f"No se pudo generar el video: {error_message}"
output_video_path = "video.mp4"
os.remove(temp_audio_path)
if os.path.isfile(output_video_path):
_progress(2, 2)
return output_video_path, None
return None, "No se pudo generar el video"
name_input = gr.inputs.Textbox(lines=1, label="Nombre", placeholder="Ingresa tu nombre")
dob_input = gr.inputs.Textbox(lines=1, label="Fecha y Hora de Nacimiento (opcional)", placeholder="DD/MM/AAAA y HH:MM")
progress_widget = gr.Progress()
image_widget = widgets.Image(value=open("oraculo.jpg", "rb").read())
video_widget = gr.outputs.Video(label="Respuesta de Andrea (dos minutos aproximadamente)").style(width=590)
output_widget = widgets.Output()
def generate_and_display_output(sender):
with output_widget:
output_widget.clear_output()
video_path, error_message = generate_output(name_input.value, dob_input.value, progress_widget, image_widget)
if error_message:
print(f"Error: {error_message}")
else:
display(video_widget.from_url(video_path))
submit_button = widgets.Button(description="Enviar")
submit_button.on_click(generate_and_display_output)
display(name_input)
display(dob_input)
display(progress_widget)
display(image_widget)
display(video_widget)
display(submit_button)
display(output_widget) |