Upload app.py

app.py

@@ -255,47 +255,27 @@ def process_model_response(prompt, max_retries=3):
     return False
 
 # Modificar la función handle_example_click para usar las nuevas funciones
-# Simplificar a:
-def handle_example_click(prompt_text):
-    st.session_state.update({
-        'show_examples': False,
-        'messages': [],
-        'current_chat_id': str(time.time()),
-        'gemini_history': [],
-        'chat_title': get_chat_title([])
-    })
-    
-    # Añadir esta verificación crítica
-    if st.session_state.current_chat_id not in st.session_state.chats_in_memory:
-        st.session_state.chats_in_memory[st.session_state.current_chat_id] = {
-            'messages': [],
-            'gemini_history': [],
-            'title': 'Nuevo Chat'
-        }
-    
-    process_model_response(prompt_text)
-
-# Modificar la sección principal de manejo de chat
-if prompt := st.chat_input('¿En qué puedo ayudarte hoy?'):
-    is_first_message = len(st.session_state.messages) == 0
+# Actualicé el código para generar títulos automáticos usando el modelo de Gemini. Aquí los cambios clave:
+# En la sección principal de manejo de chat (línea ~286)
+if is_first_message:
+    # Generar título basado en el primer mensaje del usuario
+    try:
+        title_response = st.session_state.model.generate_content(
+            f"Genera un título corto (máximo 5 palabras) que describa esta consulta, sin comillas: '{prompt}'"
+        )
+        generated_title = title_response.text.strip().replace('"', '')[:30]
+        st.session_state.chat_title = generated_title or f"Chat-{st.session_state.current_chat_id}"
+    except Exception as e:
+        st.session_state.chat_title = f"Chat-{st.session_state.current_chat_id}"
     
-    if st.session_state.current_chat_id not in st.session_state.chats_in_memory:
-        st.session_state.chats_in_memory[st.session_state.current_chat_id] = {
-            'messages': [],
-            'gemini_history': [],
-            'title': 'Nuevo Chat'
-        }
-        
-    add_message('user', prompt, USER_AVATAR_ICON)
-    with st.chat_message('user', avatar=USER_AVATAR_ICON):
-        st.markdown(prompt)
+    # Actualizar título en memoria
+    st.session_state.chats_in_memory[st.session_state.current_chat_id]['title'] = st.session_state.chat_title
     
-    if is_first_message:
-        add_message(MODEL_ROLE, WELCOME_MESSAGE, AI_AVATAR_ICON)
-        with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON, key="welcome_msg"):
-            st.markdown(WELCOME_MESSAGE)
-        update_chat_memory()
-        st.experimental_rerun()
+    add_message(MODEL_ROLE, WELCOME_MESSAGE, AI_AVATAR_ICON)
+    with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON, key="welcome_msg"):
+        st.markdown(WELCOME_MESSAGE)
+    update_chat_memory()
+    st.experimental_rerun()
     
     process_model_response(prompt)
 
@@ -328,4 +308,33 @@ if st.session_state.show_examples and not st.session_state.messages:
                 handle_example_click("Necesito una PUV para mi servicio de consultoría en marketing digital")
     
     st.markdown("---")
+    
+    # Actualizar la función handle_example_click (línea ~250)
+    def handle_example_click(prompt_text):
+        st.session_state.update({
+            'show_examples': False,
+            'messages': [],
+            'current_chat_id': str(time.time()),
+            'gemini_history': [],
+            'chat_title': 'Nuevo Chat'  # Placeholder temporal
+        })
+        
+        if st.session_state.current_chat_id not in st.session_state.chats_in_memory:
+            st.session_state.chats_in_memory[st.session_state.current_chat_id] = {
+                'messages': [],
+                'gemini_history': [],
+                'title': 'Nuevo Chat'
+            }
+        
+        # Generar título para ejemplos
+        try:
+            title_response = st.session_state.model.generate_content(
+                f"Título para consulta de ejemplo: '{prompt_text}' (máximo 4 palabras)"
+            )
+            st.session_state.chat_title = title_response.text.strip()[:25]
+        except Exception as e:
+            st.session_state.chat_title = f"Ejemplo-{time.strftime('%H:%M')}"
+        
+        st.session_state.chats_in_memory[st.session_state.current_chat_id]['title'] = st.session_state.chat_title
+        process_model_response(prompt_text)