Use helper to update flash in streams

app/helpers/application_helper.rb

@@ -8,4 +8,10 @@ module ApplicationHelper
   def current_user
     @current_user ||= User.find(session[:user_id]) if logged_in?
   end
+
+  def update_flash
+    turbo_stream.replace "flash_wrapper" do
+      render "shared/flash"
+    end
+  end
 end

app/views/sessions/new.turbo_stream.erb

@@ -1,7 +1,6 @@
 <% @title = "Log in" %>
-<%= turbo_stream.replace "flash_wrapper" do %>
-  <%= render "shared/flash" %>
-<% end %>
+
+<%= update_flash %>
 
 <%= turbo_stream.replace "session_form" do %>
   <%= form_with url: sessions_path, method: :post, id: "session_form" do |f| %>

app/views/users/edit.turbo_stream.erb

@@ -1,7 +1,6 @@
 <% @title = "Edit account details" %>
-<%= turbo_stream.replace "flash_wrapper" do %>
-  <%= render "shared/flash" %>
-<% end %>
+
+<%= update_flash %>
 
 <%= turbo_stream.replace "user_form" do %>
   <%= render partial: "form",

app/views/users/new.turbo_stream.erb

@@ -1,7 +1,6 @@
 <% @title = "Register" %>
-<%= turbo_stream.replace "flash_wrapper" do %>
-  <%= render "shared/flash" %>
-<% end %>
+
+<%= update_flash %>
 
 <%= turbo_stream.replace "user_form" do %>
   <%= render partial: "form",