Upload app.py

app.py

@@ -1,102 +1,197 @@
+from dotenv import load_dotenv
 import streamlit as st
-import google.generativeai as genai
 import os
-from dotenv import load_dotenv
-from styles import get_custom_css
-
-# Set page to wide mode to use full width
-st.set_page_config(layout="wide")
+import google.generativeai as genai
+from style import styles
+from prompts import create_instruction
 
-# Load environment variables
+# Cargar las variables de entorno
 load_dotenv()
 
-# Configure Google Gemini API
-genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
-model = genai.GenerativeModel('gemini-2.0-flash')
+# Configurar la API de Google
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 
-# Initialize session state variables if they don't exist
-if 'submitted' not in st.session_state:
-    st.session_state.submitted = False
-if 'offer_result' not in st.session_state:
-    st.session_state.offer_result = ""
+# Inicializar variables de estado en session_state si no existen
+if 'perfil_cliente' not in st.session_state:
+    st.session_state.perfil_cliente = None
+if 'producto' not in st.session_state:
+    st.session_state.producto = ""
+if 'habilidades' not in st.session_state:
+    st.session_state.habilidades = ""
+if 'creatividad' not in st.session_state:
+    st.session_state.creatividad = 1.0
 
-# Hide Streamlit menu and footer
-st.markdown("""
-<style>
-#MainMenu {visibility: hidden;}
-footer {visibility: hidden;}
-header {visibility: hidden;}
-</style>
-""", unsafe_allow_html=True)
+# Función para generar el perfil de cliente ideal
+@st.cache_resource
+def get_model(temperature):
+    generation_config = {
+        "temperature": temperature,
+    }
+    return genai.GenerativeModel('gemini-2.0-flash', generation_config=generation_config)
 
-# Custom CSS
-st.markdown(get_custom_css(), unsafe_allow_html=True)
+def generate_buyer_persona(product, skills, target_audience, temperature):
+    if not product or not skills:
+        return "Por favor, completa los campos de producto y habilidades."
+    
+    model = get_model(temperature)
+    instruction = create_instruction(
+        product_service=product,
+        skills=skills,
+        target_audience=target_audience
+    )
+    
+    # Añadir instrucción explícita para respuesta en español
+    instruction += "\n\nIMPORTANTE: La respuesta debe estar completamente en español."
+    
+    response = model.generate_content([instruction], generation_config={"temperature": temperature})
+    return response.parts[0].text if response and response.parts else "Error generando el perfil de cliente ideal."
 
-# App title and description
-st.title('🚀 Great Offer Generator')
-st.markdown('''### Transform your skills into compelling offers!''')
+# Modificar la función update_profile para que no use spinner
+def update_profile():
+    # Solo actualizar la variable de sesión
+    st.session_state.submitted = True
 
-# Create two columns for layout - left column 40%, right column 60%
-col1, col2 = st.columns([4, 6])
+# Configurar la interfaz de usuario con Streamlit
+st.set_page_config(page_title="Generador de Cliente Ideal", page_icon="👤", layout="wide")
 
-# Main input section in left column
-with col1:
-    skills = st.text_area('💪 Your Skills', height=70, 
-                        help='List your key skills and expertise')
-    product_service = st.text_area('🎯 Product/Service', height=70,
-                                help='Describe your product or service')
+# Leer el contenido del archivo manual.md si existe
+try:
+    with open("manual.md", "r", encoding="utf-8") as file:
+        manual_content = file.read()
+    # Mostrar el contenido del manual en el sidebar
+    st.sidebar.markdown(manual_content)
+except FileNotFoundError:
+    st.sidebar.warning("Manual not found. Please create a manual.md file.")
+except Exception as e:
+    st.sidebar.error(f"Error loading manual: {str(e)}")
 
-    # Accordion for additional settings
-    with st.expander('⚙️ Advanced Settings'):
-        target_audience = st.text_area('👥 Target Audience', height=70,
-                                    help='Describe your ideal customer or client')
-        temperature = st.slider('🌡️ Creativity Level', min_value=0.0, max_value=2.0, value=0.7,
-                            help='Higher values make the output more creative but less focused')
+# Ocultar elementos de la interfaz
+st.markdown(styles["main_layout"], unsafe_allow_html=True)
 
-    # Generate button with callback
-    def generate_offer():
-        if not skills or not product_service:
-            st.error('Please fill in both Skills and Product/Service fields')
-        else:
-            # Set submitted flag to True
-            st.session_state.submitted = True
-            # Store input values in session state
-            st.session_state.skills = skills
-            st.session_state.product_service = product_service
-            st.session_state.target_audience = target_audience
-            st.session_state.temperature = temperature
-            
-    st.button('Generate Offer 🎉', on_click=generate_offer)
+# Centrar el título y el subtítulo
+st.markdown("<h1 style='text-align: center;'>Generador de Perfil de Cliente Ideal</h1>", unsafe_allow_html=True)
+st.markdown("<h4 style='text-align: center;'>Crea un perfil detallado de tu cliente ideal basado en tu producto y habilidades.</h4>", unsafe_allow_html=True)
+
+# Añadir CSS personalizado para el botón
+st.markdown(styles["button"], unsafe_allow_html=True)
+# Añadir CSS personalizado para el botón de descarga
+st.markdown(styles["download_button"], unsafe_allow_html=True)
 
