Update app.py

app.py

@@ -1,3 +1,16 @@
+from dotenv import load_dotenv
+import streamlit as st
+import os
+import google.generativeai as genai
+import langchain
+
+# Cargar las variables de entorno
+load_dotenv()
+
+# Configurar la API de Google
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+
+# Generar el resultado utilizando el modelo con la instrucción de bullets específica
 def generate_bullets(number_of_bullets, target_audience, product, call_to_action, temperature):
     # Configuración del modelo
     generation_config = {
@@ -41,3 +54,64 @@ def generate_bullets(number_of_bullets, target_audience, product, call_to_action
             raise ValueError("No se generaron bullets válidos.")
     except Exception as e:
         raise ValueError(f"Error al generar los bullets: {str(e)}")
+
+# Configurar la interfaz de usuario con Streamlit
+st.set_page_config(page_title="Impact Bullet Generator", layout="wide")
+
+# Centrar el título y el subtítulo
+st.markdown("<h1 style='text-align: center;'>Impact Bullet Generator</h1>", unsafe_allow_html=True)
+st.markdown("<h4 style='text-align: center;'>Transforma los pensamientos de tu audiencia en balas persuasivas que inspiren a la acción.</h4>", unsafe_allow_html=True)
+
+# Añadir CSS personalizado para el botón
+st.markdown("""
+    <style>
+    div.stButton > button {
+        background-color: #FFCC00;
+        color: black;
+        width: 90%;
+        height: 60px;
+        font-weight: bold;
+        font-size: 22px;
+        text-transform: uppercase;
+        border: 1px solid #000000;
+        border-radius: 8px;
+        display: block;
+        margin: 0 auto;
+    }
+    div.stButton > button:hover {
+        background-color: #FFD700;
+        color: black;
+    }
+    </style>
+""", unsafe_allow_html=True)
+
+# Crear columnas
+col1, col2 = st.columns([1, 2])  
+
+# Columnas de entrada
+with col1:
+    target_audience = st.text_input("¿Quién es tu público objetivo?", placeholder="Ejemplo: Estudiantes Universitarios")
+    product = st.text_input("¿Qué producto tienes en mente?", placeholder="Ejemplo: Curso de Inglés")
+    call_to_action = st.text_input("¿Qué acción deseas que tomen?", placeholder="Ejemplo: Inscribirse al curso")
+    number_of_bullets = st.selectbox("Número de bullets", options=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10], index=4)
+    temperature = st.slider("Creatividad", min_value=0.0, max_value=1.0, value=0.5, step=0.1)
+
+    # Botón de enviar
+    submit = st.button("Generar Beneficios")
+
+# Mostrar los beneficios generados
+if submit:
+    if target_audience and product and call_to_action:
+        try:
+            # Obtener la respuesta del modelo
+            generated_bullets = generate_bullets(number_of_bullets, target_audience, product, call_to_action, temperature)
+            col2.markdown(f"""
+                <div style="border: 1px solid #000000; padding: 5px; border-radius: 8px; background-color: #ffffff;">
+                    <h4>Mira los bullets generados:</h4>
+                    <p style="font-size: 22px;">{generated_bullets}</p>
+                </div>
+            """, unsafe_allow_html=True)
+        except Exception as e:
+            st.error(f"Error al generar los bullets: {str(e)}")
+    else:
+        st.error("Por favor, completa todos los campos.")