Update app.py

app.py

@@ -4,13 +4,11 @@ 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:"
+def generate_output(name, horoscope_type):
+    prompt = f"{name}, tu horóscopo {horoscope_type} y oráculo astrológico de hoy es:"
     response = openai.Completion.create(
         engine="text-davinci-003",
         prompt=prompt,
@@ -24,10 +22,6 @@ def generate_output(name, date_of_birth, progress=gr.Progress(), image_widget=No
     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"
@@ -36,7 +30,6 @@ def generate_output(name, date_of_birth, progress=gr.Progress(), image_widget=No
         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)}"
 
@@ -50,48 +43,41 @@ def generate_output(name, date_of_birth, progress=gr.Progress(), image_widget=No
     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")
-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, progress_widget, image_widget)
-        if error_message:
-            print(f"Error: {error_message}")
-        else:
-            display(video_widget.from_url(video_path))
+horoscope_type_options = [
+    {"label": "Egipcio", "value": "egipcio"},
+    {"label": "Griego", "value": "griego"},
+    {"label": "Maya", "value": "maya"},
+    {"label": "Chino", "value": "chino"}
+]
+
+horoscope_type_radio = gr.inputs.Radio(horoscope_type_options, label="Tipo de Horóscopo")
 
-submit_button = widgets.Button(description="Enviar")
-submit_button.on_click(generate_and_display_output)
+output = gr.outputs.Video(label="Respuesta de Andrea (dos minutos aproximadamente)").style(width=590)
 
-display(name_input)
-display(dob_input)
-display(progress_widget)
-display(image_widget)
-display(video_widget)
-display(submit_button)
-display(output_widget)
+def generate_and_display_output(name, horoscope_type):
+    video_path, error_message = generate_output(name, horoscope_type)
+    if error_message:
+        print(f"Error: {error_message}")
+    else:
+        return video_path
 
 iface = gr.Interface(
-    fn=generate_output,
+    fn=generate_and_display_output,
     inputs=[
         name_input,
-        dob_input
+        horoscope_type_radio
     ],
-    outputs=video_widget,
+    outputs=output,
     title="Horóscopo con Inteligencia Artificial v2.1",
-    description="Por favor, ingresa tu nombre completo.",
-    layout="vertical-reverse",
+    description="Por favor, ingresa tu nombre y selecciona el tipo de horóscopo.",
+    layout="vertical",
     theme="dark",
     live=False
 )
 
-iface.launch()
+iface.launch()