Reset modal image when a new image is uploaded

InvenTree/company/templates/company/company_base.html

@@ -158,6 +158,12 @@
     function reloadImage(data) {
         if (data.image) {
             $('#company-image').attr('src', data.image);
+
+            // Reset the "modal image" view
+            $('#company-image').click(function() {
+                showModalImage(data.image);
+            });
+
         } else {
             location.reload();
         }

InvenTree/part/templates/part/part_base.html

@@ -328,6 +328,12 @@
         // If image / thumbnail data present, live update
         if (data.image) {
             $('#part-image').attr('src', data.image);
+
+            // Reset the "modal image" view
+            $('#part-thumb').click(function() {
+                showModalImage(data.image);
+            });
+
         } else {
             // Otherwise, reload the page
             location.reload();