fix: upon successful training of newly uploaded src doc, the latest doc is auto set as selected doc

2 weeks ago · 45145685d5
parent 2fbec6f21f
commit 45145685d5
1 changed files with 16 additions and 2 deletions
--- 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 && {