-# Results column
+# Crear columnas
+col1, col2 = st.columns([1, 2])  
+
+# Columna de entrada
+with col1:
+    producto = st.text_area("¿Qué producto o servicio ofreces?", 
+                          value=st.session_state.producto,
+                          placeholder="Ejemplo: Curso de Inglés",
+                          key="producto_input",
+                          height=70)
+    st.session_state.producto = producto
+    
+    habilidades = st.text_area("¿Cuáles son tus habilidades principales?", 
+                             value=st.session_state.habilidades,
+                             placeholder="Ejemplo: Enseñanza, comunicación, diseño de contenidos",
+                             key="habilidades_input",
+                             height=70)
+    st.session_state.habilidades = habilidades
+    
+    # Crear un acordeón para las opciones de personalización
+    with st.expander("Personaliza Tu Cliente Ideal Soñado"):
+        # Nuevo campo para público objetivo
+        if 'publico_objetivo' not in st.session_state:
+            st.session_state.publico_objetivo = ""
+        
+        publico_objetivo = st.text_area("¿Cuál es tu público objetivo? (opcional)", 
+                                      value=st.session_state.publico_objetivo,
+                                      placeholder="Ejemplo: Profesionales entre 25-40 años interesados en desarrollo personal",
+                                      key="publico_objetivo_input",
+                                      height=70)
+        st.session_state.publico_objetivo = publico_objetivo
+        
+        # Nivel de creatividad con slider
+        creatividad = st.slider("Nivel de creatividad", 
+                               min_value=0.0, 
+                               max_value=2.0, 
+                               value=st.session_state.creatividad, 
+                               step=0.1,
+                               key="creatividad_slider")
+        st.session_state.creatividad = creatividad
+    
+    # Botón para generar
+    submit = st.button("GENERAR PERFIL DE CLIENTE IDEAL", on_click=update_profile)
+
+# Columna de resultados
 with col2:
-    # Check if form has been submitted
-    if st.session_state.submitted:
-        with st.spinner('Creating your perfect offer...'):
-            prompt = f"""Based on the following information, create a compelling offer:
-            Skills: {st.session_state.skills}
-            Product/Service: {st.session_state.product_service}
-            Target Audience: {st.session_state.target_audience if st.session_state.target_audience else 'General audience'}
-            
-            Please create a professional and engaging offer that highlights the value proposition
-            and appeals to the target audience. Include a clear call to action."""
-            
-            try:
-                # Create generation config with temperature
-                generation_config = genai.GenerationConfig(temperature=st.session_state.temperature)
+    # Verificar si se ha enviado el formulario
+    if 'submitted' in st.session_state and st.session_state.submitted:
+        if st.session_state.producto and st.session_state.habilidades:
+            with st.spinner("Creando tu Cliente Ideal Soñado..."):
+                # Generar el perfil del cliente
+                perfil_cliente = generate_buyer_persona(
+                    st.session_state.producto,
+                    st.session_state.habilidades,
+                    st.session_state.publico_objetivo,
+                    st.session_state.creatividad
+                )
+                # Guardar en session_state
+                st.session_state.perfil_cliente = perfil_cliente
+                # Resetear el estado de envío
+                st.session_state.submitted = False
                 
-                # Pass the generation config to generate_content
-                response = model.generate_content(prompt, generation_config=generation_config)
-                st.session_state.offer_result = response.text
-                st.success('✨ Your offer is ready!')
+            # Mostrar resultados
+            if not isinstance(st.session_state.perfil_cliente, str):
+                st.error("Error al generar el perfil de cliente ideal")
+            else:
+                # Crear un contenedor con estilo personalizado
+                st.markdown(f"""
+                <style>
+                .results-box {{
+                    padding: 15px;
+                    border: 1px solid #ddd;
+                    border-radius: 8px;
+                    margin-bottom: 20px;
+                }}
+                </style>
+                """, unsafe_allow_html=True)
                 
-                # Display result
-                st.markdown('### 📝 Generated Offer')
-                st.markdown(st.session_state.offer_result)
-            except Exception as e:
-                st.error(f'An error occurred: {str(e)}')
-                st.session_state.submitted = False
-
-# Footer
-st.markdown('---')
-st.markdown('Made with ❤️ by Jesús Cabrera')
+                # Usar un expander sin título para contener todo el resultado
+                with st.expander("", expanded=True):
+                    st.markdown("<h3>Tu Cliente Ideal</h3>", unsafe_allow_html=True)
+                    st.markdown(st.session_state.perfil_cliente)
+                
+                # Opción para descargar
+                st.download_button(
+                    label="Descargar Perfil",
+                    data=st.session_state.perfil_cliente,
+                    file_name="cliente_ideal.txt",
+                    mime="text/plain"
+                )
+        else:
+            st.warning("Por favor, completa los campos de producto y habilidades antes de generar el perfil.")
+    # Mostrar resultados anteriores si existen
+    elif st.session_state.perfil_cliente:
+        # Crear un contenedor con estilo personalizado
+        st.markdown(f"""
+        <style>
+        .results-box {{
+            padding: 15px;
+            border: 1px solid #ddd;
+            border-radius: 8px;
+            margin-bottom: 20px;
+        }}
+        </style>
+        """, unsafe_allow_html=True)
+        
+        # Usar un expander sin título para contener todo el resultado
+        with st.expander("", expanded=True):
+            st.markdown("<h3>Tu Cliente Ideal</h3>", unsafe_allow_html=True)
+            st.markdown(st.session_state.perfil_cliente)
+        
+        # Opción para descargar
+        st.download_button(
+            label="Descargar Perfil",
+            data=st.session_state.perfil_cliente,
+            file_name="cliente_ideal.md",
+            mime="text/markdown"
+        )