Update app.py

app.py

@@ -1,8 +1,7 @@
-import gradio as gr
+import streamlit as st
 import google.generativeai as genai
 import os
 from dotenv import load_dotenv
-from gradio import Markdown
 import textwrap
 
 # Cargar variables de entorno
@@ -12,7 +11,7 @@ load_dotenv()
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
 def to_markdown(text):
-    return Markdown(textwrap.indent(text, '> ', predicate=lambda _: True))
+    return textwrap.indent(text, '> ', predicate=lambda _: True)
 
 def generate_opening_paragraph(target_audience, product, temperature, text_type, model_name):
     # Crear la configuración del modelo
@@ -72,42 +71,34 @@ 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)
 
-# Configurar la interfaz de usuario con Gradio
-css = """
-<style>
-    body { font-family: Arial, sans-serif; }
-    .gradio-container { max-width: 800px; margin: auto; }
-</style>
-"""
+# Configuración de la interfaz de Streamlit
+st.title("Generador de Párrafos de Apertura")
 
-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)
+# 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"])
 
-    # 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]
-    )
+# 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"
 
-# Lanzar la interfaz
-iface.launch()
+# Mostrar resultado (si se generó uno)
+if 'output_text' in locals():
+    st.markdown("**Párrafo de Apertura Generado:**")
+    st.markdown(output_text)