JeCabrera commited on
Commit
8b67c3a
·
verified ·
1 Parent(s): 7e7706a

Upload app (11).py

Browse files
Files changed (1) hide show
  1. app (11).py +324 -0
app (11).py ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ # Configurar la interfaz de usuario con Streamlit - MUST BE FIRST!
3
+ st.set_page_config(page_title="Generador de Cliente Ideal", page_icon="👤", layout="wide")
4
+
5
+ from dotenv import load_dotenv
6
+ import os
7
+ import google.generativeai as genai
8
+ from style import styles
9
+ from prompts import create_instruction
10
+ from consciousness_levels import CONSCIOUSNESS_LEVELS
11
+
12
+ # Cargar las variables de entorno
13
+ load_dotenv()
14
+
15
+ # Configurar la API de Google
16
+ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
17
+
18
+ # Inicializar variables de estado en session_state si no existen
19
+ if 'perfil_cliente' not in st.session_state:
20
+ st.session_state.perfil_cliente = None
21
+ if 'producto' not in st.session_state:
22
+ st.session_state.producto = ""
23
+ if 'habilidades' not in st.session_state:
24
+ st.session_state.habilidades = ""
25
+ if 'creatividad' not in st.session_state:
26
+ st.session_state.creatividad = 1.0
27
+ # Update the default format in session state
28
+ if 'formato' not in st.session_state:
29
+ st.session_state.formato = "Jung's_Avatar" # Changed from previous format name
30
+ if 'nivel_conciencia' not in st.session_state:
31
+ # Usar el primer nivel del diccionario como valor predeterminado
32
+ first_key = list(CONSCIOUSNESS_LEVELS.keys())[0]
33
+ st.session_state.nivel_conciencia = first_key.replace("_", " ")
34
+
35
+ # Función para generar el perfil de cliente ideal
36
+ @st.cache_resource
37
+ def get_model(temperature):
38
+ generation_config = {
39
+ "temperature": temperature,
40
+ }
41
+ return genai.GenerativeModel('gemini-2.0-flash', generation_config=generation_config)
42
+
43
+ # REMOVED: The format selection dropdown that was here
44
+
45
+ def generate_buyer_persona(product, skills, target_audience, temperature, consciousness_level="Ninguno", format_type="Jung's_Avatar"):
46
+ if not product or not skills:
47
+ return "Por favor, completa los campos de producto y habilidades."
48
+
49
+ try:
50
+ model = get_model(temperature)
51
+ instruction = create_instruction(
52
+ product_service=product,
53
+ skills=skills,
54
+ target_audience=target_audience,
55
+ consciousness_level=consciousness_level,
56
+ format_type=format_type
57
+ )
58
+
59
+ # Añadir instrucción explícita para respuesta en español
60
+ instruction += "\n\nIMPORTANTE: La respuesta debe estar completamente en español."
61
+
62
+ response = model.generate_content([instruction], generation_config={"temperature": temperature})
63
+ return response.parts[0].text if response and response.parts else "Error generando el perfil de cliente ideal."
64
+ except Exception as e:
65
+ return f"Error al generar el perfil: {str(e)}"
66
+
67
+ # Modificar la función update_profile para que no use spinner
68
+ def update_profile():
69
+ # Solo actualizar la variable de sesión
70
+ st.session_state.submitted = True
71
+
72
+ # Leer el contenido del archivo manual.md si existe
73
+ try:
74
+ with open("manual.md", "r", encoding="utf-8") as file:
75
+ manual_content = file.read()
76
+ # Mostrar el contenido del manual en el sidebar
77
+ st.sidebar.markdown(manual_content)
78
+ except FileNotFoundError:
79
+ st.sidebar.warning("Manual not found. Please create a manual.md file.")
80
+ except Exception as e:
81
+ st.sidebar.error(f"Error loading manual: {str(e)}")
82
+
83
+ # Ocultar elementos de la interfaz
84
+ st.markdown(styles["main_layout"], unsafe_allow_html=True)
85
+
86
+ # Centrar el título y el subtítulo
87
+ st.markdown("<h1 style='text-align: center;'>Generador de Perfil de Cliente Ideal</h1>", unsafe_allow_html=True)
88
+ 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)
89
+
90
+ # Añadir CSS personalizado para el botón
91
+ st.markdown(styles["button"], unsafe_allow_html=True)
92
+ # Añadir CSS personalizado para el botón de descarga
93
+ st.markdown(styles["download_button"], unsafe_allow_html=True)
94
+
95
+ # Crear columnas
96
+ col1, col2 = st.columns([1, 2])
97
+
98
+ # Columna de entrada
99
+ with col1:
100
+ producto = st.text_area("¿Qué producto o servicio ofreces?",
101
+ value=st.session_state.producto,
102
+ placeholder="Ejemplo: Curso de Inglés",
103
+ key="producto_input",
104
+ height=70)
105
+ st.session_state.producto = producto
106
+
107
+ habilidades = st.text_area("¿Cuáles son tus habilidades principales?",
108
+ value=st.session_state.habilidades,
109
+ placeholder="Ejemplo: Enseñanza, comunicación, diseño de contenidos",
110
+ key="habilidades_input",
111
+ height=70)
112
+ st.session_state.habilidades = habilidades
113
+
114
+ # Botón para generar - Movido arriba del acordeón
115
+ submit = st.button("CREAR MI CLIENTE IDEAL SOÑADO ➤➤", on_click=update_profile)
116
+
117
+ # Crear un acordeón para las opciones de personalización
118
+ with st.expander("Personaliza Tu Cliente Ideal Soñado"):
119
+ # Nuevo campo para público objetivo
120
+ if 'publico_objetivo' not in st.session_state:
121
+ st.session_state.publico_objetivo = ""
122
+
123
+ publico_objetivo = st.text_area("¿Cuál es tu público objetivo? (opcional)",
124
+ value=st.session_state.publico_objetivo,
125
+ placeholder="Ejemplo: Profesionales entre 25-40 años interesados en desarrollo personal",
126
+ key="publico_objetivo_input",
127
+ height=70)
128
+ st.session_state.publico_objetivo = publico_objetivo
129
+
130
+ # Selector de formato
131
+ from format.format import buyer_persona_formats
132
+
133
+ # Usar directamente las claves del diccionario sin filtrar
134
+ formato = st.selectbox(
135
+ "Formato del perfil",
136
+ options=list(buyer_persona_formats.keys()),
137
+ format_func=lambda x: x.capitalize(),
138
+ index=list(buyer_persona_formats.keys()).index(st.session_state.formato) if st.session_state.formato in buyer_persona_formats else 0,
139
+ help="Selecciona el formato en el que se presentará el perfil del cliente ideal"
140
+ )
141
+ st.session_state.formato = formato
142
+
143
+ # Nivel de creatividad con slider
144
+ creatividad = st.slider("Nivel de creatividad",
145
+ min_value=0.0,
146
+ max_value=2.0,
147
+ value=st.session_state.creatividad,
148
+ step=0.1,
149
+ key="creatividad_slider")
150
+ st.session_state.creatividad = creatividad
151
+
152
+ # Selector de nivel de conciencia
153
+ consciousness_options = []
154
+ for i, key in enumerate(CONSCIOUSNESS_LEVELS.keys(), 1):
155
+ # Replace underscores with spaces in the key
156
+ display_name = key.replace("_", " ")
157
+ consciousness_options.append(f"Nivel {i} - {display_name}")
158
+
159
+ nivel_conciencia_display = st.selectbox(
160
+ "Nivel de conciencia del cliente ideal",
161
+ consciousness_options,
162
+ index=0,
163
+ help="Selecciona el nivel de conciencia en el que se encuentra tu cliente ideal"
164
+ )
165
+
166
+ # Extract the original key from the display name
167
+ level_number = nivel_conciencia_display.split(" - ")[0].replace("Nivel ", "")
168
+ original_key = list(CONSCIOUSNESS_LEVELS.keys())[int(level_number) - 1]
169
+ nivel_conciencia = original_key.replace("_", " ")
170
+
171
+ # Get the description from the CONSCIOUSNESS_LEVELS dictionary
172
+ if original_key in CONSCIOUSNESS_LEVELS:
173
+ nivel_info = CONSCIOUSNESS_LEVELS[original_key]["estado_mental"]
174
+ st.info(f"**{nivel_conciencia}**: {nivel_info}")
175
+
176
+ st.session_state.nivel_conciencia = nivel_conciencia
177
+
178
+ # Columna de resultados
179
+ with col2:
180
+ # Verificar si se ha enviado el formulario
181
+ if 'submitted' in st.session_state and st.session_state.submitted:
182
+ if st.session_state.producto and st.session_state.habilidades:
183
+ with st.spinner("Creando tu Cliente Ideal Soñado..."):
184
+ # Generar el perfil del cliente
185
+ perfil_cliente = generate_buyer_persona(
186
+ st.session_state.producto,
187
+ st.session_state.habilidades,
188
+ st.session_state.publico_objetivo,
189
+ st.session_state.creatividad,
190
+ st.session_state.nivel_conciencia,
191
+ st.session_state.formato
192
+ )
193
+
194
+ # Mejorar la limpieza del perfil para eliminar estructuras JSON/diccionario
195
+ if isinstance(perfil_cliente, str):
196
+ import re
197
+ import json
198
+
199
+ # Eliminar marcadores de código markdown como ```json, ```python, etc.
200
+ perfil_cliente = re.sub(r'```[a-z]*\n', '', perfil_cliente)
201
+ perfil_cliente = re.sub(r'```', '', perfil_cliente)
202
+
203
+ # Intentar detectar y limpiar formato JSON
204
+ if '{' in perfil_cliente and '}' in perfil_cliente:
205
+ # Intentar extraer solo el contenido textual, eliminando estructuras JSON
206
+ # Primero, intentar encontrar el JSON completo
207
+ json_pattern = r'(\{.*?\})'
208
+ json_matches = re.findall(json_pattern, perfil_cliente, re.DOTALL)
209
+
210
+ if json_matches:
211
+ for json_match in json_matches:
212
+ # Intentar extraer el contenido real del JSON
213
+ try:
214
+ # Reemplazar el JSON con una cadena vacía
215
+ perfil_cliente = perfil_cliente.replace(json_match, '')
216
+ except:
217
+ pass
218
+
219
+ # Limpiar líneas que parecen ser parte de un diccionario
220
+ lines = perfil_cliente.split('\n')
221
+ cleaned_lines = []
222
+ for line in lines:
223
+ # Omitir líneas que parecen ser claves de diccionario
224
+ if not re.match(r'^\s*["\']?[a-zA-Z_]+["\']?\s*:', line):
225
+ cleaned_lines.append(line)
226
+
227
+ perfil_cliente = '\n'.join(cleaned_lines)
228
+
229
+ # Eliminar llaves sueltas y corchetes
230
+ perfil_cliente = re.sub(r'[{}[\]]', '', perfil_cliente)
231
+
232
+ # Eliminar comillas y dos puntos que parecen ser de un diccionario
233
+ perfil_cliente = re.sub(r'["\']\s*:\s*["\']', '', perfil_cliente)
234
+
235
+ # Eliminar "template", "description", "example" y otras palabras clave comunes en el formato
236
+ keywords = ["template", "description", "example", "Nivel de conciencia"]
237
+ for keyword in keywords:
238
+ perfil_cliente = re.sub(rf'["\']?{keyword}["\']?\s*:\s*["\']?', '', perfil_cliente)
239
+
240
+ # Eliminar comillas sueltas
241
+ perfil_cliente = re.sub(r'^\s*["\']|["\']$', '', perfil_cliente)
242
+
243
+ # Normalizar espacios en blanco y sangrías
244
+ lines = perfil_cliente.split('\n')
245
+ cleaned_lines = []
246
+ for line in lines:
247
+ # Eliminar sangrías excesivas pero mantener estructura básica
248
+ cleaned_line = line.strip()
249
+ if cleaned_line: # Solo agregar líneas no vacías
250
+ cleaned_lines.append(cleaned_line)
251
+
252
+ # Unir las líneas con saltos de línea adecuados
253
+ perfil_cliente = '\n'.join(cleaned_lines)
254
+
255
+ # Eliminar espacios en blanco adicionales
256
+ perfil_cliente = re.sub(r'\n\s*\n', '\n\n', perfil_cliente)
257
+ perfil_cliente = perfil_cliente.strip()
258
+
259
+ # Convertir los asteriscos de formato Markdown a HTML para una correcta visualización
260
+ # Convertir **texto** a <strong>texto</strong>
261
+ perfil_cliente = re.sub(r'\*\*(.*?)\*\*', r'<strong>\1</strong>', perfil_cliente)
262
+ # Convertir *texto* a <em>texto</em>
263
+ perfil_cliente = re.sub(r'\*(.*?)\*', r'<em>\1</em>', perfil_cliente)
264
+
265
+ # Mejorar el formato HTML para una mejor visualización
266
+ perfil_cliente = """
267
+ <div style="line-height: 1.3; text-align: left;">
268
+ {}
269
+ </div>
270
+ """.format(perfil_cliente.replace('\n', '<br><br>'))
271
+
272
+ # Guardar en session_state
273
+ st.session_state.perfil_cliente = perfil_cliente
274
+ # Resetear el estado de envío
275
+ st.session_state.submitted = False
276
+
277
+ # Mostrar resultados
278
+ if not isinstance(st.session_state.perfil_cliente, str):
279
+ st.error("Error al generar el perfil de cliente ideal")
280
+ else:
281
+ # Crear un contenedor para el resultado
282
+ result_container = st.container()
283
+
284
+ # Aplicar estilos del contenedor
285
+ st.markdown(styles["results_container"], unsafe_allow_html=True)
286
+
287
+ with result_container:
288
+ # Mostrar el título
289
+ st.markdown("<div class='results-title'>Tu Cliente Ideal</div>", unsafe_allow_html=True)
290
+
291
+ # Mostrar el contenido como HTML para asegurar el formato correcto
292
+ st.markdown(st.session_state.perfil_cliente, unsafe_allow_html=True)
293
+
294
+ # Opción para descargar
295
+ st.download_button(
296
+ label="DESCARGAR MI CLIENTE SOÑADO ➤➤",
297
+ data=st.session_state.perfil_cliente,
298
+ file_name="cliente_ideal.txt",
299
+ mime="text/plain"
300
+ )
301
+ else:
302
+ st.warning("Por favor, completa los campos de producto y habilidades antes de generar el perfil.")
303
+ # Mostrar resultados anteriores si existen
304
+ elif st.session_state.perfil_cliente:
305
+ # Crear un contenedor para el resultado
306
+ result_container = st.container()
307
+
308
+ # Aplicar estilos del contenedor
309
+ st.markdown(styles["results_container"], unsafe_allow_html=True)
310
+
311
+ with result_container:
312
+ # Mostrar el título
313
+ st.markdown("<div class='results-title'>Tu Cliente Ideal</div>", unsafe_allow_html=True)
314
+
315
+ # Mostrar el contenido como HTML para asegurar el formato correcto
316
+ st.markdown(st.session_state.perfil_cliente, unsafe_allow_html=True)
317
+
318
+ # Opción para descargar
319
+ st.download_button(
320
+ label="DESCARGAR MI CLIENTE SOÑADO ➤➤",
321
+ data=st.session_state.perfil_cliente,
322
+ file_name="cliente_ideal.txt",
323
+ mime="text/plain"
324
+ )