Update app.py

app.py

@@ -16,6 +16,8 @@ from bullets.generator import create_bullet_instruction
 from bonuses.generator import create_bonus_instruction
 # Import the offer instruction generator
 from prompts import create_offer_instruction
+# Import sophistication levels
+from sophistication import sophistication_levels
 
 # Set page to wide mode to use full width
 st.set_page_config(layout="wide")
@@ -99,6 +101,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
 
     # Keep only the manual input tab
     with st.container():
@@ -115,6 +118,13 @@ with col1:
         target_audience = st.text_area('👥 Público Objetivo', height=70,
                                     help='Describe tu cliente o público ideal')
         
+        # Add sophistication level selector
+        sophistication_level = st.selectbox(
+            '🧠 Nivel de Sofisticación del Mercado',
+            options=list(sophistication_levels.keys()),
+            help='Selecciona el nivel de conocimiento que tiene tu público sobre soluciones similares'
+        )
+        
         # Add file/image uploader here
         uploaded_file = st.file_uploader("📄 Sube un archivo o imagen", 
                                         type=['txt', 'pdf', 'docx', 'jpg', 'jpeg', 'png'])
@@ -212,7 +222,8 @@ with col2:
                 product_service=product_service_value,
                 selected_formula_name=st.session_state.formula_type,
                 file_content=file_content,
-                skills=skills_value
+                skills=skills_value,
+                sophistication_level=sophistication_levels.get(st.session_state.sophistication_level) if hasattr(st.session_state, 'sophistication_level') else None
             )
                 
             # Add instruction for generating benefit bullets based on the promise