Update app.py

app.py

@@ -5,6 +5,7 @@ import time
 from dotenv import load_dotenv
 from styles import get_custom_css, get_response_html_wrapper
 from formulas import offer_formulas
+from sophistication_levels import sophistication_levels  # Import the sophistication levels
 import PyPDF2
 import docx
 from PIL import Image
@@ -99,6 +100,7 @@ with col1:
         st.session_state.target_audience = target_audience
         st.session_state.temperature = temperature
         st.session_state.formula_type = formula_type
+        st.session_state.sophistication_level = sophistication_level  # Store the sophistication level
 
     # Keep only the manual input tab
     with st.container():
@@ -184,6 +186,17 @@ with col1:
             help='Selecciona el tipo de fórmula para tu oferta'
         )
         
+        # Add the sophistication level selector
+        sophistication_level = st.selectbox(
+            '🎯 Nivel de Sofisticación del Mercado',
+            options=list(sophistication_levels.keys()),
+            help='Adapta tu mensaje según la madurez del mercado (Eugene Schwartz)'
+        )
+        
+        # Show example and strategy for the selected level
+        if sophistication_level:
+            st.info(f"**Estrategia:** {sophistication_levels[sophistication_level]['strategy']}\n\n**Ejemplo:** *{sophistication_levels[sophistication_level]['example']}*")
+        
         temperature = st.slider('🌡️ Nivel de Creatividad', min_value=0.0, max_value=2.0, value=1.0,
                             help='Valores más altos hacen que el resultado sea más creativo pero menos enfocado')
 
@@ -200,6 +213,10 @@ with col2:
             
             # Get skills from session state or use empty string
             skills_value = st.session_state.skills if hasattr(st.session_state, 'skills') and st.session_state.skills else ""
+            
+            # Get sophistication level from session state or use default
+            sophistication_level_value = st.session_state.sophistication_level if hasattr(st.session_state, 'sophistication_level') and st.session_state.sophistication_level else list(sophistication_levels.keys())[0]
+            sophistication_data = sophistication_levels[sophistication_level_value]
                 
             # Preparar el contenido del archivo si existe
             file_content = ""
@@ -212,7 +229,8 @@ with col2:
                 product_name=product_service_value,
                 selected_formula_name=st.session_state.formula_type,
                 file_content=file_content,
-                skills=skills_value
+                skills=skills_value,
+                sophistication_level=sophistication_data  # Pass the sophistication level data
             )
                 
             # Add instruction for generating benefit bullets based on the promise