From 45145685d5a64fa535cc25f79f152fd42b7ad211 Mon Sep 17 00:00:00 2001
From: utin-francis-peter <utinfrancis.dev@gmail.com>
Date: Wed, 19 Jun 2024 23:41:38 +0100
Subject: [PATCH] fix: upon successful training of newly uploaded src doc, the
 latest doc is auto set as selected doc

---
 frontend/src/upload/Upload.tsx | 18 ++++++++++++++++--
 1 file changed, 16 insertions(+), 2 deletions(-)

diff --git a/frontend/src/upload/Upload.tsx b/frontend/src/upload/Upload.tsx
index 242070a..f170ed1 100644
--- a/frontend/src/upload/Upload.tsx
+++ b/frontend/src/upload/Upload.tsx
@@ -104,7 +104,14 @@ export default function Upload({
             .then((data) => {
               if (data.status == 'SUCCESS') {
                 if (data.result.limited === true) {
-                  getDocs().then((data) => dispatch(setSourceDocs(data)));
+                  getDocs().then((data) => {
+                    dispatch(setSourceDocs(data));
+                    dispatch(
+                      setSelectedDocs(
+                        data?.find((d) => d.location.toLowerCase() === 'local'),
+                      ),
+                    );
+                  });
                   setProgress(
                     (progress) =>
                       progress && {
@@ -114,7 +121,14 @@ export default function Upload({
                       },
                   );
                 } else {
-                  getDocs().then((data) => dispatch(setSourceDocs(data)));
+                  getDocs().then((data) => {
+                    dispatch(setSourceDocs(data));
+                    dispatch(
+                      setSelectedDocs(
+                        data?.find((d) => d.location.toLowerCase() === 'local'),
+                      ),
+                    );
+                  });
                   setProgress(
                     (progress) =>
                       progress && {