Update app.py

app.py

@@ -1,24 +1,14 @@
-import os
+ import os
+import openai
 import gradio as gr
 import subprocess
 from gtts import gTTS
 from pydub import AudioSegment
 
-horoscope_images = {
-    "Griego": "griego.jpg",
-    "Egipcio": "egipcio.jpg",
-    "Maya": "maya.jpg",
-    "Chino": "chino.jpg",
-    "Celta": "celta.jpg",
-    "Hindu": "hindu.jpg"
-}
-
-name_input = gr.inputs.Textbox(lines=1, placeholder="Escribe tu Nombre Completo")
-horoscope_type_radio = gr.inputs.Radio(["Griego", "Egipcio", "Maya", "Chino", "Celta", "Hindu"], label="Tipo de Horóscopo")
-output = gr.outputs.Video(label=None)
+openai.api_key = os.environ.get("openai_api_key")
 
 def generate_output(name, horoscope_type):
-    image_filename = horoscope_images[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",
@@ -57,6 +47,10 @@ def generate_output(name, horoscope_type):
         return output_video_path, None
     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(["Griego", "Egipcio", "Maya", "Chino", "Celta", "Hindu"], label="Tipo de Horóscopo")
+output = gr.outputs.Video(label=None)
+
 def generate_and_display_output(name, horoscope_type):
     video_path, error_message = generate_output(name, horoscope_type)
     if error_message:
@@ -71,8 +65,7 @@ iface = gr.Interface(
     title="Horóscopo con Inteligencia Artificial v2.1",
     description="Ingresa tu nombre y selecciona el tipo de horóscopo.",
     layout="vertical",
-    theme="dark",
-    examples=[],
+    theme="dark"
 )
 
 iface.launch()