From 633010ff62266f425e53706966ed7c2e10c0b500 Mon Sep 17 00:00:00 2001
From: Dan Birman <danbirman@gmail.com>
Date: Mon, 4 Nov 2024 11:25:31 -0800
Subject: [PATCH] feat: adding redirects that return correctly to the same page

---
 src/aind_qc_portal/panel/metric.py          | 2 +-
 src/aind_qc_portal/panel/quality_control.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/aind_qc_portal/panel/metric.py b/src/aind_qc_portal/panel/metric.py
index fc0886c..4e287bd 100644
--- a/src/aind_qc_portal/panel/metric.py
+++ b/src/aind_qc_portal/panel/metric.py
@@ -60,7 +60,7 @@ def _set_status(self, status: Status | str):
         status : Status or str
         """
         if pn.state.user == 'guest':
-            self.hidden_html.object = "<script>window.location.href = '/login';</script>"
+            self.hidden_html.object = f"<script>window.location.href = '/login?next={pn.state.location.href}';</script>"
             return
 
         if not isinstance(status, Status):
diff --git a/src/aind_qc_portal/panel/quality_control.py b/src/aind_qc_portal/panel/quality_control.py
index fcb8f63..da96295 100644
--- a/src/aind_qc_portal/panel/quality_control.py
+++ b/src/aind_qc_portal/panel/quality_control.py
@@ -93,7 +93,7 @@ def submit_changes(self, *event):
 
         # redirect users to login 
         if pn.state.user == 'guest':
-            self.hidden_html.object = "<script>window.location.href = '/login';</script>"
+            self.hidden_html.object = f"<script>window.location.href = '/login?next={pn.state.location.href}';</script>"
             return
 
         qc_update_to_id(self.id, self.data)