JeCabrera commited on
Commit
71e4038
·
verified ·
1 Parent(s): bfaca36

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -24
app.py CHANGED
@@ -229,30 +229,9 @@ if prompt := st.chat_input('¿Cuál es tu producto o servicio?'): # Mensaje má
229
  # Procesamiento normal para otras preguntas
230
  # Save this as a chat for later
231
  if st.session_state.chat_id not in past_chats.keys():
232
- # Es una nueva conversación, generemos un título basado en el primer mensaje
233
- # Primero, guardamos un título temporal
234
- temp_title = f'PUV-{st.session_state.chat_id}'
235
- past_chats[st.session_state.chat_id] = temp_title
236
-
237
- # Generamos un título basado en el contenido del mensaje
238
- try:
239
- # Usamos el mismo modelo para generar un título corto
240
- title_generator = genai.GenerativeModel('gemini-2.0-flash')
241
- title_response = title_generator.generate_content(
242
- f"Genera un título corto (máximo 5 palabras) que describa de qué producto o servicio trata esta consulta, sin usar comillas ni puntuación: '{prompt}'")
243
-
244
- # Obtenemos el título generado
245
- generated_title = title_response.text.strip()
246
-
247
- # Actualizamos el título en past_chats
248
- if generated_title:
249
- st.session_state.chat_title = f"PUV: {generated_title}"
250
- past_chats[st.session_state.chat_id] = f"PUV: {generated_title}"
251
- else:
252
- st.session_state.chat_title = temp_title
253
- except Exception as e:
254
- print(f"Error al generar título: {e}")
255
- st.session_state.chat_title = temp_title
256
  else:
257
  # Ya existe esta conversación, usamos el título guardado
258
  st.session_state.chat_title = past_chats[st.session_state.chat_id]
 
229
  # Procesamiento normal para otras preguntas
230
  # Save this as a chat for later
231
  if st.session_state.chat_id not in past_chats.keys():
232
+ # Es una nueva conversación, usamos un título fijo
233
+ st.session_state.chat_title = f"PUV Session {time.strftime('%d/%m/%Y')}"
234
+ past_chats[st.session_state.chat_id] = st.session_state.chat_title
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
235
  else:
236
  # Ya existe esta conversación, usamos el título guardado
237
  st.session_state.chat_title = past_chats[st.session_state.chat_id]