feat: create workout fitness model

models-server/models/fitness_model.py

@@ -252,6 +252,7 @@ class FitnessModel:
 
     @classmethod
     def load(cls):
+
         with open(FITNESS_MODEL_PATH, "rb") as f:
             fitness_model = pickle.load(f)
 

src/configs/config.ts

@@ -14,6 +14,7 @@ export interface Config {
     expiresIn: string;
   };
   saltRounds: number;
+  modelsServerUrl: string;
 }
 
 export const config: Config = {
@@ -28,4 +29,5 @@ export const config: Config = {
     expiresIn: Env.get("JWT_EXPIRES_IN").toString(),
   },
   saltRounds: Env.get("SALT_ROUNDS", 5).toNumber(),
+  modelsServerUrl: `${Env.get("MODELS_HOST", 'http://127.0.0.1').toString()}:${Env.get("MODELS_PORT", '3030').toString()}`,
 };

src/lib/models/fitness-model.ts

@@ -1,6 +1,59 @@
-import path from "path"
+import { FitnessGoal } from "@common/enums/fitness-goal.enum";
+import { FitnessLevel } from "@common/enums/fitness-level.enum";
+import { Gender } from "@common/enums/gender.enum";
+import { config } from "@configs/config";
 
-path.join(__dirname, ..."../../resources/models/fitness_model.pkl".split("/"));
-const modelPath = `${__dirname}/../../resources/models/fitness_model.pkl`
+const endpoint = '/fitness';
 
-export class FitnessModel {}
+export interface IFitnessPredictionItem {
+  bodyPart: string;
+  equipment: string;
+  name: string;
+  repetitions: number;
+  sets: number;
+  target: string;
+  type: string;
+  weights_or_duration: number;
+}
+
+// Fitness Workout Params
+export interface IFWParams {
+  home_or_gym: 0 | 1; // 0 for home, 1 for gym
+  level: FitnessLevel;
+  goal: FitnessGoal;
+  gender: Gender;
+  age: number;
+  feedback: boolean;
+  old_weight: number;
+  equipments: string[];
+}
+
+export class FitnessModel {
+  public static async predictWorkout(
+    params: IFWParams
+  ): Promise<IFitnessPredictionItem[][]> {
+    params.level = params.level.split(' ').map((s) => s.charAt(0).toUpperCase() + s.slice(1)).join(' ') as FitnessLevel;
+    params.goal = params.goal.split(' ').map((s) => s.charAt(0).toUpperCase() + s.slice(1)).join(' ') as FitnessGoal;
+    params.gender = params.gender.toUpperCase() as Gender;
+
+    const response = await fetch(
+      `${config.modelsServerUrl}${endpoint}`,
+      {
+        method: "POST",
+        headers: {
+          "Content-Type": "application/json",
+        },
+        body: JSON.stringify(params),
+      }
+    );
+
+    if (!response.ok) {
+      console.error(await response.text());
+      throw new Error("Failed to fetch data from the server");
+    }
+
+    return response.json().then((data) => {
+      return data.result;
+    });
+  }
+}