Zipstack · nehabagdia · Mar 7, 2024 · Mar 6, 2024 · Mar 6, 2024 · Mar 6, 2024
diff --git a/frontend/src/components/custom-tools/document-manager/DocumentManager.css b/frontend/src/components/custom-tools/document-manager/DocumentManager.css
@@ -30,14 +30,6 @@
     overflow-y: auto;
 }
 
-.rpv-default-layout__sidebar-headers {
-    background-color: #F8FBFF;
-}
-
-.rpv-default-layout__toolbar {
-    background-color: #F8FBFF;
-}
-
 .rpv-default-layout__container {
     border: 0 !important;
 }
diff --git a/frontend/src/components/custom-tools/editable-text/EditableText.jsx b/frontend/src/components/custom-tools/editable-text/EditableText.jsx
@@ -25,7 +25,7 @@ function EditableText({
 
   useEffect(() => {
     setText(defaultText);
-  }, [defaultText]);
+  }, []);
 
   useEffect(() => {
     // Attach the event listener when the component mounts

diff --git a/frontend/src/components/custom-tools/manage-docs-modal/ManageDocsModal.jsx b/frontend/src/components/custom-tools/manage-docs-modal/ManageDocsModal.jsx
@@ -1,4 +1,8 @@
-import { DeleteOutlined, PlusOutlined } from "@ant-design/icons";
+import {
+  DeleteOutlined,
+  PlusOutlined,
+  ReloadOutlined,
+} from "@ant-design/icons";
 import {
   Button,
   Divider,
@@ -49,6 +53,12 @@ function ManageDocsModal({
       dataIndex: "document",
       key: "document",
     },
+    {
+      title: "",
+      dataIndex: "reindex",
+      key: "reindex",
+      width: 30,
+    },
     {
       title: "",
       dataIndex: "delete",
@@ -68,6 +78,17 @@ function ManageDocsModal({
       return {
         key: doc,
         document: doc || "",
+        reindex: (
+          <Tooltip title="Re-Index">
+            <Button
+              size="small"
+              className="display-flex-align-center"
+              onClick={() => generateIndex(doc)}
+            >
+              <ReloadOutlined className="manage-llm-pro-icon" />
+            </Button>
+          </Tooltip>
+        ),
         delete: (
           <ConfirmModal
             handleConfirm={() => handleDelete(doc)}

diff --git a/frontend/src/components/custom-tools/output-analyzer-card/OutputAnalyzerCard.jsx b/frontend/src/components/custom-tools/output-analyzer-card/OutputAnalyzerCard.jsx
@@ -78,26 +78,24 @@ function OutputAnalyzerCard({ doc, totalFields }) {
         </Flex>
       </div>
       <div className="output-analyzer-main">
-        {fileUrl?.length > 0 && (
-          <Row className="height-100">
-            <Col span={12} className="height-100">
-              <div className="output-analyzer-left-box">
-                <DocumentViewer
-                  doc={doc}
-                  isLoading={isDocLoading}
-                  isContentAvailable={fileUrl?.length > 0}
-                >
-                  <PdfViewer fileUrl={fileUrl} />
-                </DocumentViewer>
-              </div>
-            </Col>
-            <Col span={12} className="height-100">
-              <div className="output-analyzer-right-box">
-                <CombinedOutput doc={doc} setFilledFields={setFilledFields} />
-              </div>
-            </Col>
-          </Row>
-        )}
+        <Row className="height-100">
+          <Col span={12} className="height-100">
+            <div className="output-analyzer-left-box">
+              <DocumentViewer
+                doc={doc}
+                isLoading={isDocLoading}
+                isContentAvailable={fileUrl?.length > 0}
+              >
+                <PdfViewer fileUrl={fileUrl} />
+              </DocumentViewer>
+            </div>
+          </Col>
+          <Col span={12} className="height-100">
+            <div className="output-analyzer-right-box">
+              <CombinedOutput doc={doc} setFilledFields={setFilledFields} />
+            </div>
+          </Col>
+        </Row>
       </div>
     </div>
   );

diff --git a/frontend/src/components/custom-tools/output-analyzer-header/OutputAnalyzerHeader.jsx b/frontend/src/components/custom-tools/output-analyzer-header/OutputAnalyzerHeader.jsx
@@ -20,7 +20,7 @@ function OutputAnalyzerHeader() {
             <ArrowLeftOutlined />
           </Button>
           <Typography.Text className="font-size-16" strong>
-            Output Viewer
+            Output Analyzer
           </Typography.Text>
         </Space>
       </div>

diff --git a/frontend/src/components/custom-tools/select-llm-profile-modal/SelectLlmProfileModal.jsx b/frontend/src/components/custom-tools/select-llm-profile-modal/SelectLlmProfileModal.jsx
@@ -36,7 +36,8 @@ function SelectLlmProfileModal({
   const [isContext, setIsContext] = useState(false);
   const [isSource, setIsSource] = useState(false);
   const [isModified, setIsModified] = useState(false);
-  const { details } = useCustomToolStore();
+  const [isLoading, setIsLoading] = useState(false);
+  const { details, updateCustomTool } = useCustomToolStore();
   const { setAlertDetails } = useAlertStore();
 
   useEffect(() => {
@@ -94,19 +95,31 @@ function SelectLlmProfileModal({
       [fieldNames.SUMMARIZE_AS_SOURCE]: isSource,
     };
 
+    setIsLoading(true);
     handleUpdateTool(body)
       .then(() => {
         handleBtnText(selectedLlm);
+        const updatedDetails = { ...details };
+        updatedDetails["summarize_llm_profile"] = selectedLlm;
+        updatedDetails["summarize_prompt"] = prompt;
+        updatedDetails["summarize_context"] = isContext;
+        updatedDetails["summarize_as_source"] = isSource;
+        updateCustomTool({ details: updatedDetails });
+
         setAlertDetails({
           type: "success",
           content: "Successfully updated the LLM profile",
         });
         setIsModified(false);
+        setOpen(false);
       })
       .catch((err) => {
         setAlertDetails(
           handleException(err, "Failed to update the LLM profile")
         );
+      })
+      .finally(() => {
+        setIsLoading(false);
       });
   };
 
@@ -192,6 +205,7 @@ function SelectLlmProfileModal({
                 type="primary"
                 onClick={handleSave}
                 disabled={!isModified}
+                loading={isLoading}
               >
                 Save
               </CustomButton>