Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import google.generativeai as genai
 import os
 from dotenv import load_dotenv
+from gradio import Markdown
+import textwrap
 
 # Cargar variables de entorno
 load_dotenv()
@@ -9,6 +11,10 @@ load_dotenv()
 # Configurar la API de Google Gemini
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
+def to_markdown(text):
+    text = text.replace('•', '  *')  # Convertir los puntos en listas con asteriscos
+    return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))
+
 def generate_headlines(number_of_headlines, target_audience, product, temperature):
     # Crear la configuración del modelo
     generation_config = {
@@ -22,7 +28,7 @@ def generate_headlines(number_of_headlines, target_audience, product, temperatur
     model = genai.GenerativeModel(
         model_name="gemini-1.5-pro",
         generation_config=generation_config,
-        system_instruction="Eres un copywriter de clase mundial, con experiencia en la creación de ganchos, titulares y líneas de asunto que capturan la atención de inmediato. Tu habilidad radica en comprender profundamente las emociones, deseos y desafíos de una audiencia específica, lo que te permite diseñar estrategias de marketing personalizadas que resuenan y motivan la acción. Sabes cómo utilizar estructuras probadas para atraer a tu audiencia objetivo, generando interés y logrando una conexión poderosa que impulsa los resultados deseados en campañas publicitarias y de contenido. Responde en español, en tipo lista enumerando los encabezados, ponle un título que tenga que ver con los encabezados generados y el producto, sin utilizar Markdown.",
+        system_instruction="Eres un copywriter de clase mundial, con experiencia en la creación de ganchos, titulares y líneas de asunto que capturan la atención de inmediato. Tu habilidad radica en comprender profundamente las emociones, deseos y desafíos de una audiencia específica, lo que te permite diseñar estrategias de marketing personalizadas que resuenan y motivan la acción. Sabes cómo utilizar estructuras probadas para atraer a tu audiencia objetivo, generando interés y logrando una conexión poderosa que impulsa los resultados deseados en campañas publicitarias y de contenido. Responde en español, en tipo lista enumerando los encabezados, ponle un título que tenga que ver con los encabezados generados y el producto",
     )
 
     chat_session = model.start_chat(
@@ -37,21 +43,18 @@ def generate_headlines(number_of_headlines, target_audience, product, temperatur
     )
 
     response = chat_session.send_message("Genera los titulares")  # Enviar mensaje para obtener la respuesta
-    return response.text
+    return to_markdown(response.text)  # Usar to_markdown para formatear la respuesta
 
 # Configurar la interfaz de usuario con Gradio
-def gradio_generate_headlines(number_of_headlines, target_audience, product, temperature):
-    return generate_headlines(number_of_headlines, target_audience, product, temperature)
-
 iface = gr.Interface(
-    fn=gradio_generate_headlines,
+    fn=generate_headlines,
     inputs=[
         gr.Dropdown(choices=[str(i) for i in range(1, 11)], label="Número de Titulares", value="5"),
         gr.Textbox(label="Público Objetivo", placeholder="Ejemplo: Estudiantes Universitarios"),
         gr.Textbox(label="Producto", placeholder="Ejemplo: Curso de Inglés"),
         gr.Slider(minimum=0, maximum=1, value=0, step=0.1, label="Creatividad")
     ],
-    outputs=gr.Textbox(label="Titulares Generados", lines=10, placeholder="Los titulares aparecerán aquí..."),
+    outputs=gr.Markdown(label="Titulares Generados"),  # Eliminado el placeholder
     title="Generador de Titulares",
     description="Usa el poder de Gemini AI para crear titulares atractivos. Ajusta los parámetros para generar titulares que capturen la atención de tu audiencia."
 )