Upload app.py

app.py

@@ -293,12 +293,12 @@ with chat_container:
 chat_input_container = st.container()
 
 # Mostrar mensajes existentes
-for message in st.session_state.messages:
-    with st.chat_message(
-        name=message['role'],
-        avatar=AI_AVATAR_ICON if message['role'] == MODEL_ROLE else USER_AVATAR_ICON
-    ):
-        st.markdown(message['content'])
+    for message in st.session_state.messages:
+        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:
@@ -309,14 +309,14 @@ for message in st.session_state.messages:
             st.markdown(prompt)
         add_message("user", prompt, USER_AVATAR_ICON)
         
-        # Actualizar título antes de procesar
+        # Actualizar título y procesar respuesta
         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.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,
@@ -324,32 +324,29 @@ for message in st.session_state.messages:
             'title': st.session_state.chat_title
         }
         
-        # Procesar la respuesta una sola vez
         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()
+    # Chat input al final
+    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)
+        
+        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
+        }
+        
+        process_model_response(new_prompt)
+        update_chat_memory()