Update sophistication/generator.py

sophistication/generator.py

@@ -1,29 +1,33 @@
-from .sophistication_levels import sophistication_levels
-
-def create_sophistication_instruction(sophistication_level_key):
-    """
-    Creates an instruction for the AI based on the selected sophistication level.
-    
-    Args:
-        sophistication_level_key: The key of the selected sophistication level
-        
-    Returns:
-        str: Instruction for the AI based on the sophistication level
-    """
-    if sophistication_level_key not in sophistication_levels:
-        # Default to level 1 if the key is not found
-        sophistication_level_key = list(sophistication_levels.keys())[0]
-    
-    level_data = sophistication_levels[sophistication_level_key]
-    
-    instruction = f"""
-    NIVEL DE SOFISTICACIÓN DEL MERCADO: {level_data['description']}
-    
-    ESTRATEGIA A APLICAR: {level_data['strategy']}
-    
-    EJEMPLO DE REFERENCIA: {level_data['example']}
-    
-    Adapta la promesa y el tono según este nivel de sofisticación del mercado.
-    """
-    
+from sophistication.sophistication_levels import sophistication_levels
+
+def create_sophistication_instruction(sophistication_key):
+    """
+    Creates instruction text based on the selected sophistication level key.
+    
+    Args:
+        sophistication_key: String key for the sophistication level (e.g., "Nivel 1 - Mercado Nuevo")
+    
+    Returns:
+        str: Instruction text for the selected sophistication level
+    """
+    # Get the sophistication level data from the dictionary using the key
+    level_data = sophistication_levels.get(sophistication_key, {})
+    
+    if not level_data:
+        return ""
+    
+    # Create the instruction text using the level data
+    instruction = f"""NIVEL DE SOFISTICACIÓN DEL MERCADO:
+{level_data.get('description', '')}
+
+Estrategia recomendada:
+{level_data.get('strategy', '')}
+
+Características de este nivel:
+- {'\n- '.join(level_data.get('characteristics', []))}
+
+Ejemplo de gancho para este nivel:
+"{level_data.get('example', '')}"
+"""
+    
     return instruction