Update app.py

app.py

@@ -14,7 +14,7 @@ genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 def to_markdown(text):
     return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))
 
-def generate_opening_paragraph(target_audience, product, temperature, text_type):
+def generate_opening_paragraph(target_audience, product, temperature, text_type, model_name):
     # Crear la configuración del modelo
     generation_config = {
         "temperature": temperature,
@@ -25,7 +25,7 @@ def generate_opening_paragraph(target_audience, product, temperature, text_type)
     }
 
     model = genai.GenerativeModel(
-        model_name="gemini-1.5-flash",
+        model_name=model_name,
         generation_config=generation_config,
         system_instruction=f"Eres un especialista en copywriting altamente capacitado, enfocado en la creación de textos persuasivos que logran captar la atención del lector, mantener su interés y llevarlo a la acción. Tienes una profunda comprensión de la psicología del consumidor, y utilizas la persuasión de manera efectiva en cada palabra que escribes, conectando con las emociones y necesidades de tu audiencia. Dominas técnicas avanzadas de redacción, como el uso de encabezados poderosos, aperturas intrigantes, llamados a la acción efectivos y storytelling cautivador. Tu habilidad para ajustar el tono y estilo al contexto asegura que cada mensaje resuene profundamente. Tienes una amplia experiencia demostrada en generar resultados de ventas tanto para negocios en línea como offline. Las respuestas deben presentarse en formato de texto, en párrafos, sin mencionar directamente el producto o servicio, ni referirse al público objetivo de manera explícita. Al responder escribe un encabezado que diga: 'Este es tu parrafo de apertura para cautivar a {target_audience}'"
     )
@@ -79,7 +79,8 @@ iface = gr.Interface(
         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"),
-        gr.Dropdown(choices=["Página de Ventas", "Correo", "Historia"], label="Tipo de Texto", value="Página de Ventas")
+        gr.Dropdown(choices=["Página de Ventas", "Correo", "Historia"], label="Tipo de Texto", value="Página de Ventas"),
+        gr.Dropdown(choices=["gemini-1.5-pro", "gemini-1.5-flash"], label="Selecciona el modelo", value="gemini-1.5-pro")
     ],
     outputs=gr.Markdown(label="Párrafo de Apertura Generado"),
     title="Generador de Párrafos de Apertura",