Merge pull request #943 from singnet/training

Training
singnet · Nov 19, 2024 · c61e5da · c61e5da
2 parents 8d99571 + 374ac46
commit c61e5da
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 7 deletions.
diff --git a/src/Redux/actionCreators/ServiceDetailsActions.js b/src/Redux/actionCreators/ServiceDetailsActions.js
@@ -3,7 +3,7 @@ import { getAPI, initializeAPIOptions } from "../../utility/API";
 import { fetchAuthenticatedUser } from "./UserActions";
 import { loaderActions } from "./";
 import { LoaderContent } from "../../utility/constants/LoaderContent";
-// import { cacheS3Url } from "../../utility/image";
+import { isEmpty } from "lodash";
 
 export const UPDATE_SERVICE_DETAILS = "UPDATE_SERVICE_DETAILS";
 export const RESET_SERVICE_DETAILS = "RESET_SERVICE_DETAILS";
@@ -87,3 +87,19 @@ export const fetchMeteringData =
     const usageData = await meteringAPI(token, orgId, serviceId, groupId, email);
     return dispatch(fetchMeteringDataSuccess(usageData));
   };
+
+export const getIsTrainingAvailable = (detailsTraining, isLoggedIn) => {
+  if (isEmpty(detailsTraining)) {
+    return false;
+  }
+
+  if (!Object.prototype.hasOwnProperty.call(detailsTraining, "trainingMethods")) {
+    return false;
+  }
+
+  return (
+    process.env.REACT_APP_TRAINING_ENABLE === "true" &&
+    Object.keys(detailsTraining.trainingMethods).length &&
+    isLoggedIn
+  );
+};
diff --git a/src/components/ServiceDetails/ExistingModel/ModelDetails/index.js b/src/components/ServiceDetails/ExistingModel/ModelDetails/index.js
@@ -80,7 +80,7 @@ const ModelDetails = ({ classes, openEditModel, model, address }) => {
           </div>
         </div>
         <div className={classes.actionButtons}>
-          <Button className={classes.testBtn} onClick={handleSetModel} disabled={!isInferenceAvailable}>
+          <Button className={classes.inferenceBtn} onClick={handleSetModel} disabled={!isInferenceAvailable}>
             <NearMeOutlinedIcon />
             <span>Inference</span>
           </Button>

diff --git a/src/components/ServiceDetails/ExistingModel/ModelDetails/styles.js b/src/components/ServiceDetails/ExistingModel/ModelDetails/styles.js
@@ -87,12 +87,16 @@ export const useStyles = (theme) => ({
     },
   },
   updateBtn: { color: theme.palette.text.darkShadedGray },
-  testBtn: {
+  inferenceBtn: {
     background: theme.palette.text.primary,
     color: theme.palette.text.white,
     "&:hover": {
       background: theme.palette.text.customHoverBlue,
     },
+    "&:disabled": {
+      backgroundColor: theme.palette.text.lightGray,
+      color: theme.palette.text.white,
+    },
   },
   deleteBtn: { color: theme.palette.text.redBtnText },
   deleteModalContent: {

diff --git a/src/components/ServiceDetails/index.js b/src/components/ServiceDetails/index.js
@@ -16,7 +16,11 @@ import AboutService from "./AboutService";
 import InstallAndRunService from "./InstallAndRunService";
 import NotificationBar, { notificationBarTypes } from "../common/NotificationBar";
 
-import { fetchTrainingModel, fetchServiceDetails } from "../../Redux/actionCreators/ServiceDetailsActions";
+import {
+  fetchTrainingModel,
+  fetchServiceDetails,
+  getIsTrainingAvailable,
+} from "../../Redux/actionCreators/ServiceDetailsActions";
 import {
   pricing as getPricing,
   serviceDetails as getServiceDetails,
@@ -43,7 +47,7 @@ const ServiceDetails = ({ classes }) => {
   const { orgId, serviceId, tabId } = useParams();
 
   const isLoggedIn = useSelector((state) => state.userReducer.login.isLoggedIn);
-  const training = useSelector((state) => state.serviceDetailsReducer.detailsTraining);
+  const detailsTraining = useSelector((state) => state.serviceDetailsReducer.detailsTraining);
   const service = useSelector((state) => getServiceDetails(state, orgId, serviceId));
   const groupInfo = useSelector((state) => getGroupInfo(state));
   const pricing = useSelector((state) => getPricing(state));
@@ -89,8 +93,7 @@ const ServiceDetails = ({ classes }) => {
     );
   }
 
-  const isTrainingAvailable =
-    process.env.REACT_APP_TRAINING_ENABLE === "true" && Object.keys(training).length && isLoggedIn;
+  const isTrainingAvailable = getIsTrainingAvailable(detailsTraining, isLoggedIn);
 
   const tabs = [
     {