add helper for training_target

Author: ramsayleung

app/controllers/trained_messages_controller.rb

@@ -96,7 +96,7 @@ def bulk_update
       end
     end
 
-    message_type_name = TrainedMessage.message_types[new_message_type_symbol]
+    message_type_name = new_message_type_symbol.to_s
     flash[:notice] = "Successfully marked #{update_count} messages as #{message_type_name}"
     redirect_to trained_messages_path(params.except(:commit, :trained_message_ids, :authenticity_token, :controller, :action).to_unsafe_h)
   end

app/helpers/trained_messages_helper.rb

@@ -15,4 +15,15 @@ def message_type_classes(message_type)
       "bg-gray-100 text-gray-800"
     end
   end
+
+  def training_target_classes(training_target)
+    case training_target
+    when "message_content"
+      "bg-blue-100 text-blue-800"
+    when "user_name"
+      "bg-purple-100 text-purple-800"
+    else
+      "bg-yellow-100 text-yellow-800"
+    end
+  end
 end

app/views/trained_messages/index.html.erb

@@ -168,8 +168,7 @@
                   </td>
                   <td class="px-6 py-4 whitespace-nowrap">
                     <% if trained_message.training_target %>
-                        <span class="inline-flex items-center px-2.5 py-0.5 rounded-full text-xs font-medium
-                                     <%= trained_message.training_target == 'message_content' ? 'bg-green-100 text-green-800' :  'bg-yellow-100 text-green-800'%>">
+                        <span class="inline-flex items-center px-2.5 py-0.5 rounded-full text-xs font-medium <%= training_target_classes(trained_message.training_target) %>">
                         <%= trained_message.training_target.humanize %>
                       </span>
                     <% else %>