Upload 12 files

app.py

@@ -29,10 +29,43 @@ def process_message(prompt, is_example=False):
     handle_chat_title(prompt)
     
     # Mostrar y guardar mensaje del usuario
+    # Mostrar mensaje del usuario
     with st.chat_message('user', avatar=USER_AVATAR_ICON):
         st.markdown(prompt)
+    
+    # Añadir mensaje del usuario al historial
     state.add_message('user', prompt, USER_AVATAR_ICON)
     
+    # Preparar el prompt según el tipo de mensaje
+    if is_greeting(prompt):
+        enhanced_prompt = f"El usuario te ha saludado con '{prompt}'. Preséntate brevemente..."
+    elif is_example:
+        enhanced_prompt = f"El usuario ha seleccionado un ejemplo: '{prompt}'..."
+    else:
+        enhanced_prompt = prompt
+    
+    # Mover la respuesta del modelo después del mensaje del usuario
+    with st.chat_message(MODEL_ROLE, avatar=AI_AVATAR_ICON):
+        try:
+            message_placeholder = st.empty()
+            typing_indicator = st.empty()
+            typing_indicator.markdown("*Generando respuesta...*")
+            
+            response = state.send_message(enhanced_prompt)
+            full_response = ''
+            
+            try:
+                for chunk in response:
+                    if chunk.text:
+                        for ch in chunk.text:
+                            full_response += ch
+                            time.sleep(0.01)
+                            typing_indicator.markdown("*Generando respuesta...*")
+                            message_placeholder.markdown(full_response + '▌')
+            except Exception as e:
+                st.error(f"Error en el streaming: {str(e)}")
+                return
+    
     # Obtener el prompt mejorado
     enhanced_prompt = get_enhanced_prompt(prompt, is_example)