Update app.py

app.py

@@ -7,9 +7,9 @@ from pydub import AudioSegment
 
 openai.api_key = os.environ.get("openai_api_key")
 
-def generate_output(name, horoscope_type):
+def generate_output(name, horoscope_type, birth_date):
     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:"
+    prompt = f"Tu horóscopo {horoscope_type} y oráculo de hoy para {name} nacido el {birth_date} es:"
     response = openai.Completion.create(
         engine="text-davinci-003",
         prompt=prompt,
@@ -19,12 +19,12 @@ def generate_output(name, horoscope_type):
         stop=None,
     )
     gpt3_output = response.choices[0].text.strip()
-    generated_text = gpt3_output.replace(prompt, "").strip()
+    generated_text = gpt3_output.strip()
     if len(response.choices) == 0 or 'text' not in response.choices[0]:
         return None, "No se pudo generar el texto."
 
     try:
-        tts = gTTS(generated_text, lang='es')
+        tts = gTTS(f"Tu horóscopo {horoscope_type} para {name} nacido el {birth_date} es: {generated_text}", lang='es')
         temp_audio_path = "temp_audio.mp3"
         tts.save(temp_audio_path)
         audio_path = "audio.wav"
@@ -49,10 +49,11 @@ def generate_output(name, horoscope_type):
 
 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")
+birth_date_input = gr.inputs.Date(label="Fecha de Nacimiento")
 output = gr.outputs.Video(label=None)
 
-def generate_and_display_output(name, horoscope_type):
-    video_path, error_message = generate_output(name, horoscope_type)
+def generate_and_display_output(name, horoscope_type, birth_date):
+    video_path, error_message = generate_output(name, horoscope_type, birth_date)
     if error_message:
         print(f"Error: {error_message}")
     else:
@@ -60,10 +61,10 @@ def generate_and_display_output(name, horoscope_type):
 
 iface = gr.Interface(
     fn=generate_and_display_output,
-    inputs=[name_input, horoscope_type_radio],
+    inputs=[name_input, horoscope_type_radio, birth_date_input],
     outputs=output,
     title="Horóscopo con Inteligencia Artificial v2.1",
-    description="Ingresa tu nombre y selecciona el tipo de horóscopo.",
+    description="Ingresa tu nombre, tipo de horóscopo y fecha de nacimiento.",
     layout="vertical",
     theme="dark"
 )