From 95d16690ef23e99b0c54d7d7bd75a403695f38c6 Mon Sep 17 00:00:00 2001
From: Tim Lorsbach <tim@lorsba.ch>
Date: Tue, 11 Nov 2025 20:38:13 +0100
Subject: [PATCH] added retrain code

---
 epdb/views.py                                 |  6 ++
 .../modals/objects/retrain_model_modal.html   | 86 +++++++++++--------
 2 files changed, 55 insertions(+), 37 deletions(-)

diff --git a/epdb/views.py b/epdb/views.py
index 9af2ebe5..784287ca 100644
--- a/epdb/views.py
+++ b/epdb/views.py
@@ -955,6 +955,12 @@ def package_model(request, package_uuid, model_uuid):
                 ]
                 dispatch(current_user, evaluate_model, current_model.pk, multigen, eval_package_ids)
 
+                return redirect(current_model.url)
+            elif hidden == "retrain":
+                from .tasks import dispatch, retrain
+
+                dispatch(current_user, retrain, current_model.pk)
+
                 return redirect(current_model.url)
             else:
                 return HttpResponseBadRequest()
diff --git a/templates/modals/objects/retrain_model_modal.html b/templates/modals/objects/retrain_model_modal.html
index 9ed7745f..8bc90301 100644
--- a/templates/modals/objects/retrain_model_modal.html
+++ b/templates/modals/objects/retrain_model_modal.html
@@ -1,43 +1,55 @@
-<div class="modal fade" tabindex="-1" id="retrain_model_modal" role="dialog" aria-labelledby="retrain_model_modal"
-     aria-hidden="true">
-    <div class="modal-dialog modal-lg">
-        <div class="modal-content">
-            <div class="modal-header">
-                <button type="button" class="close" data-dismiss="modal">
-                    <span aria-hidden="true">&times;</span>
-                    <span class="sr-only">Close</span>
-                </button>
-                <h4 class="modal-title">Retrain Model</h4>
-            </div>
-            <div class="modal-body">
-                <form id="retrain_model_form" accept-charset="UTF-8" action="{{ meta.current_package.url }}/model"
-                      data-remote="true" method="post">
-                    <div class="jumbotron">
-                        To reflect changes in the rule or data packages, you can use the "Retrain" button,
-                        to let the model reflect the changes without creating a new model.
-                        While the model is retraining, it will be unavailable for prediction.
-                    </div>
-                    {% csrf_token %}
-                    <input type="hidden" name="action" value="retrain">
-                </form>
-            </div>
-            <div class="modal-footer">
-                <a id="retrain_model_form_submit" class="btn btn-primary" href="#">Retrain</a>
-                <button type="button" class="btn btn-default" data-dismiss="modal">Cancel</button>
-            </div>
-        </div>
+<div
+  class="modal fade"
+  tabindex="-1"
+  id="retrain_model_modal"
+  role="dialog"
+  aria-labelledby="retrain_model_modal"
+  aria-hidden="true"
+>
+  <div class="modal-dialog modal-lg">
+    <div class="modal-content">
+      <div class="modal-header">
+        <button type="button" class="close" data-dismiss="modal">
+          <span aria-hidden="true">&times;</span>
+          <span class="sr-only">Close</span>
+        </button>
+        <h4 class="modal-title">Retrain Model</h4>
+      </div>
+      <div class="modal-body">
+        <form
+          id="retrain_model_form"
+          accept-charset="UTF-8"
+          action="{{ meta.current_object.url }}"
+          data-remote="true"
+          method="post"
+        >
+          <div class="jumbotron">
+            To reflect changes in the rule or data packages, you can use the
+            "Retrain" button, to let the model reflect the changes without
+            creating a new model. While the model is retraining, it will be
+            unavailable for prediction.
+          </div>
+          {% csrf_token %}
+          <input type="hidden" name="hidden" value="retrain" />
+        </form>
+      </div>
+      <div class="modal-footer">
+        <a id="retrain_model_form_submit" class="btn btn-primary" href="#"
+          >Retrain</a
+        >
+        <button type="button" class="btn btn-default" data-dismiss="modal">
+          Cancel
+        </button>
+      </div>
     </div>
+  </div>
 </div>
 
 <script>
-
-    $(function () {
-
-        $('#retrain_model_form_submit').on('click', function (e) {
-            e.preventDefault();
-            $('#retrain_model_form').submit();
-        });
-
+  $(function () {
+    $("#retrain_model_form_submit").on("click", function (e) {
+      e.preventDefault();
+      $("#retrain_model_form").submit();
     });
-
+  });
 </script>