Upload app.py

app.py

@@ -102,11 +102,50 @@ def update_chat_memory():
 def handle_model_error(error, retry_count, max_retries):
     if retry_count >= max_retries:
         error_message = f"Lo siento, estoy experimentando problemas para procesar tu solicitud. Por favor, intenta de nuevo más tarde. Error: {str(error)}"
-        with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON):
-            st.error(error_message)
-        add_message(MODEL_ROLE, error_message, AI_AVATAR_ICON)
-        update_chat_memory()
-        return True
+        return error_message
+    return None
+
+def process_model_response(prompt, max_retries=3):
+    retry_count = 0
+    while retry_count < max_retries:
+        try:
+            # Crear el mensaje del modelo primero
+            with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON):
+                # Añade un indicador de "escribiendo..."
+                typing_indicator = st.empty()
+                typing_indicator.markdown("*🤖 RoboCopy está escribiendo...*")
+                
+                response = st.session_state.chat.send_message(prompt, stream=True)
+                mensaje_completo = ""
+                mensaje_actual = ""
+                
+                # Procesar la respuesta por chunks con efecto de escritura
+                for chunk in response:
+                    mensaje_completo += chunk.text
+                    for caracter in chunk.text:
+                        mensaje_actual += caracter
+                        typing_indicator.markdown(mensaje_actual + "▌")
+                        time.sleep(0.02)  # Velocidad de escritura ajustada
+                
+                # Mostrar mensaje final y limpiar indicador
+                typing_indicator.markdown(mensaje_completo)
+            
+            add_message(MODEL_ROLE, mensaje_completo, AI_AVATAR_ICON)
+            st.session_state.gemini_history = st.session_state.chat.history
+            update_chat_memory()
+            return True
+            
+        except Exception as e:
+            retry_count += 1
+            error_msg = handle_model_error(e, retry_count, max_retries)
+            if error_msg:
+                with st.chat_message(name=MODEL_ROLE, avatar=AI_AVATAR_ICON):
+                    st.error(error_msg)
+                add_message(MODEL_ROLE, error_msg, AI_AVATAR_ICON)
+                update_chat_memory()
+                return False
+            wait_time = (1.5 ** retry_count)
+            time.sleep(wait_time)
     return False
 
 def mostrar_con_efecto_escritura(texto, velocidad=0.05):