Upload app.py

app.py

@@ -150,9 +150,12 @@ def handle_example_click(prompt_text):
         'current_chat_id': str(time.time()),
         'gemini_history': [],
         'chat_title': 'Nuevo Chat',
-        'user_input': prompt_text  # Añadir el texto al estado
+        'user_input': prompt_text
     })
+    st.rerun()  # Forzar recarga de la página
     
+    # Eliminar todo el código restante de esta función
+    # El procesamiento se hará en la sección de entrada principal
     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': [],
@@ -292,43 +295,55 @@ if st.session_state.show_examples and not st.session_state.messages:
 # === ENTRADA DEL USUARIO ===
 if 'user_input' in st.session_state:
     prompt = st.session_state.user_input
-    del st.session_state.user_input  # Limpiar después de usar
-else:
-    prompt = st.chat_input('¿En qué puedo ayudarte hoy?')
-
-if prompt:
+    del st.session_state.user_input
+    
     # Simular el envío del mensaje
     with st.chat_message("user", avatar=USER_AVATAR_ICON):
-        st.markdown(prompt)  # Cambiado de prompt_text a prompt
-    add_message("user", prompt, USER_AVATAR_ICON)  # Cambiado de prompt_text a prompt
-    
-    # Procesar la respuesta
-    process_model_response(prompt)  # Cambiado de prompt_text a prompt
-    
-    is_first_message = len(st.session_state.messages) == 0
-    
-    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'
-        }
+        st.markdown(prompt)
+    add_message("user", prompt, USER_AVATAR_ICON)
     
+    # Actualizar título antes de procesar
     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}'"
+            f"Título para consulta: '{prompt}' (máximo 4 palabras)"
         )
-        generated_title = title_response.text.strip().replace('"', '')[:30]
-        st.session_state.chat_title = generated_title or f"Chat-{st.session_state.current_chat_id}"
+        st.session_state.chat_title = title_response.text.strip()[:25]
     except Exception as e:
-        st.session_state.chat_title = f"Chat-{st.session_state.current_chat_id}"
+        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
+    st.session_state.chats_in_memory[st.session_state.current_chat_id] = {
+        'messages': st.session_state.messages,
+        'gemini_history': st.session_state.gemini_history,
+        '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):
-            st.markdown(WELCOME_MESSAGE)
+    # Procesar la respuesta una sola vez
+    process_model_response(prompt)
     update_chat_memory()
-    
-    process_model_response(prompt)
+
+else:
+    prompt = st.chat_input('¿En qué puedo ayudarte hoy?')
+    if prompt:
+        # Simular el envío del mensaje
+        with st.chat_message("user", avatar=USER_AVATAR_ICON):
+            st.markdown(prompt)
+        add_message("user", prompt, USER_AVATAR_ICON)
+        
+        # Actualizar título antes de procesar
+        try:
+            title_response = st.session_state.model.generate_content(
+                f"Título para consulta: '{prompt}' (máximo 4 palabras)"
+            )
+            st.session_state.chat_title = title_response.text.strip()[:25]
+        except Exception as e:
+            st.session_state.chat_title = f"Chat-{time.strftime('%H:%M')}"
+        
+        st.session_state.chats_in_memory[st.session_state.current_chat_id] = {
+            'messages': st.session_state.messages,
+            'gemini_history': st.session_state.gemini_history,
+            'title': st.session_state.chat_title
+        }
+        
+        # Procesar la respuesta una sola vez
+        process_model_response(prompt)
+        update_chat_memory()