Update app.py

app.py

@@ -1,7 +1,8 @@
-import streamlit as st
+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
@@ -11,7 +12,7 @@ load_dotenv()
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
 def to_markdown(text):
-    return textwrap.indent(text, '> ', predicate=lambda _: True)
+    return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))
 
 def generate_opening_paragraph(target_audience, product, temperature, text_type, model_name):
     # Crear la configuración del modelo
@@ -71,34 +72,42 @@ def generate_opening_paragraph(target_audience, product, temperature, text_type,
     response = chat_session.send_message("Genera el párrafo de apertura")
     return to_markdown(response.text)
 
-# Configuración de la interfaz de Streamlit
-st.title("Generador de Párrafos de Apertura")
+# Configurar la interfaz de usuario con Gradio
+css = """
+<style>
+    body { font-family: Arial, sans-serif; }
+    .gradio-container { max-width: 800px; margin: auto; }
+</style>
+"""
 
-# Entradas
-target_audience = st.text_input("Público Objetivo", "Ejemplo: Estudiantes Universitarios")
-product = st.text_input("Producto", "Ejemplo: Curso de Inglés")
-temperature = st.slider("Creatividad", 0.0, 1.0, 0.5, 0.1)
-text_type = st.selectbox("Tipo de Texto", ["Página de Ventas", "Correo", "Historia"])
-model_selector = st.selectbox("Selecciona el modelo", ["gemini-1.5-flash", "gemini-1.5-pro"])
+with gr.Blocks(css=css) as iface:
+    gr.HTML("<h1 class='text-3xl font-bold text-center mb-6'>Generador de Párrafos de Apertura</h1>")
+    
+    with gr.Row():
+        with gr.Column():
+            target_audience = gr.Textbox(label="Público Objetivo", placeholder="Ejemplo: Estudiantes Universitarios")
+            product = gr.Textbox(label="Producto", placeholder="Ejemplo: Curso de Inglés")
+            temperature = gr.Slider(minimum=0, maximum=1, value=0.5, step=0.1, label="Creatividad")
+            text_type = gr.Dropdown(choices=["Página de Ventas", "Correo", "Historia"], label="Tipo de Texto", value="Página de Ventas")
+            model_selector = gr.Dropdown(choices=["gemini-1.5-flash", "gemini-1.5-pro"], value="gemini-1.5-flash", label="Selecciona el modelo")
+        
+            # Colocar los botones en una misma fila
+            with gr.Row():
+                submit_btn = gr.Button("Generar Párrafo de Apertura", variant="primary")
+                clear_btn = gr.Button("Empezar de nuevo", variant="secondary")
+        
+        with gr.Column():
+            output_text = gr.Markdown(label="Párrafo de Apertura Generado")
+    
+    # Funcionalidad del botón Generar
+    submit_btn.click(fn=generate_opening_paragraph, inputs=[target_audience, product, temperature, text_type, model_selector], outputs=output_text)
 
-# Botones
-col1, col2 = st.columns(2)
-
-with col1:
-    if st.button("Generar Párrafo de Apertura"):
-        output_text = generate_opening_paragraph(target_audience, product, temperature, text_type, model_selector)
-        st.markdown(output_text)
-
-with col2:
-    if st.button("Empezar de nuevo"):
-        # Limpiar campos de entrada
-        target_audience = ""
-        product = ""
-        temperature = 0.5
-        text_type = "Página de Ventas"
-        model_selector = "gemini-1.5-flash"
+    # Funcionalidad del botón Limpiar: restablecer valores a los predeterminados
+    clear_btn.click(
+        fn=lambda: ("", "", 0.5, "Página de Ventas", "gemini-1.5-flash"),
+        inputs=[],
+        outputs=[target_audience, product, temperature, text_type, model_selector]
+    )
 
-# Mostrar resultado (si se generó uno)
-if 'output_text' in locals():
-    st.markdown("**Párrafo de Apertura Generado:**")
-    st.markdown(output_text)
+# Lanzar la interfaz
+iface.launch()