Upload 12 files

app.py

@@ -66,41 +66,55 @@ def process_message(prompt, is_example=False):
     
     with st.chat_message(MODEL_ROLE, avatar=AI_AVATAR_ICON):
         try:
-            # Forzar streaming siempre
+            # Configurar el placeholder y el indicador de escritura
+            message_placeholder = st.empty()
+            typing_indicator = st.empty()
+            typing_indicator.markdown("*Generando respuesta...*")
+            
+            # Asegurar que siempre usamos streaming
             response = state.chat.send_message(
                 enhanced_prompt,
-                stream=True,  # Asegurar que siempre sea True
-                generation_config={"temperature": 0.9}  # Mantener configuración consistente
+                stream=True,
+                generation_config={"temperature": 0.9}
             )
             
-            message_placeholder = st.empty()
             full_response = ''
             
-            # Añadir indicador de "escribiendo..."
-            typing_indicator = st.empty()
-            typing_indicator.markdown("*Generando respuesta...*")
-            
-            # Mostrar respuesta por fragmentos
-            for chunk in response:
-                for ch in chunk.text:
-                    full_response += ch
-                    time.sleep(0.01)  # Mantener delay consistente
-                    message_placeholder.write(full_response + '▌')
+            # Mejorar el manejo del streaming
+            try:
+                for chunk in response:
+                    if chunk.text:  # Verificar que hay texto en el chunk
+                        for ch in chunk.text:
+                            full_response += ch
+                            time.sleep(0.01)
+                            # Mantener el indicador mientras se escribe
+                            typing_indicator.markdown("*Generando respuesta...*")
+                            message_placeholder.markdown(full_response + '▌')
+            except Exception as e:
+                st.error(f"Error en el streaming: {str(e)}")
+                return
             
-            # Eliminar indicador de escritura
+            # Limpiar el indicador de escritura al finalizar
             typing_indicator.empty()
             
-            # Mostrar respuesta completa
-            message_placeholder.write(full_response)
+            # Mostrar respuesta final
+            message_placeholder.markdown(full_response)
             
-            # Añadir respuesta al historial
+            # Actualizar el historial
             state.add_message(
                 role=MODEL_ROLE,
-                content=state.chat.history[-1].parts[0].text,
-                avatar=AI_AVATAR_ICON,
+                content=full_response,
+                avatar=AI_AVATAR_ICON
             )
             
-            state.gemini_history = state.chat.history
+            # Guardar el historial actualizado
+            state.save_chat_history()
+            
+        except Exception as e:
+            st.error(f"Error: {str(e)}")
+            return
+
+    state.gemini_history = state.chat.history
             
             # Guardar historial actualizado
             state.save_chat_history()