Upload app.py

app.py

@@ -262,10 +262,9 @@ if not st.session_state.gemini_history:
     st.session_state.gemini_history = st.session_state.chat.history
 
 # === MOSTRAR MENSAJES DEL HISTORIAL ===
-messages_container = st.container()
-chat_input_container = st.container()
+chat_container = st.container()
 
-with messages_container:
+with chat_container:
     # === SECCIÓN DE EJEMPLOS ===
     if st.session_state.show_examples and not st.session_state.messages:
         st.title("💡 RoboCopy - Asistente de PUVs")
@@ -295,54 +294,29 @@ chat_input_container = st.container()
 
 # Mostrar mensajes existentes
 for message in st.session_state.messages:
-    with st.chat_message(name=message['role'], avatar=message.get('avatar', USER_AVATAR_ICON if message['role'] == 'user' else AI_AVATAR_ICON)):
+    with st.chat_message(
+        name=message['role'],
+        avatar=AI_AVATAR_ICON if message['role'] == MODEL_ROLE else USER_AVATAR_ICON
+    ):
         st.markdown(message['content'])
-
-# Procesar entrada del usuario si existe
-if 'user_input' in st.session_state:
-    prompt = st.session_state.user_input
-    del st.session_state.user_input
-    
-    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"Ejemplo-{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()
-
-# Mantener el chat_input siempre visible al final
-with chat_input_container:
-    new_prompt = st.chat_input('¿En qué puedo ayudarte hoy?')
-    if new_prompt:
-        # Simular el envío del mensaje
+    # Procesar entrada del usuario si existe
+    if 'user_input' in st.session_state:
+        prompt = st.session_state.user_input
+        del st.session_state.user_input
+        
         with st.chat_message("user", avatar=USER_AVATAR_ICON):
-            st.markdown(new_prompt)
-        add_message("user", new_prompt, 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: '{new_prompt}' (máximo 4 palabras)"
+                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.chat_title = f"Ejemplo-{time.strftime('%H:%M')}"
         
         st.session_state.chats_in_memory[st.session_state.current_chat_id] = {
             'messages': st.session_state.messages,
@@ -351,5 +325,31 @@ with chat_input_container:
         }
         
         # Procesar la respuesta una sola vez
-        process_model_response(new_prompt)
-        update_chat_memory()
+        process_model_response(prompt)
+        update_chat_memory()
+
+# Chat input al final del contenedor
+new_prompt = st.chat_input('¿En qué puedo ayudarte hoy?')
+if new_prompt:
+    with st.chat_message("user", avatar=USER_AVATAR_ICON):
+        st.markdown(new_prompt)
+    add_message("user", new_prompt, USER_AVATAR_ICON)
+    
+    # Actualizar título antes de procesar
+    try:
+        title_response = st.session_state.model.generate_content(
+            f"Título para consulta: '{new_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(new_prompt)
+    update_chat_memory()