fix greyed save button

Author: hanna-paasivirta

lib/lightning_web/live/ai_assistant/component.ex

@@ -271,19 +271,12 @@ defmodule LightningWeb.AiAssistant.Component do
           streaming_status: nil
         )
 
-        # Call callback if there's code to apply
+        # Always call callback to notify message received (sets sending_ai_message: false)
         case socket.assigns.callbacks[:on_message_received] do
           callback when is_function(callback, 2) ->
             code = payload_data[:code]
-
-            if code != nil and code != "" do
-              Logger.info("[Component] Applying workflow code from saved message")
-              last_message = final_session.messages |> List.last()
-              callback.(code, last_message)
-            else
-              Logger.info("[Component] No workflow code to apply")
-            end
-
+            last_message = final_session.messages |> List.last()
+            callback.(code, last_message)
             socket
 
           _ ->

lib/lightning_web/live/workflow_live/workflow_ai_chat_component.ex

@@ -31,10 +31,16 @@ defmodule LightningWeb.WorkflowLive.WorkflowAiChatComponent do
         },
         socket
       ) do
-    {:ok,
-     socket
-     |> assign(session_or_message: session_or_message)
-     |> push_event("template_selected", %{template: code})}
+    socket = assign(socket, session_or_message: session_or_message)
+
+    socket =
+      if code != nil and code != "" do
+        push_event(socket, "template_selected", %{template: code})
+      else
+        socket
+      end
+
+    {:ok, socket}
   end
 
   def update(assigns, socket) do