Deploying Spring Boot

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+/mvnw text eol=lf
+*.cmd text eol=crlf

.mvn/wrapper/maven-wrapper.properties

@@ -0,0 +1,3 @@
+wrapperVersion=3.3.4
+distributionType=only-script
+distributionUrl=https://repo.maven.apache.org/maven2/org/apache/maven/apache-maven/3.9.11/apache-maven-3.9.11-bin.zip

Dockerfile

@@ -0,0 +1,17 @@
+FROM maven:3.9.6-eclipse-temurin-17 AS build
+WORKDIR /app
+COPY pom.xml .
+COPY src ./src
+RUN mvn clean package -DskipTests
+
+
+FROM eclipse-temurin:17-jre-alpine
+WORKDIR /app
+
+
+COPY --from=build /app/target/*.jar app.jar
+
+EXPOSE 7860
+
+# Run the jar
+ENTRYPOINT ["java", "-jar", "app.jar"]

mvnw

@@ -0,0 +1,295 @@
+#!/bin/sh
+# ----------------------------------------------------------------------------
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+# ----------------------------------------------------------------------------
+
+# ----------------------------------------------------------------------------
+# Apache Maven Wrapper startup batch script, version 3.3.4
+#
+# Optional ENV vars
+# -----------------
+#   JAVA_HOME - location of a JDK home dir, required when download maven via java source
+#   MVNW_REPOURL - repo url base for downloading maven distribution
+#   MVNW_USERNAME/MVNW_PASSWORD - user and password for downloading maven
+#   MVNW_VERBOSE - true: enable verbose log; debug: trace the mvnw script; others: silence the output
+# ----------------------------------------------------------------------------
+
+set -euf
+[ "${MVNW_VERBOSE-}" != debug ] || set -x
+
+# OS specific support.
+native_path() { printf %s\\n "$1"; }
+case "$(uname)" in
+CYGWIN* | MINGW*)
+  [ -z "${JAVA_HOME-}" ] || JAVA_HOME="$(cygpath --unix "$JAVA_HOME")"
+  native_path() { cygpath --path --windows "$1"; }
+  ;;
+esac
+
+# set JAVACMD and JAVACCMD
+set_java_home() {
+  # For Cygwin and MinGW, ensure paths are in Unix format before anything is touched
+  if [ -n "${JAVA_HOME-}" ]; then
+    if [ -x "$JAVA_HOME/jre/sh/java" ]; then
+      # IBM's JDK on AIX uses strange locations for the executables
+      JAVACMD="$JAVA_HOME/jre/sh/java"
+      JAVACCMD="$JAVA_HOME/jre/sh/javac"
+    else
+      JAVACMD="$JAVA_HOME/bin/java"
+      JAVACCMD="$JAVA_HOME/bin/javac"
+
+      if [ ! -x "$JAVACMD" ] || [ ! -x "$JAVACCMD" ]; then
+        echo "The JAVA_HOME environment variable is not defined correctly, so mvnw cannot run." >&2
+        echo "JAVA_HOME is set to \"$JAVA_HOME\", but \"\$JAVA_HOME/bin/java\" or \"\$JAVA_HOME/bin/javac\" does not exist." >&2
+        return 1
+      fi
+    fi
+  else
+    JAVACMD="$(
+      'set' +e
+      'unset' -f command 2>/dev/null
+      'command' -v java
+    )" || :
+    JAVACCMD="$(
+      'set' +e
+      'unset' -f command 2>/dev/null
+      'command' -v javac
+    )" || :
+
+    if [ ! -x "${JAVACMD-}" ] || [ ! -x "${JAVACCMD-}" ]; then
+      echo "The java/javac command does not exist in PATH nor is JAVA_HOME set, so mvnw cannot run." >&2
+      return 1
+    fi
+  fi
+}
+
+# hash string like Java String::hashCode
+hash_string() {
+  str="${1:-}" h=0
+  while [ -n "$str" ]; do
+    char="${str%"${str#?}"}"
+    h=$(((h * 31 + $(LC_CTYPE=C printf %d "'$char")) % 4294967296))
+    str="${str#?}"
+  done
+  printf %x\\n $h
+}
+
+verbose() { :; }
+[ "${MVNW_VERBOSE-}" != true ] || verbose() { printf %s\\n "${1-}"; }
+
+die() {
+  printf %s\\n "$1" >&2
+  exit 1
+}
+
+trim() {
+  # MWRAPPER-139:
+  #   Trims trailing and leading whitespace, carriage returns, tabs, and linefeeds.
+  #   Needed for removing poorly interpreted newline sequences when running in more
+  #   exotic environments such as mingw bash on Windows.
+  printf "%s" "${1}" | tr -d '[:space:]'
+}
+
+scriptDir="$(dirname "$0")"
+scriptName="$(basename "$0")"
+
+# parse distributionUrl and optional distributionSha256Sum, requires .mvn/wrapper/maven-wrapper.properties
+while IFS="=" read -r key value; do
+  case "${key-}" in
+  distributionUrl) distributionUrl=$(trim "${value-}") ;;
+  distributionSha256Sum) distributionSha256Sum=$(trim "${value-}") ;;
+  esac
+done <"$scriptDir/.mvn/wrapper/maven-wrapper.properties"
+[ -n "${distributionUrl-}" ] || die "cannot read distributionUrl property in $scriptDir/.mvn/wrapper/maven-wrapper.properties"
+
+case "${distributionUrl##*/}" in
+maven-mvnd-*bin.*)
+  MVN_CMD=mvnd.sh _MVNW_REPO_PATTERN=/maven/mvnd/
+  case "${PROCESSOR_ARCHITECTURE-}${PROCESSOR_ARCHITEW6432-}:$(uname -a)" in
+  *AMD64:CYGWIN* | *AMD64:MINGW*) distributionPlatform=windows-amd64 ;;
+  :Darwin*x86_64) distributionPlatform=darwin-amd64 ;;
+  :Darwin*arm64) distributionPlatform=darwin-aarch64 ;;
+  :Linux*x86_64*) distributionPlatform=linux-amd64 ;;
+  *)
+    echo "Cannot detect native platform for mvnd on $(uname)-$(uname -m), use pure java version" >&2
+    distributionPlatform=linux-amd64
+    ;;
+  esac
+  distributionUrl="${distributionUrl%-bin.*}-$distributionPlatform.zip"
+  ;;
+maven-mvnd-*) MVN_CMD=mvnd.sh _MVNW_REPO_PATTERN=/maven/mvnd/ ;;
+*) MVN_CMD="mvn${scriptName#mvnw}" _MVNW_REPO_PATTERN=/org/apache/maven/ ;;
+esac
+
+# apply MVNW_REPOURL and calculate MAVEN_HOME
+# maven home pattern: ~/.m2/wrapper/dists/{apache-maven-<version>,maven-mvnd-<version>-<platform>}/<hash>
+[ -z "${MVNW_REPOURL-}" ] || distributionUrl="$MVNW_REPOURL$_MVNW_REPO_PATTERN${distributionUrl#*"$_MVNW_REPO_PATTERN"}"
+distributionUrlName="${distributionUrl##*/}"
+distributionUrlNameMain="${distributionUrlName%.*}"
+distributionUrlNameMain="${distributionUrlNameMain%-bin}"
+MAVEN_USER_HOME="${MAVEN_USER_HOME:-${HOME}/.m2}"
+MAVEN_HOME="${MAVEN_USER_HOME}/wrapper/dists/${distributionUrlNameMain-}/$(hash_string "$distributionUrl")"
+
+exec_maven() {
+  unset MVNW_VERBOSE MVNW_USERNAME MVNW_PASSWORD MVNW_REPOURL || :
+  exec "$MAVEN_HOME/bin/$MVN_CMD" "$@" || die "cannot exec $MAVEN_HOME/bin/$MVN_CMD"
+}
+
+if [ -d "$MAVEN_HOME" ]; then
+  verbose "found existing MAVEN_HOME at $MAVEN_HOME"
+  exec_maven "$@"
+fi
+
+case "${distributionUrl-}" in
+*?-bin.zip | *?maven-mvnd-?*-?*.zip) ;;
+*) die "distributionUrl is not valid, must match *-bin.zip or maven-mvnd-*.zip, but found '${distributionUrl-}'" ;;
+esac
+
+# prepare tmp dir
+if TMP_DOWNLOAD_DIR="$(mktemp -d)" && [ -d "$TMP_DOWNLOAD_DIR" ]; then
+  clean() { rm -rf -- "$TMP_DOWNLOAD_DIR"; }
+  trap clean HUP INT TERM EXIT
+else
+  die "cannot create temp dir"
+fi
+
+mkdir -p -- "${MAVEN_HOME%/*}"
+
+# Download and Install Apache Maven
+verbose "Couldn't find MAVEN_HOME, downloading and installing it ..."
+verbose "Downloading from: $distributionUrl"
+verbose "Downloading to: $TMP_DOWNLOAD_DIR/$distributionUrlName"
+
+# select .zip or .tar.gz
+if ! command -v unzip >/dev/null; then
+  distributionUrl="${distributionUrl%.zip}.tar.gz"
+  distributionUrlName="${distributionUrl##*/}"
+fi
+
+# verbose opt
+__MVNW_QUIET_WGET=--quiet __MVNW_QUIET_CURL=--silent __MVNW_QUIET_UNZIP=-q __MVNW_QUIET_TAR=''
+[ "${MVNW_VERBOSE-}" != true ] || __MVNW_QUIET_WGET='' __MVNW_QUIET_CURL='' __MVNW_QUIET_UNZIP='' __MVNW_QUIET_TAR=v
+
+# normalize http auth
+case "${MVNW_PASSWORD:+has-password}" in
+'') MVNW_USERNAME='' MVNW_PASSWORD='' ;;
+has-password) [ -n "${MVNW_USERNAME-}" ] || MVNW_USERNAME='' MVNW_PASSWORD='' ;;
+esac
+
+if [ -z "${MVNW_USERNAME-}" ] && command -v wget >/dev/null; then
+  verbose "Found wget ... using wget"
+  wget ${__MVNW_QUIET_WGET:+"$__MVNW_QUIET_WGET"} "$distributionUrl" -O "$TMP_DOWNLOAD_DIR/$distributionUrlName" || die "wget: Failed to fetch $distributionUrl"
+elif [ -z "${MVNW_USERNAME-}" ] && command -v curl >/dev/null; then
+  verbose "Found curl ... using curl"
+  curl ${__MVNW_QUIET_CURL:+"$__MVNW_QUIET_CURL"} -f -L -o "$TMP_DOWNLOAD_DIR/$distributionUrlName" "$distributionUrl" || die "curl: Failed to fetch $distributionUrl"
+elif set_java_home; then
+  verbose "Falling back to use Java to download"
+  javaSource="$TMP_DOWNLOAD_DIR/Downloader.java"
+  targetZip="$TMP_DOWNLOAD_DIR/$distributionUrlName"
+  cat >"$javaSource" <<-END
+	public class Downloader extends java.net.Authenticator
+	{
+	  protected java.net.PasswordAuthentication getPasswordAuthentication()
+	  {
+	    return new java.net.PasswordAuthentication( System.getenv( "MVNW_USERNAME" ), System.getenv( "MVNW_PASSWORD" ).toCharArray() );
+	  }
+	  public static void main( String[] args ) throws Exception
+	  {
+	    setDefault( new Downloader() );
+	    java.nio.file.Files.copy( java.net.URI.create( args[0] ).toURL().openStream(), java.nio.file.Paths.get( args[1] ).toAbsolutePath().normalize() );
+	  }
+	}
+	END
+  # For Cygwin/MinGW, switch paths to Windows format before running javac and java
+  verbose " - Compiling Downloader.java ..."
+  "$(native_path "$JAVACCMD")" "$(native_path "$javaSource")" || die "Failed to compile Downloader.java"
+  verbose " - Running Downloader.java ..."
+  "$(native_path "$JAVACMD")" -cp "$(native_path "$TMP_DOWNLOAD_DIR")" Downloader "$distributionUrl" "$(native_path "$targetZip")"
+fi
+
+# If specified, validate the SHA-256 sum of the Maven distribution zip file
+if [ -n "${distributionSha256Sum-}" ]; then
+  distributionSha256Result=false
+  if [ "$MVN_CMD" = mvnd.sh ]; then
+    echo "Checksum validation is not supported for maven-mvnd." >&2
+    echo "Please disable validation by removing 'distributionSha256Sum' from your maven-wrapper.properties." >&2
+    exit 1
+  elif command -v sha256sum >/dev/null; then
+    if echo "$distributionSha256Sum  $TMP_DOWNLOAD_DIR/$distributionUrlName" | sha256sum -c - >/dev/null 2>&1; then
+      distributionSha256Result=true
+    fi
+  elif command -v shasum >/dev/null; then
+    if echo "$distributionSha256Sum  $TMP_DOWNLOAD_DIR/$distributionUrlName" | shasum -a 256 -c >/dev/null 2>&1; then
+      distributionSha256Result=true
+    fi
+  else
+    echo "Checksum validation was requested but neither 'sha256sum' or 'shasum' are available." >&2
+    echo "Please install either command, or disable validation by removing 'distributionSha256Sum' from your maven-wrapper.properties." >&2
+    exit 1
+  fi
+  if [ $distributionSha256Result = false ]; then
+    echo "Error: Failed to validate Maven distribution SHA-256, your Maven distribution might be compromised." >&2
+    echo "If you updated your Maven version, you need to update the specified distributionSha256Sum property." >&2
+    exit 1
+  fi
+fi
+
+# unzip and move
+if command -v unzip >/dev/null; then
+  unzip ${__MVNW_QUIET_UNZIP:+"$__MVNW_QUIET_UNZIP"} "$TMP_DOWNLOAD_DIR/$distributionUrlName" -d "$TMP_DOWNLOAD_DIR" || die "failed to unzip"
+else
+  tar xzf${__MVNW_QUIET_TAR:+"$__MVNW_QUIET_TAR"} "$TMP_DOWNLOAD_DIR/$distributionUrlName" -C "$TMP_DOWNLOAD_DIR" || die "failed to untar"
+fi
+
+# Find the actual extracted directory name (handles snapshots where filename != directory name)
+actualDistributionDir=""
+
+# First try the expected directory name (for regular distributions)
+if [ -d "$TMP_DOWNLOAD_DIR/$distributionUrlNameMain" ]; then
+  if [ -f "$TMP_DOWNLOAD_DIR/$distributionUrlNameMain/bin/$MVN_CMD" ]; then
+    actualDistributionDir="$distributionUrlNameMain"
+  fi
+fi
+
+# If not found, search for any directory with the Maven executable (for snapshots)
+if [ -z "$actualDistributionDir" ]; then
+  # enable globbing to iterate over items
+  set +f
+  for dir in "$TMP_DOWNLOAD_DIR"/*; do
+    if [ -d "$dir" ]; then
+      if [ -f "$dir/bin/$MVN_CMD" ]; then
+        actualDistributionDir="$(basename "$dir")"
+        break
+      fi
+    fi
+  done
+  set -f
+fi
+
+if [ -z "$actualDistributionDir" ]; then
+  verbose "Contents of $TMP_DOWNLOAD_DIR:"
+  verbose "$(ls -la "$TMP_DOWNLOAD_DIR")"
+  die "Could not find Maven distribution directory in extracted archive"
+fi
+
+verbose "Found extracted Maven distribution directory: $actualDistributionDir"
+printf %s\\n "$distributionUrl" >"$TMP_DOWNLOAD_DIR/$actualDistributionDir/mvnw.url"
+mv -- "$TMP_DOWNLOAD_DIR/$actualDistributionDir" "$MAVEN_HOME" || [ -d "$MAVEN_HOME" ] || die "fail to move MAVEN_HOME"
+
+clean || :
+exec_maven "$@"

mvnw.cmd

@@ -0,0 +1,189 @@
+<# : batch portion
+@REM ----------------------------------------------------------------------------
+@REM Licensed to the Apache Software Foundation (ASF) under one
+@REM or more contributor license agreements.  See the NOTICE file
+@REM distributed with this work for additional information
+@REM regarding copyright ownership.  The ASF licenses this file
+@REM to you under the Apache License, Version 2.0 (the
+@REM "License"); you may not use this file except in compliance
+@REM with the License.  You may obtain a copy of the License at
+@REM
+@REM    http://www.apache.org/licenses/LICENSE-2.0
+@REM
+@REM Unless required by applicable law or agreed to in writing,
+@REM software distributed under the License is distributed on an
+@REM "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+@REM KIND, either express or implied.  See the License for the
+@REM specific language governing permissions and limitations
+@REM under the License.
+@REM ----------------------------------------------------------------------------
+
+@REM ----------------------------------------------------------------------------
+@REM Apache Maven Wrapper startup batch script, version 3.3.4
+@REM
+@REM Optional ENV vars
+@REM   MVNW_REPOURL - repo url base for downloading maven distribution
+@REM   MVNW_USERNAME/MVNW_PASSWORD - userDTO and password for downloading maven
+@REM   MVNW_VERBOSE - true: enable verbose log; others: silence the output
+@REM ----------------------------------------------------------------------------
+
+@IF "%__MVNW_ARG0_NAME__%"=="" (SET __MVNW_ARG0_NAME__=%~nx0)
+@SET __MVNW_CMD__=
+@SET __MVNW_ERROR__=
+@SET __MVNW_PSMODULEP_SAVE=%PSModulePath%
+@SET PSModulePath=
+@FOR /F "usebackq tokens=1* delims==" %%A IN (`powershell -noprofile "& {$scriptDir='%~dp0'; $script='%__MVNW_ARG0_NAME__%'; icm -ScriptBlock ([Scriptblock]::Create((Get-Content -Raw '%~f0'))) -NoNewScope}"`) DO @(
+  IF "%%A"=="MVN_CMD" (set __MVNW_CMD__=%%B) ELSE IF "%%B"=="" (echo %%A) ELSE (echo %%A=%%B)
+)
+@SET PSModulePath=%__MVNW_PSMODULEP_SAVE%
+@SET __MVNW_PSMODULEP_SAVE=
+@SET __MVNW_ARG0_NAME__=
+@SET MVNW_USERNAME=
+@SET MVNW_PASSWORD=
+@IF NOT "%__MVNW_CMD__%"=="" ("%__MVNW_CMD__%" %*)
+@echo Cannot start maven from wrapper >&2 && exit /b 1
+@GOTO :EOF
+: end batch / begin powershell #>
+
+$ErrorActionPreference = "Stop"
+if ($env:MVNW_VERBOSE -eq "true") {
+  $VerbosePreference = "Continue"
+}
+
+# calculate distributionUrl, requires .mvn/wrapper/maven-wrapper.properties
+$distributionUrl = (Get-Content -Raw "$scriptDir/.mvn/wrapper/maven-wrapper.properties" | ConvertFrom-StringData).distributionUrl
+if (!$distributionUrl) {
+  Write-Error "cannot read distributionUrl property in $scriptDir/.mvn/wrapper/maven-wrapper.properties"
+}
+
+switch -wildcard -casesensitive ( $($distributionUrl -replace '^.*/','') ) {
+  "maven-mvnd-*" {
+    $USE_MVND = $true
+    $distributionUrl = $distributionUrl -replace '-bin\.[^.]*$',"-windows-amd64.zip"
+    $MVN_CMD = "mvnd.cmd"
+    break
+  }
+  default {
+    $USE_MVND = $false
+    $MVN_CMD = $script -replace '^mvnw','mvn'
+    break
+  }
+}
+
+# apply MVNW_REPOURL and calculate MAVEN_HOME
+# maven home pattern: ~/.m2/wrapper/dists/{apache-maven-<version>,maven-mvnd-<version>-<platform>}/<hash>
+if ($env:MVNW_REPOURL) {
+  $MVNW_REPO_PATTERN = if ($USE_MVND -eq $False) { "/org/apache/maven/" } else { "/maven/mvnd/" }
+  $distributionUrl = "$env:MVNW_REPOURL$MVNW_REPO_PATTERN$($distributionUrl -replace "^.*$MVNW_REPO_PATTERN",'')"
+}
+$distributionUrlName = $distributionUrl -replace '^.*/',''
+$distributionUrlNameMain = $distributionUrlName -replace '\.[^.]*$','' -replace '-bin$',''
+
+$MAVEN_M2_PATH = "$HOME/.m2"
+if ($env:MAVEN_USER_HOME) {
+  $MAVEN_M2_PATH = "$env:MAVEN_USER_HOME"
+}
+
+if (-not (Test-Path -Path $MAVEN_M2_PATH)) {
+    New-Item -Path $MAVEN_M2_PATH -ItemType Directory | Out-Null
+}
+
+$MAVEN_WRAPPER_DISTS = $null
+if ((Get-Item $MAVEN_M2_PATH).Target[0] -eq $null) {
+  $MAVEN_WRAPPER_DISTS = "$MAVEN_M2_PATH/wrapper/dists"
+} else {
+  $MAVEN_WRAPPER_DISTS = (Get-Item $MAVEN_M2_PATH).Target[0] + "/wrapper/dists"
+}
+
+$MAVEN_HOME_PARENT = "$MAVEN_WRAPPER_DISTS/$distributionUrlNameMain"
+$MAVEN_HOME_NAME = ([System.Security.Cryptography.SHA256]::Create().ComputeHash([byte[]][char[]]$distributionUrl) | ForEach-Object {$_.ToString("x2")}) -join ''
+$MAVEN_HOME = "$MAVEN_HOME_PARENT/$MAVEN_HOME_NAME"
+
+if (Test-Path -Path "$MAVEN_HOME" -PathType Container) {
+  Write-Verbose "found existing MAVEN_HOME at $MAVEN_HOME"
+  Write-Output "MVN_CMD=$MAVEN_HOME/bin/$MVN_CMD"
+  exit $?
+}
+
+if (! $distributionUrlNameMain -or ($distributionUrlName -eq $distributionUrlNameMain)) {
+  Write-Error "distributionUrl is not valid, must end with *-bin.zip, but found $distributionUrl"
+}
+
+# prepare tmp dir
+$TMP_DOWNLOAD_DIR_HOLDER = New-TemporaryFile
+$TMP_DOWNLOAD_DIR = New-Item -Itemtype Directory -Path "$TMP_DOWNLOAD_DIR_HOLDER.dir"
+$TMP_DOWNLOAD_DIR_HOLDER.Delete() | Out-Null
+trap {
+  if ($TMP_DOWNLOAD_DIR.Exists) {
+    try { Remove-Item $TMP_DOWNLOAD_DIR -Recurse -Force | Out-Null }
+    catch { Write-Warning "Cannot remove $TMP_DOWNLOAD_DIR" }
+  }
+}
+
+New-Item -Itemtype Directory -Path "$MAVEN_HOME_PARENT" -Force | Out-Null
+
+# Download and Install Apache Maven
+Write-Verbose "Couldn't find MAVEN_HOME, downloading and installing it ..."
+Write-Verbose "Downloading from: $distributionUrl"
+Write-Verbose "Downloading to: $TMP_DOWNLOAD_DIR/$distributionUrlName"
+
+$webclient = New-Object System.Net.WebClient
+if ($env:MVNW_USERNAME -and $env:MVNW_PASSWORD) {
+  $webclient.Credentials = New-Object System.Net.NetworkCredential($env:MVNW_USERNAME, $env:MVNW_PASSWORD)
+}
+[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12
+$webclient.DownloadFile($distributionUrl, "$TMP_DOWNLOAD_DIR/$distributionUrlName") | Out-Null
+
+# If specified, validate the SHA-256 sum of the Maven distribution zip file
+$distributionSha256Sum = (Get-Content -Raw "$scriptDir/.mvn/wrapper/maven-wrapper.properties" | ConvertFrom-StringData).distributionSha256Sum
+if ($distributionSha256Sum) {
+  if ($USE_MVND) {
+    Write-Error "Checksum validation is not supported for maven-mvnd. `nPlease disable validation by removing 'distributionSha256Sum' from your maven-wrapper.properties."
+  }
+  Import-Module $PSHOME\Modules\Microsoft.PowerShell.Utility -Function Get-FileHash
+  if ((Get-FileHash "$TMP_DOWNLOAD_DIR/$distributionUrlName" -Algorithm SHA256).Hash.ToLower() -ne $distributionSha256Sum) {
+    Write-Error "Error: Failed to validate Maven distribution SHA-256, your Maven distribution might be compromised. If you updated your Maven version, you need to update the specified distributionSha256Sum property."
+  }
+}
+
+# unzip and move
+Expand-Archive "$TMP_DOWNLOAD_DIR/$distributionUrlName" -DestinationPath "$TMP_DOWNLOAD_DIR" | Out-Null
+
+# Find the actual extracted directory name (handles snapshots where filename != directory name)
+$actualDistributionDir = ""
+
+# First try the expected directory name (for regular distributions)
+$expectedPath = Join-Path "$TMP_DOWNLOAD_DIR" "$distributionUrlNameMain"
+$expectedMvnPath = Join-Path "$expectedPath" "bin/$MVN_CMD"
+if ((Test-Path -Path $expectedPath -PathType Container) -and (Test-Path -Path $expectedMvnPath -PathType Leaf)) {
+  $actualDistributionDir = $distributionUrlNameMain
+}
+
+# If not found, search for any directory with the Maven executable (for snapshots)
+if (!$actualDistributionDir) {
+  Get-ChildItem -Path "$TMP_DOWNLOAD_DIR" -Directory | ForEach-Object {
+    $testPath = Join-Path $_.FullName "bin/$MVN_CMD"
+    if (Test-Path -Path $testPath -PathType Leaf) {
+      $actualDistributionDir = $_.Name
+    }
+  }
+}
+
+if (!$actualDistributionDir) {
+  Write-Error "Could not find Maven distribution directory in extracted archive"
+}
+
+Write-Verbose "Found extracted Maven distribution directory: $actualDistributionDir"
+Rename-Item -Path "$TMP_DOWNLOAD_DIR/$actualDistributionDir" -NewName $MAVEN_HOME_NAME | Out-Null
+try {
+  Move-Item -Path "$TMP_DOWNLOAD_DIR/$MAVEN_HOME_NAME" -Destination $MAVEN_HOME_PARENT | Out-Null
+} catch {
+  if (! (Test-Path -Path "$MAVEN_HOME" -PathType Container)) {
+    Write-Error "fail to move MAVEN_HOME"
+  }
+} finally {
+  try { Remove-Item $TMP_DOWNLOAD_DIR -Recurse -Force | Out-Null }
+  catch { Write-Warning "Cannot remove $TMP_DOWNLOAD_DIR" }
+}
+
+Write-Output "MVN_CMD=$MAVEN_HOME/bin/$MVN_CMD"

pom.xml

@@ -0,0 +1,167 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+    <parent>
+        <groupId>org.springframework.boot</groupId>
+        <artifactId>spring-boot-starter-parent</artifactId>
+        <version>3.5.6</version>
+        <relativePath/> <!-- lookup parent from repository -->
+    </parent>
+    <groupId>edu.alexu</groupId>
+    <artifactId>fitfinder</artifactId>
+    <version>0.0.1-SNAPSHOT</version>
+    <name>FItFinder</name>
+    <description>Demo project for Spring Boot</description>
+    <url/>
+    <licenses>
+        <license/>
+    </licenses>
+    <developers>
+        <developer/>
+    </developers>
+    <scm>
+        <connection/>
+        <developerConnection/>
+        <tag/>
+        <url/>
+    </scm>
+    <properties>
+        <maven.compiler.source>17</maven.compiler.source>
+        <maven.compiler.target>17</maven.compiler.target>
+        <maven.compiler.release>17</maven.compiler.release>
+        <java.version>21</java.version>
+    </properties>
+
+    <dependencies>
+
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-actuator</artifactId>
+        </dependency>
+
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-web</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-devtools</artifactId>
+            <scope>runtime</scope>
+            <optional>true</optional>
+        </dependency>
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+            <optional>true</optional>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-test</artifactId>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.postgresql</groupId>
+            <artifactId>postgresql</artifactId>
+            <scope>runtime</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-data-jpa</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>me.paulschwarz</groupId>
+            <artifactId>spring-dotenv</artifactId>
+            <version>4.0.0</version>
+        </dependency>
+        <dependency>
+            <groupId>org.mindrot</groupId>
+            <artifactId>jbcrypt</artifactId>
+            <version>0.4</version>
+        </dependency>
+        <dependency>
+            <groupId>commons-validator</groupId>
+            <artifactId>commons-validator</artifactId>
+            <version>1.10.0</version>
+        </dependency>
+        <dependency>
+            <groupId>io.jsonwebtoken</groupId>
+            <artifactId>jjwt-api</artifactId>
+            <version>0.11.5</version>
+        </dependency>
+        <dependency>
+            <groupId>io.jsonwebtoken</groupId>
+            <artifactId>jjwt-impl</artifactId>
+            <version>0.11.5</version>
+        </dependency>
+        <dependency>
+            <groupId>io.jsonwebtoken</groupId>
+            <artifactId>jjwt-jackson</artifactId>
+            <version>0.11.5</version>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-security</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>com.cloudinary</groupId>
+            <artifactId>cloudinary-http5</artifactId>
+            <version>2.0.0</version>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-webflux</artifactId>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-websocket</artifactId>
+        </dependency>
+    </dependencies>
+
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <configuration>
+                    <annotationProcessorPaths>
+                        <path>
+                            <groupId>org.projectlombok</groupId>
+                            <artifactId>lombok</artifactId>
+                        </path>
+                    </annotationProcessorPaths>
+                </configuration>
+            </plugin>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <version>3.14.0</version>
+                <configuration>
+                    <source>17</source>
+                    <target>17</target>
+                    <release>17</release>
+                    <annotationProcessorPaths>
+                        <path>
+                            <groupId>org.projectlombok</groupId>
+                            <artifactId>lombok</artifactId>
+                            <version>1.18.34</version>
+                        </path>
+                    </annotationProcessorPaths>
+                </configuration>
+            </plugin>
+            <plugin>
+                <groupId>org.springframework.boot</groupId>
+                <artifactId>spring-boot-maven-plugin</artifactId>
+                <configuration>
+                    <excludes>
+                        <exclude>
+                            <groupId>org.projectlombok</groupId>
+                            <artifactId>lombok</artifactId>
+                        </exclude>
+                    </excludes>
+                </configuration>
+            </plugin>
+        </plugins>
+    </build>
+
+</project>

src/main/java/edu/alexu/fitfinder/FitFinderApplication.java

@@ -0,0 +1,12 @@
+package edu.alexu.fitfinder;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+
+@SpringBootApplication
+
+public class FitFinderApplication {
+  public static void main(String[] args) {
+    SpringApplication.run(FitFinderApplication.class, args);
+  }
+}

src/main/java/edu/alexu/fitfinder/component/JobRegistry.java

@@ -0,0 +1,26 @@
+package edu.alexu.fitfinder.component;
+
+import org.springframework.stereotype.Component;
+import org.springframework.web.socket.WebSocketSession;
+import java.util.concurrent.ConcurrentHashMap;
+
+@Component
+public class JobRegistry {
+
+  private final ConcurrentHashMap<String, WebSocketSession> jobToSession =
+      new ConcurrentHashMap<>();
+
+  public void registerJob(String jobId, WebSocketSession session) {
+    System.out.println("Job " + jobId + " has been registered with session " + session.getId());
+    jobToSession.put(jobId, session);
+  }
+
+  public WebSocketSession getSession(String jobId) {
+    return jobToSession.get(jobId);
+  }
+
+  public void removeJob(String jobId) {
+    System.out.println("The websocket session is no longer registered with job " + jobId);
+    jobToSession.remove(jobId);
+  }
+}

src/main/java/edu/alexu/fitfinder/configs/CloudinaryConfig.java

@@ -0,0 +1,18 @@
+package edu.alexu.fitfinder.configs;
+
+import com.cloudinary.Cloudinary;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+public class CloudinaryConfig {
+
+    @Value("${cloudinary.url}")
+    private String cloudinaryUrl;
+
+    @Bean
+    public Cloudinary cloudinary() {
+        return new Cloudinary(cloudinaryUrl);
+    }
+}

src/main/java/edu/alexu/fitfinder/configs/JwtAuthenticationFilter.java

@@ -0,0 +1,60 @@
+package edu.alexu.fitfinder.configs;
+
+import jakarta.servlet.FilterChain;
+import jakarta.servlet.ServletException;
+import jakarta.servlet.http.HttpServletRequest;
+import jakarta.servlet.http.HttpServletResponse;
+import lombok.AllArgsConstructor;
+import org.springframework.lang.NonNull;
+import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.authority.SimpleGrantedAuthority;
+import org.springframework.security.core.context.SecurityContextHolder;
+import org.springframework.security.core.userdetails.UserDetails;
+import org.springframework.security.core.userdetails.User;
+import org.springframework.security.web.authentication.WebAuthenticationDetailsSource;
+import org.springframework.stereotype.Component;
+import org.springframework.web.filter.OncePerRequestFilter;
+import edu.alexu.fitfinder.service.JwtService;
+import java.io.IOException;
+import java.util.List;
+
+@AllArgsConstructor
+@Component
+public class JwtAuthenticationFilter extends OncePerRequestFilter {
+
+  private final JwtService jwtService;
+
+  @Override
+  protected void doFilterInternal(
+      @NonNull HttpServletRequest request,
+      @NonNull HttpServletResponse response,
+      @NonNull FilterChain filterChain)
+      throws ServletException, IOException {
+
+    final String authHeader = request.getHeader("Authorization");
+
+    if (authHeader == null || !authHeader.startsWith("Bearer ")) {
+      filterChain.doFilter(request, response);
+      return;
+    }
+
+    final String jwt = authHeader.substring(7);
+    final String userId = jwtService.extractUserId(jwt);
+
+    Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+    if (userId != null && authentication == null) {
+      UserDetails userDetails =
+          new User(userId, "", List.of(new SimpleGrantedAuthority("GENERAL")));
+
+      if (!jwtService.isTokenExpired(jwt)) {
+        UsernamePasswordAuthenticationToken authToken =
+            new UsernamePasswordAuthenticationToken(
+                userDetails, null, userDetails.getAuthorities());
+        authToken.setDetails(new WebAuthenticationDetailsSource().buildDetails(request));
+        SecurityContextHolder.getContext().setAuthentication(authToken);
+      }
+    }
+    filterChain.doFilter(request, response);
+  }
+}

src/main/java/edu/alexu/fitfinder/configs/SecurityConfig.java

@@ -0,0 +1,85 @@
+package edu.alexu.fitfinder.configs;
+
+import java.util.List;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.security.authentication.AuthenticationManager;
+import org.springframework.security.config.annotation.authentication.configuration.AuthenticationConfiguration;
+import org.springframework.security.config.annotation.method.configuration.EnableMethodSecurity;
+import org.springframework.security.config.annotation.web.builders.HttpSecurity;
+import org.springframework.security.config.annotation.web.configuration.EnableWebSecurity;
+import org.springframework.security.config.annotation.web.configurers.AbstractHttpConfigurer;
+import org.springframework.security.config.http.SessionCreationPolicy;
+import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
+import org.springframework.security.crypto.password.PasswordEncoder;
+import org.springframework.security.web.SecurityFilterChain;
+import org.springframework.security.web.authentication.UsernamePasswordAuthenticationFilter;
+
+@Configuration
+@EnableWebSecurity
+@EnableMethodSecurity
+public class SecurityConfig {
+
+  @Autowired private JwtAuthenticationFilter authFilter;
+
+  @Bean
+  public SecurityFilterChain securityFilterChain(HttpSecurity http) throws Exception {
+    http.csrf(AbstractHttpConfigurer::disable)
+        .cors(
+            cors ->
+                cors.configurationSource(
+                    request -> {
+                      var corsConfiguration = new org.springframework.web.cors.CorsConfiguration();
+                      // Allow common frontend origins (GitHub Pages, local dev) and ngrok domain
+                      corsConfiguration.setAllowedOriginPatterns(
+                          List.of(
+                              "https://*.github.io",
+                              "https://*.githubusercontent.com",
+                              "http://localhost:*",
+                              "http://127.0.0.1:*",
+                              "http://192.168.*.*",
+                              "https://*.ngrok-free.dev",
+                              "https://*.ngrok-free.app",
+                              "https://telescopic-ungodlily-wilbert.ngrok-free.dev"
+                          ));
+                      corsConfiguration.setAllowCredentials(true); // allow cookies/authorization
+                      corsConfiguration.setAllowedMethods(
+                          List.of("GET", "POST", "PUT", "DELETE", "OPTIONS"));
+                      corsConfiguration.setAllowedHeaders(List.of("*"));
+                      corsConfiguration.setExposedHeaders(List.of("*"));
+                      return corsConfiguration;
+                    }))
+        .authorizeHttpRequests(
+            auth ->
+                    auth.requestMatchers(
+                        "/api/v1/auth/**",
+                        "/api/v1/items/**",
+                        "/api/v1/favorites/**",
+                        "/segment/upload",
+                        "/re-segment",
+                        "/segmentation/callback",
+                        "/ws"
+                    )
+                    .permitAll()
+                    .anyRequest()
+                    .authenticated())
+        .sessionManagement(
+            session -> session.sessionCreationPolicy(SessionCreationPolicy.STATELESS) // No sessions
+            )
+        .addFilterBefore(authFilter, UsernamePasswordAuthenticationFilter.class);
+
+    return http.build();
+  }
+
+  @Bean
+  public PasswordEncoder passwordEncoder() {
+    return new BCryptPasswordEncoder(); // Password encoding
+  }
+
+  @Bean
+  public AuthenticationManager authenticationManager(AuthenticationConfiguration config)
+      throws Exception {
+    return config.getAuthenticationManager();
+  }
+}

src/main/java/edu/alexu/fitfinder/configs/WebSocketConfig.java

@@ -0,0 +1,17 @@
+package edu.alexu.fitfinder.configs;
+
+import edu.alexu.fitfinder.handler.MyWebSocketHandler;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Bean;
+import org.springframework.web.socket.config.annotation.*;
+import org.springframework.web.socket.server.standard.ServletServerContainerFactoryBean;
+
+@Configuration
+@EnableWebSocket
+public class WebSocketConfig implements WebSocketConfigurer {
+
+  @Override
+  public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
+    registry.addHandler(new MyWebSocketHandler(), "/ws").setAllowedOriginPatterns("*");
+  }
+}

src/main/java/edu/alexu/fitfinder/controller/FavoriteController.java

@@ -0,0 +1,88 @@
+package edu.alexu.fitfinder.controller;
+
+import edu.alexu.fitfinder.dto.ItemDTO;
+import edu.alexu.fitfinder.exception.FavoriteNotFoundException;
+import edu.alexu.fitfinder.exception.ItemNotFoundException;
+import edu.alexu.fitfinder.exception.UserNotFoundException;
+import edu.alexu.fitfinder.service.FavoriteService;
+import edu.alexu.fitfinder.service.JwtService;
+import lombok.RequiredArgsConstructor;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.*;
+
+import java.util.List;
+
+@RestController
+@RequestMapping("/api/v1/favorites")
+@CrossOrigin
+@RequiredArgsConstructor
+public class FavoriteController {
+
+    private final JwtService jwtService;
+    private final FavoriteService favoriteService;
+
+
+    @PostMapping("/{itemId}")
+    public ResponseEntity<?> addToFavorites(
+            @PathVariable Long itemId,
+            @RequestHeader("Authorization") String token) {
+
+        try {
+            Long userId = jwtService.extractUserFromToken(token);
+            favoriteService.addFavorite(userId, itemId);
+            return ResponseEntity.ok("Item added to favorites");
+        } catch (IllegalArgumentException e) {
+            return ResponseEntity.status(HttpStatus.UNAUTHORIZED).build();
+        }
+    }
+
+    // FIX: Changed path to avoid conflict with /{itemId}
+    @DeleteMapping("/id/{favId}")
+    public ResponseEntity<?> deleteById(
+            @PathVariable Long favId,
+            @RequestHeader("Authorization") String token) {
+
+        try {
+            Long userId = jwtService.extractUserFromToken(token);
+            favoriteService.deleteFavoriteById(userId, favId);
+            return ResponseEntity.ok("Item removed from favorites");
+
+        } catch (IllegalArgumentException e) {
+            return ResponseEntity.status(HttpStatus.UNAUTHORIZED).build();
+        } catch (FavoriteNotFoundException e) {
+            return ResponseEntity.status(HttpStatus.NOT_FOUND).body(e.getMessage());
+        } catch (RuntimeException e) {
+            // Catches the "Unauthorized: You do not own this favorite" from Service
+            return ResponseEntity.status(HttpStatus.FORBIDDEN).body(e.getMessage());
+        }
+    }
+
+    // This handles DELETE /favorites/5 (Assumes 5 is Item ID)
+    @DeleteMapping("/{itemId}")
+    public ResponseEntity<?> deleteByItem(
+            @PathVariable Long itemId,
+            @RequestHeader("Authorization") String token) {
+
+        try {
+            Long userId = jwtService.extractUserFromToken(token);
+            favoriteService.deleteFavoriteByItem(userId, itemId);
+            return ResponseEntity.ok("Item removed from favorites");
+
+        } catch (IllegalArgumentException e) {
+            return ResponseEntity.status(HttpStatus.UNAUTHORIZED).build();
+        } catch (FavoriteNotFoundException | UserNotFoundException | ItemNotFoundException e) {
+            return ResponseEntity.status(HttpStatus.NOT_FOUND).body(e.getMessage());
+        }
+    }
+
+    @GetMapping
+    public ResponseEntity<List<ItemDTO>> getAllFavorites(@RequestHeader("Authorization") String token) {
+        try{
+            Long userId = jwtService.extractUserFromToken(token);
+            return ResponseEntity.ok(favoriteService.getUserFavorites(userId));
+        }catch (IllegalArgumentException e) {
+            return ResponseEntity.status(HttpStatus.UNAUTHORIZED).build();
+        }
+    }
+}

src/main/java/edu/alexu/fitfinder/controller/GlobalExceptionHandler.java

@@ -0,0 +1,62 @@
+package edu.alexu.fitfinder.controller;
+
+import edu.alexu.fitfinder.exception.InvalidInputException;
+import edu.alexu.fitfinder.exception.UnauthorizedException;
+import edu.alexu.fitfinder.exception.UserAlreadyExistsException;
+import edu.alexu.fitfinder.exception.SocketException;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.ExceptionHandler;
+import org.springframework.web.bind.annotation.RestControllerAdvice;
+import org.springframework.web.multipart.MaxUploadSizeExceededException;
+import org.springframework.web.multipart.MultipartException;
+
+import java.io.IOException;
+import java.util.Map;
+
+@RestControllerAdvice
+public class GlobalExceptionHandler {
+  @ExceptionHandler(UserAlreadyExistsException.class)
+  public ResponseEntity<Map<String, String>> handleUserExists(UserAlreadyExistsException e) {
+    return ResponseEntity.status(HttpStatus.CONFLICT).body(Map.of("error", e.getMessage()));
+  }
+
+  @ExceptionHandler(InvalidInputException.class)
+  public ResponseEntity<Map<String, String>> handleInvalidInput(InvalidInputException e) {
+    return ResponseEntity.status(HttpStatus.UNPROCESSABLE_ENTITY)
+        .body(Map.of("error", e.getMessage()));
+  }
+
+  @ExceptionHandler(UnauthorizedException.class)
+  public ResponseEntity<Map<String, String>> handleUnauthorizedRequest(UnauthorizedException e) {
+    return ResponseEntity.status(HttpStatus.UNAUTHORIZED).body(Map.of("error", e.getMessage()));
+  }
+
+  @ExceptionHandler(MaxUploadSizeExceededException.class)
+  public ResponseEntity<?> handleMaxSizeException(MaxUploadSizeExceededException exc) {
+    return ResponseEntity.status(HttpStatus.PAYLOAD_TOO_LARGE)
+        .body("File size shouldn't exceeds 10MB.");
+  }
+
+  @ExceptionHandler(MultipartException.class)
+  public ResponseEntity<?> handleMultipartException(MultipartException exc) {
+    return ResponseEntity.status(HttpStatus.PAYLOAD_TOO_LARGE).body("Invalid file upload.");
+  }
+
+  @ExceptionHandler(SocketException.class)
+  public ResponseEntity<Map<String, String>> handleSocketException(SocketException e) {
+    return ResponseEntity.status(HttpStatus.BAD_REQUEST).body(Map.of("error", e.getMessage()));
+  }
+
+  @ExceptionHandler(IOException.class)
+  public ResponseEntity<Map<String, String>> handleIOException(IOException e) {
+    return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
+        .body(Map.of("error", e.getMessage()));
+  }
+
+  @ExceptionHandler(Exception.class)
+  public ResponseEntity<Map<String, String>> handleGeneric(Exception e) {
+    return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
+        .body(Map.of("error", "Unexpected error occurred"));
+  }
+}

src/main/java/edu/alexu/fitfinder/controller/SearchController.java

@@ -0,0 +1,58 @@
+package edu.alexu.fitfinder.controller;
+
+import edu.alexu.fitfinder.dto.ItemDTO;
+import edu.alexu.fitfinder.dto.SearchRequestDTO;
+import edu.alexu.fitfinder.repository.ItemRepo;
+import edu.alexu.fitfinder.service.JwtService;
+import edu.alexu.fitfinder.service.SearchService;
+import edu.alexu.fitfinder.service.StoredItemService;
+import lombok.RequiredArgsConstructor;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.*;
+import java.util.LinkedList;
+import java.util.List;
+
+@RestController
+@RequestMapping("/api/v1/items")
+@CrossOrigin
+@RequiredArgsConstructor
+public class SearchController {
+
+  private final JwtService jwtService;
+  private final ItemRepo findRandomEntities;
+  private final SearchService searchService;
+  private final StoredItemService storedItemService;
+
+  @PostMapping("/search")
+  public ResponseEntity<?> searchByImageMask(
+          @RequestBody SearchRequestDTO searchInfo,
+          @RequestHeader("Authorization") String token)
+      throws Exception {
+
+
+    try {
+      Long userId = jwtService.extractUserFromToken(token);
+      List<Long> vectorIds = searchService.getSimilarIndices(searchInfo);
+      if (vectorIds.isEmpty()) return ResponseEntity.ok().body(new LinkedList<>());
+      return ResponseEntity.ok().body(storedItemService.getProductsByVectorIds(vectorIds,userId));
+    } catch (IllegalArgumentException e) {
+      return ResponseEntity.status(HttpStatus.UNAUTHORIZED).build();
+    }
+  }
+
+  @GetMapping("/random")
+  public List<ItemDTO> getRandomItems() {
+    List<Long> randomIds = findRandomEntities.findRandomIds();
+    List<ItemDTO> items = storedItemService.getProducts(randomIds);
+    java.util.Collections.shuffle(items);
+    return items;
+  }
+
+  @PostMapping("/fav/:id")
+  private ResponseEntity<?> addFav(@RequestParam("id") Long item_id){
+    
+    return ResponseEntity.status(HttpStatus.CREATED).build();
+  }
+}

src/main/java/edu/alexu/fitfinder/controller/SegmentationController.java

@@ -0,0 +1,65 @@
+package edu.alexu.fitfinder.controller;
+
+import edu.alexu.fitfinder.component.JobRegistry;
+import edu.alexu.fitfinder.dto.ImageMasksDTO;
+import edu.alexu.fitfinder.dto.ResegmentImageDTO;
+import edu.alexu.fitfinder.exception.InvalidInputException;
+import edu.alexu.fitfinder.service.SegmentationService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
+import org.springframework.web.socket.BinaryMessage;
+import org.springframework.web.socket.TextMessage;
+import org.springframework.web.socket.WebSocketSession;
+import java.io.IOException;
+import java.nio.charset.StandardCharsets;
+import java.util.Map;
+
+@RestController
+public class SegmentationController {
+
+  @Autowired SegmentationService segmentationService;
+  @Autowired JobRegistry jobRegistry;
+
+  @PostMapping("/segment/upload")
+  public ResponseEntity<?> uploadImage(
+      @RequestParam MultipartFile image, @RequestParam String sessionId)
+      throws IOException, Exception {
+
+    Map<String, String> response = segmentationService.UploadImageAndSegment(image, sessionId);
+    return ResponseEntity.status(HttpStatus.OK).body(response);
+  }
+
+  @PostMapping("/re-segment")
+  public ResponseEntity<?> resegment(
+      @RequestParam String sessionId, @RequestBody ResegmentImageDTO resegmentInfo)
+      throws InvalidInputException, Exception {
+
+    segmentationService.Resegment(sessionId, resegmentInfo);
+    return new ResponseEntity<>(HttpStatus.OK);
+  }
+
+  @PutMapping("/segmentation/callback")
+  public void segmentationCallback(@RequestBody ImageMasksDTO masks) throws IOException {
+    System.out.println("mask has been received from hugging face!");
+    WebSocketSession session = jobRegistry.getSession(masks.getJob_id());
+    // make sure that session exists and still open
+    if (session != null && session.isOpen()) {
+      if (masks.getStatus().equals("re-segmented"))
+        session.sendMessage(new TextMessage(masks.getMasks().toString()));
+      else
+        session.sendMessage(
+            new TextMessage(
+                "{\"masks\": "
+                    + masks.getMasks().toString()
+                    + ", \"boxes\": "
+                    + masks.getBoxes().toString()
+                    + "}"));
+
+      System.out.println("mask has been sent to client!");
+      jobRegistry.removeJob(masks.getJob_id());
+    }
+  }
+}

src/main/java/edu/alexu/fitfinder/controller/UserController.java

@@ -0,0 +1,118 @@
+package edu.alexu.fitfinder.controller;
+
+import edu.alexu.fitfinder.dto.UserDTO;
+import edu.alexu.fitfinder.exception.UserNotFoundException;
+import edu.alexu.fitfinder.service.ImageService;
+import edu.alexu.fitfinder.service.UserService;
+import jakarta.servlet.http.HttpServletRequest;
+import jakarta.servlet.http.HttpServletResponse;
+import lombok.RequiredArgsConstructor;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.IOException;
+import java.util.Map;
+
+@RestController()
+@RequestMapping("/api/v1/auth")
+@RequiredArgsConstructor
+public class UserController {
+
+  private final UserService userService;
+
+  @PostMapping("/signup")
+  public ResponseEntity<Map<String, String>> signup(
+      @RequestBody UserDTO user, HttpServletResponse response) {
+    Map<String, String> accessToken = userService.SignUP(user, response);
+    return ResponseEntity.status(HttpStatus.CREATED).body(accessToken);
+  }
+
+  @PostMapping("/login")
+  public ResponseEntity<Map<String, String>> login(
+      @RequestBody UserDTO user, HttpServletResponse response) {
+    Map<String, String> accessToken = userService.LogIn(user, response);
+    return ResponseEntity.status(HttpStatus.OK).body(accessToken);
+  }
+
+  @PostMapping("/refresh")
+  public ResponseEntity<Map<String, String>> refresh(HttpServletRequest request) {
+    Map<String, String> accessToken = userService.RefreshToken(request);
+    return ResponseEntity.status(HttpStatus.OK).body(accessToken);
+  }
+
+  @PostMapping("logout")
+  public ResponseEntity<?> logout(HttpServletResponse response) {
+    userService.LogOut(response);
+    return ResponseEntity.status(HttpStatus.OK).build();
+  }
+
+  @GetMapping("/profile")
+  public ResponseEntity<?> getProfile(@RequestHeader("Authorization") String token) {
+    UserDTO user;
+    try {
+      if (token != null && token.startsWith("Bearer ")) {
+        token = token.substring(7);
+      }
+      else {
+        return ResponseEntity.status(HttpStatus.UNAUTHORIZED).build();
+      }
+
+      user = userService.getUser(token);
+    } catch (UserNotFoundException e) {
+      return ResponseEntity.status(HttpStatus.NOT_FOUND).build();
+    }
+
+    return ResponseEntity.ok(user);
+  }
+
+  @PutMapping("/profile/photo")
+  public ResponseEntity<?> uploadImageProfile(
+          @RequestParam MultipartFile image,
+          @RequestHeader("Authorization") String token) {
+
+    UserDTO user;
+    try {
+      if (token != null && token.startsWith("Bearer ")) {
+        token = token.substring(7);
+      }
+      else {
+        return ResponseEntity.status(HttpStatus.UNAUTHORIZED).build();
+      }
+
+      user = userService.getUser(token);
+      userService.updateImageProfile(user.getId(), image);
+      return ResponseEntity.ok("Profile image Uploaded");
+    } catch (UserNotFoundException e) {
+      return ResponseEntity.status(HttpStatus.NOT_FOUND).build();
+    } catch (IOException e){
+      return ResponseEntity.internalServerError().body(e.getMessage());
+    }
+  }
+
+  @DeleteMapping("/profile/photo")
+  public ResponseEntity<?> deleteImageProfile(
+          @RequestParam MultipartFile image,
+          @RequestHeader("Authorization") String token) {
+
+    UserDTO user;
+    try {
+      if (token != null && token.startsWith("Bearer ")) {
+        token = token.substring(7);
+      }
+      else {
+        return ResponseEntity.status(HttpStatus.UNAUTHORIZED).build();
+      }
+
+      user = userService.getUser(token);
+
+      userService.deleteImageProfile(user.getId());
+      return ResponseEntity.ok("Profile image Uploaded");
+    } catch (UserNotFoundException e) {
+      return ResponseEntity.status(HttpStatus.NOT_FOUND).build();
+    } catch (IOException e){
+      return ResponseEntity.internalServerError().body(e.getMessage());
+    }
+  }
+}

src/main/java/edu/alexu/fitfinder/dto/ImageMasksDTO.java

@@ -0,0 +1,14 @@
+package edu.alexu.fitfinder.dto;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import java.util.List;
+
+@AllArgsConstructor
+@Getter
+public class ImageMasksDTO {
+  String job_id;
+  String status;
+  List<List<Byte>> masks;
+  List<List<Integer>> boxes;
+}

src/main/java/edu/alexu/fitfinder/dto/ItemDTO.java

@@ -0,0 +1,22 @@
+package edu.alexu.fitfinder.dto;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.Setter;
+import lombok.ToString;
+
+@AllArgsConstructor
+@Getter
+@Setter
+@ToString
+public class ItemDTO {
+  private Long item_id;
+  private String category;
+  private String currency;
+  private String description;
+  private String imageURL;
+  private String itemWebURL;
+  private float price;
+  private String title;
+  private boolean isFavorite;
+}

src/main/java/edu/alexu/fitfinder/dto/ResegmentImageDTO.java

@@ -0,0 +1,17 @@
+package edu.alexu.fitfinder.dto;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.Setter;
+
+@AllArgsConstructor
+@Getter
+@Setter
+public class ResegmentImageDTO {
+  String job_id;
+  String image_url;
+  int[][] pos_points;
+  int[][] neg_points;
+  int[][] boxes;
+  String callback_url;
+}

src/main/java/edu/alexu/fitfinder/dto/SearchRequestDTO.java

@@ -0,0 +1,16 @@
+package edu.alexu.fitfinder.dto;
+
+import java.util.List;
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.Setter;
+
+@AllArgsConstructor
+@Getter
+@Setter
+public class SearchRequestDTO {
+  String job_id;
+  String image_url;
+  List<List<Integer>> mask_json;
+  String prompt;
+}

src/main/java/edu/alexu/fitfinder/dto/SearchResponseDTO.java

@@ -0,0 +1,23 @@
+package edu.alexu.fitfinder.dto;
+
+import lombok.*;
+
+@AllArgsConstructor
+@NoArgsConstructor
+@Getter
+@Setter
+@ToString
+public class SearchResponseDTO {
+
+  @AllArgsConstructor
+  @Getter
+  @Setter
+  @ToString
+  public static class Result {
+    long faiss_id;
+    double distance;
+  }
+
+  String job_id;
+  Result[] results;
+}

src/main/java/edu/alexu/fitfinder/dto/UserDTO.java

@@ -0,0 +1,18 @@
+package edu.alexu.fitfinder.dto;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+
+@NoArgsConstructor
+@AllArgsConstructor
+@Getter
+@Setter
+public class UserDTO {
+  private Long id;
+  private String profileImageURL;
+  private String userName;
+  private String password;
+  private String email;
+}

src/main/java/edu/alexu/fitfinder/entity/Favorite.java

@@ -0,0 +1,30 @@
+package edu.alexu.fitfinder.entity;
+
+import jakarta.persistence.*;
+import lombok.Data;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+
+@Entity
+@Data
+@Table(name = "FAVORITES")
+@NoArgsConstructor
+@Getter
+@Setter
+public class Favorite {
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.SEQUENCE)
+    @Column(name = "fav_id")
+    private Long id;
+
+    @ManyToOne
+    @JoinColumn(name = "user_id", nullable = false) // Defines the Foreign Key column
+    private User user;
+
+    @ManyToOne
+    @JoinColumn(name = "item_id", nullable = false) // Defines the Foreign Key column
+    private StoredItem item;
+
+}

src/main/java/edu/alexu/fitfinder/entity/ItemVector.java

@@ -0,0 +1,26 @@
+package edu.alexu.fitfinder.entity;
+
+import jakarta.persistence.*;
+import lombok.*;
+import org.hibernate.annotations.CreationTimestamp;
+import java.time.LocalDateTime;
+
+@Entity
+@Table(name = "ITEM_VECTOR")
+@Data
+@NoArgsConstructor
+@Getter
+@Setter
+public class ItemVector {
+
+  @Id
+  @GeneratedValue(strategy = GenerationType.SEQUENCE)
+  private Long id;
+
+  @CreationTimestamp private LocalDateTime createdAt;
+  private Long vectorId;
+
+  @ManyToOne
+  @JoinColumn(name = "item_id", nullable = false)
+  private StoredItem item;
+}

src/main/java/edu/alexu/fitfinder/entity/StoredItem.java

@@ -0,0 +1,61 @@
+package edu.alexu.fitfinder.entity;
+
+import java.time.LocalDateTime;
+
+import org.hibernate.annotations.CreationTimestamp;
+
+import jakarta.persistence.*;
+import lombok.Data;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+
+@Entity
+@Data
+@Table(name = "STORED_ITEMS")
+@NoArgsConstructor
+@Getter
+@Setter
+public class StoredItem {
+
+  @Id
+  @GeneratedValue(strategy = GenerationType.SEQUENCE)
+  private Long itemId;
+
+  private String category;
+  @CreationTimestamp private LocalDateTime createdAt;
+  private String currency;
+
+  @Column(columnDefinition = "TEXT", name = "description")
+  private String description;
+
+  private boolean embedded;
+  private String imageURL;
+  private String itemWebURL;
+  private float price;
+  private String source;
+  private String title;
+
+  public StoredItem(
+      String category,
+      LocalDateTime createdAt,
+      String currency,
+      String description,
+      boolean embedded,
+      String imageURL,
+      String itemWebURL,
+      float price,
+      String source,
+      String title) {
+    this.category = category;
+    this.createdAt = createdAt;
+    this.currency = currency;
+    this.description = description;
+    this.embedded = embedded;
+    this.imageURL = imageURL;
+    this.itemWebURL = itemWebURL;
+    this.price = price;
+    this.source = source;
+    this.title = title;
+  }
+}

src/main/java/edu/alexu/fitfinder/entity/UploadedImage.java

@@ -0,0 +1,37 @@
+package edu.alexu.fitfinder.entity;
+
+import jakarta.persistence.*;
+import lombok.Data;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+
+@Entity
+@Data
+@Table(name = "UPLOADED_IMAGES")
+@NoArgsConstructor
+@Getter
+@Setter
+public class UploadedImage {
+  @Id
+  @GeneratedValue(strategy = GenerationType.SEQUENCE)
+  private Long uploadedImgId;
+
+  @ManyToOne
+  @JoinColumn(name = "user_id", nullable = false)
+  private User user;
+
+  private Long uploadedFAISSId;
+  private String imageURL;
+  private String croppedImageURL;
+  private String name;
+
+  //  private boolean[][] mask;
+
+  public UploadedImage(User user, Long uploadedFAISSId, String imageURL /*, boolean[][] mask*/) {
+    this.user = user;
+    this.uploadedFAISSId = uploadedFAISSId;
+    this.imageURL = imageURL;
+    //    this.mask = mask;
+  }
+}

src/main/java/edu/alexu/fitfinder/entity/User.java

@@ -0,0 +1,41 @@
+package edu.alexu.fitfinder.entity;
+
+import jakarta.persistence.*;
+import lombok.Data;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.Setter;
+
+import java.util.List;
+
+@Entity
+@Data
+@Table(name = "USERS")
+@NoArgsConstructor
+@Getter
+@Setter
+public class User {
+  @Id
+  @GeneratedValue(strategy = GenerationType.SEQUENCE)
+  @Column(name = "user_id")
+  private Long userId;
+
+  private String userName;
+  private String password;
+  private String email;
+
+  @Column(nullable = false)
+  private String profileImageURL = "";
+
+  @OneToMany(mappedBy = "user", cascade = CascadeType.ALL, orphanRemoval = true)
+  private List<UploadedImage> images;
+
+  @OneToMany(mappedBy = "user", cascade = CascadeType.ALL, orphanRemoval = true)
+  private List<Favorite> favorites;
+
+  public User(String userName, String password, String email) {
+    this.userName = userName;
+    this.password = password;
+    this.email = email;
+  }
+}

src/main/java/edu/alexu/fitfinder/exception/FavoriteNotFoundException.java

@@ -0,0 +1,7 @@
+package edu.alexu.fitfinder.exception;
+
+public class FavoriteNotFoundException extends RuntimeException {
+    public FavoriteNotFoundException(String message) {
+        super(message);
+    }
+}

src/main/java/edu/alexu/fitfinder/exception/ImageNotFoundException.java

@@ -0,0 +1,7 @@
+package edu.alexu.fitfinder.exception;
+
+public class ImageNotFoundException extends RuntimeException {
+    public ImageNotFoundException(String message) {
+        super(message);
+    }
+}

src/main/java/edu/alexu/fitfinder/exception/InvalidInputException.java

@@ -0,0 +1,7 @@
+package edu.alexu.fitfinder.exception;
+
+public class InvalidInputException extends RuntimeException {
+  public InvalidInputException(String message) {
+    super(message);
+  }
+}

src/main/java/edu/alexu/fitfinder/exception/ItemNotFoundException.java

@@ -0,0 +1,7 @@
+package edu.alexu.fitfinder.exception;
+
+public class ItemNotFoundException extends RuntimeException {
+    public ItemNotFoundException(String message) {
+        super(message);
+    }
+}

src/main/java/edu/alexu/fitfinder/exception/LogInException.java

@@ -0,0 +1,7 @@
+package edu.alexu.fitfinder.exception;
+
+public class LogInException extends Exception {
+  public LogInException(String message) {
+    super(message);
+  }
+}

src/main/java/edu/alexu/fitfinder/exception/SocketException.java

@@ -0,0 +1,7 @@
+package edu.alexu.fitfinder.exception;
+
+public class SocketException extends RuntimeException {
+  public SocketException(String message) {
+    super(message);
+  }
+}

src/main/java/edu/alexu/fitfinder/exception/UnauthorizedException.java

@@ -0,0 +1,7 @@
+package edu.alexu.fitfinder.exception;
+
+public class UnauthorizedException extends RuntimeException {
+  public UnauthorizedException(String message) {
+    super(message);
+  }
+}

src/main/java/edu/alexu/fitfinder/exception/UserAlreadyExistsException.java

@@ -0,0 +1,7 @@
+package edu.alexu.fitfinder.exception;
+
+public class UserAlreadyExistsException extends RuntimeException {
+  public UserAlreadyExistsException(String message) {
+    super(message);
+  }
+}

src/main/java/edu/alexu/fitfinder/exception/UserNotFoundException.java

@@ -0,0 +1,7 @@
+package edu.alexu.fitfinder.exception;
+
+public class UserNotFoundException extends RuntimeException {
+    public UserNotFoundException(String message) {
+        super(message);
+    }
+}

src/main/java/edu/alexu/fitfinder/exception/ValidatorException.java

@@ -0,0 +1,7 @@
+package edu.alexu.fitfinder.exception;
+
+public class ValidatorException extends Exception {
+  public ValidatorException(String message) {
+    super(message);
+  }
+}

src/main/java/edu/alexu/fitfinder/handler/MyWebSocketHandler.java

@@ -0,0 +1,26 @@
+package edu.alexu.fitfinder.handler;
+
+import org.springframework.web.socket.*;
+import org.springframework.web.socket.handler.TextWebSocketHandler;
+import java.util.concurrent.ConcurrentHashMap;
+
+public class MyWebSocketHandler extends TextWebSocketHandler {
+
+  // store all active sessions
+  public static ConcurrentHashMap<String, WebSocketSession> sessions = new ConcurrentHashMap<>();
+
+  @Override
+  public void afterConnectionEstablished(WebSocketSession session) throws Exception {
+    String sessionId = session.getId();
+    sessions.put(sessionId, session);
+    System.out.println("Websocket connected successfully with sessionId = " + sessionId);
+    session.sendMessage(new TextMessage("{\"sessionId\": \"" + sessionId + "\"}"));
+  }
+
+  @Override
+  public void afterConnectionClosed(WebSocketSession session, CloseStatus status) {
+    String sessionId = session.getId();
+    sessions.remove(sessionId);
+    System.out.println("Websocket disconnected successfully with sessionId = " + sessionId);
+  }
+}

src/main/java/edu/alexu/fitfinder/mapper/ItemMapper.java

@@ -0,0 +1,42 @@
+package edu.alexu.fitfinder.mapper;
+
+import edu.alexu.fitfinder.dto.ItemDTO;
+import edu.alexu.fitfinder.entity.StoredItem;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+import java.util.stream.Collectors;
+
+@Component
+public class ItemMapper {
+
+    public ItemDTO toDTO(StoredItem item) {
+        if (item == null) return null;
+
+        // Ensure these getters match your StoredItem Entity field names
+        return new ItemDTO(
+                item.getItemId(),
+                item.getCategory(),
+                item.getCurrency(),
+                item.getDescription(),
+                item.getImageURL(),
+                item.getItemWebURL(),
+                item.getPrice(),
+                item.getTitle(),
+                false
+        );
+    }
+
+    public ItemDTO toDTO(StoredItem item, boolean isFavorite) {
+        ItemDTO dto = toDTO(item); // Call your existing base method
+        dto.setFavorite(isFavorite); // Set the new field
+        return dto;
+    }
+
+    // Helper to map a whole list at once
+    public List<ItemDTO> toDTOList(List<StoredItem> items) {
+        return items.stream()
+                .map(this::toDTO)
+                .collect(Collectors.toList());
+    }
+}

src/main/java/edu/alexu/fitfinder/mapper/UserMapper.java

@@ -0,0 +1,35 @@
+package edu.alexu.fitfinder.mapper;
+
+import edu.alexu.fitfinder.dto.UserDTO;
+import edu.alexu.fitfinder.entity.User;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+import java.util.stream.Collectors;
+
+@Component
+public class UserMapper {
+
+    public UserDTO toDTO(User user) {
+        if (user == null) {
+            return null;
+        }
+
+        UserDTO dto = new UserDTO();
+
+        // Mapping fields
+        dto.setId(user.getUserId()); // Note the name change: userId -> id
+        dto.setUserName(user.getUserName());
+        dto.setPassword("In Your Dreams 😛"); // Be careful sending passwords in DTOs!
+        dto.setEmail(user.getEmail());
+        dto.setProfileImageURL(user.getProfileImageURL());
+
+        return dto;
+    }
+
+    public List<UserDTO> toDTOList(List<User> users) {
+        return users.stream()
+                .map(this::toDTO)
+                .collect(Collectors.toList());
+    }
+}

src/main/java/edu/alexu/fitfinder/repository/FavoriteRepo.java

@@ -0,0 +1,25 @@
+package edu.alexu.fitfinder.repository;
+
+import edu.alexu.fitfinder.entity.Favorite;
+import edu.alexu.fitfinder.entity.StoredItem;
+import edu.alexu.fitfinder.entity.User;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.jpa.repository.Query;
+import org.springframework.data.repository.query.Param;
+import org.springframework.stereotype.Repository;
+
+import java.util.List;
+import java.util.Optional;
+import java.util.Set;
+
+@Repository
+public interface FavoriteRepo extends JpaRepository<Favorite, Long> {
+    boolean existsByUserAndItem(User user, StoredItem item);
+
+    Optional<Favorite> findByUserAndItem(User user, StoredItem item);
+
+    List<Favorite> findAllByUser(User user);
+
+    @Query("SELECT f.item.itemId FROM Favorite f WHERE f.user.userId = :userId")
+    Set<Long> findFavoriteItemIds(@Param("userId") Long userId);
+}

src/main/java/edu/alexu/fitfinder/repository/ImageRepo.java

@@ -0,0 +1,9 @@
+package edu.alexu.fitfinder.repository;
+
+import edu.alexu.fitfinder.entity.UploadedImage;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface ImageRepo extends JpaRepository<UploadedImage, Long> {
+}

src/main/java/edu/alexu/fitfinder/repository/ItemRepo.java

@@ -0,0 +1,22 @@
+package edu.alexu.fitfinder.repository;
+
+import edu.alexu.fitfinder.entity.StoredItem;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.jpa.repository.Query;
+import org.springframework.data.repository.query.Param;
+import org.springframework.stereotype.Repository;
+import java.util.List;
+
+@Repository
+public interface ItemRepo extends JpaRepository<StoredItem, Long> {
+    @Query(value = "SELECT item_id FROM stored_items ORDER BY RANDOM() LIMIT 20", nativeQuery = true)
+    List<Long> findRandomIds();
+
+    @Query(value = """
+        SELECT DISTINCT si.* FROM STORED_ITEMS si
+        JOIN ITEM_VECTOR vi ON vi.item_id = si.item_id
+        WHERE vi.vector_id IN :vectorIds
+        """, nativeQuery = true)
+    List<StoredItem> findItemsByVectorIds(@Param("vectorIds") List<Long> vectorIds);
+
+}

src/main/java/edu/alexu/fitfinder/repository/UserRepo.java

@@ -0,0 +1,10 @@
+package edu.alexu.fitfinder.repository;
+
+import edu.alexu.fitfinder.entity.User;
+import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.stereotype.Repository;
+
+@Repository
+public interface UserRepo extends JpaRepository<User, Long> {
+  User findByEmail(String email);
+}

src/main/java/edu/alexu/fitfinder/service/AIService.java

@@ -0,0 +1,74 @@
+package edu.alexu.fitfinder.service;
+
+import com.cloudinary.Cloudinary;
+import com.cloudinary.utils.ObjectUtils;
+import edu.alexu.fitfinder.dto.SearchRequestDTO;
+import edu.alexu.fitfinder.exception.InvalidInputException;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.MediaType;
+import org.springframework.stereotype.Service;
+import org.springframework.web.reactive.function.client.WebClient;
+import edu.alexu.fitfinder.dto.SearchResponseDTO;
+import java.util.LinkedList;
+import java.util.List;
+
+@Service
+public class AIService {
+
+  @Value("${cloudinary.url}")
+  private Cloudinary cloudinary;
+
+  @Value("${forwarding.url}")
+  private String forwardingUrl;
+
+  private final WebClient webClient;
+
+  public AIService() {
+    this.webClient = WebClient.builder().baseUrl("https://fitfinder-ai-service.hf.space").build();
+  }
+
+  public List<Long> getSimilarIndices(SearchRequestDTO searchInfo)
+      throws InvalidInputException, Exception {
+
+    // Validate mask
+    if (searchInfo.getMask_json() == null) throw new InvalidInputException("Mask couldn't be null");
+
+    // Validate job_id
+    try {
+      searchInfo.setImage_url(
+          cloudinary
+              .api()
+              .resource(searchInfo.getJob_id(), ObjectUtils.emptyMap())
+              .get("secure_url")
+              .toString());
+    } catch (Exception e) {
+      throw new InvalidInputException("There is no image associated with job Id");
+    }
+    System.out.println(searchInfo.getImage_url());
+
+    // Call AI Service
+    SearchResponseDTO response =
+        webClient
+            .post()
+            .uri("/api/v1/job/search/")
+            .contentType(MediaType.APPLICATION_JSON)
+            .bodyValue(searchInfo)
+            .retrieve()
+            .bodyToMono(SearchResponseDTO.class)
+            .block();
+
+    // Validate response
+    System.out.println("response : " + response);
+    if (response == null) throw new Exception();
+
+    // No matched elements
+    LinkedList<Long> vectorIds = new LinkedList<>();
+    if (response.getResults() == null || response.getResults().length == 0) return vectorIds;
+
+    // Extract Vector Ids
+    for (SearchResponseDTO.Result result : response.getResults())
+      vectorIds.add(result.getFaiss_id());
+    System.out.println(vectorIds);
+    return vectorIds;
+  }
+}

src/main/java/edu/alexu/fitfinder/service/FavoriteService.java

@@ -0,0 +1,88 @@
+package edu.alexu.fitfinder.service;
+
+import edu.alexu.fitfinder.dto.ItemDTO;
+import edu.alexu.fitfinder.entity.Favorite;
+import edu.alexu.fitfinder.entity.StoredItem;
+import edu.alexu.fitfinder.entity.User;
+import edu.alexu.fitfinder.exception.FavoriteNotFoundException;
+import edu.alexu.fitfinder.exception.ItemNotFoundException;
+import edu.alexu.fitfinder.exception.UserNotFoundException;
+import edu.alexu.fitfinder.mapper.ItemMapper;
+import edu.alexu.fitfinder.repository.FavoriteRepo;
+import edu.alexu.fitfinder.repository.ItemRepo;
+import edu.alexu.fitfinder.repository.UserRepo;
+import jakarta.transaction.Transactional;
+import lombok.RequiredArgsConstructor;
+import org.springframework.stereotype.Service;
+
+import java.util.List;
+import java.util.stream.Collectors;
+
+@Service
+@RequiredArgsConstructor // Lombok for constructor injection
+public class FavoriteService {
+
+    private final FavoriteRepo favoriteRepo;
+    private final UserRepo userRepo;
+    private final ItemRepo itemRepo;
+    private final ItemMapper itemMapper;
+
+    @Transactional
+    public Long  addFavorite(Long userId, Long itemId) {
+        User user = userRepo.findById(userId)
+                .orElseThrow(() -> new UserNotFoundException("User not found"));
+
+        StoredItem item = itemRepo.findById(itemId)
+                .orElseThrow(() -> new ItemNotFoundException("Item not found"));
+
+        if (favoriteRepo.existsByUserAndItem(user, item)) {
+            throw new RuntimeException("Item is already in favorites");
+        }
+
+        Favorite favorite = new Favorite();
+        favorite.setUser(user);
+        favorite.setItem(item);
+
+        Favorite savedFav = favoriteRepo.save(favorite);
+
+        return savedFav.getId();
+    }
+
+    @Transactional
+    public void deleteFavoriteById(Long userId, Long favoriteId){
+        Favorite fav = favoriteRepo.findById(favoriteId)
+                .orElseThrow(() -> new FavoriteNotFoundException("Favorite not found"));
+
+        if (!fav.getUser().getUserId().equals(userId)) {
+            throw new RuntimeException("Unauthorized: You do not own this favorite");
+        }
+
+        favoriteRepo.delete(fav);
+    }
+
+    @Transactional
+    public void deleteFavoriteByItem(Long userId, Long itemId) {
+        User user = userRepo.findById(userId)
+                .orElseThrow(() -> new UserNotFoundException("User not found"));
+
+        StoredItem item = itemRepo.findById(itemId)
+                .orElseThrow(() -> new ItemNotFoundException("Item not found"));
+
+        Favorite fav = favoriteRepo.findByUserAndItem(user, item)
+                .orElseThrow(() -> new FavoriteNotFoundException("Favorite not found"));
+
+        favoriteRepo.delete(fav);
+    }
+
+    public List<ItemDTO> getUserFavorites(Long userId) {
+        User user = userRepo.findById(userId)
+                .orElseThrow(() -> new UserNotFoundException("User not found"));
+
+        List<Favorite> favorites = favoriteRepo.findAllByUser(user);
+
+        return favorites.stream()
+                .map(fav -> itemMapper.toDTO(fav.getItem(), true))
+                .collect(Collectors.toList());
+
+    }
+}

src/main/java/edu/alexu/fitfinder/service/ImageService.java

@@ -0,0 +1,74 @@
+package edu.alexu.fitfinder.service;
+
+import com.cloudinary.Cloudinary;
+import com.cloudinary.utils.ObjectUtils;
+import edu.alexu.fitfinder.entity.UploadedImage;
+import edu.alexu.fitfinder.entity.User;
+import edu.alexu.fitfinder.exception.ImageNotFoundException;
+import edu.alexu.fitfinder.exception.UserNotFoundException;
+import edu.alexu.fitfinder.repository.ImageRepo;
+import edu.alexu.fitfinder.repository.UserRepo;
+import lombok.RequiredArgsConstructor;
+import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.io.IOException;
+import java.util.UUID;
+
+@Service
+@RequiredArgsConstructor
+public class ImageService {
+
+  // FIX 1: Removed @Value. Spring now injects the Bean from CloudinaryConfig
+  private final Cloudinary cloudinary;
+
+  private final ImageRepo imageRepo;
+  private final UserRepo userRepo;
+
+  public String uploadImage(MultipartFile image, String imageName) throws IOException {
+    var params = ObjectUtils.asMap(
+            "public_id", imageName,
+            "overwrite", true
+    );
+    var uploadResult = cloudinary.uploader().upload(image.getBytes(), params);
+    return uploadResult.get("secure_url").toString();
+  }
+
+  @Transactional
+  public UploadedImage saveImage(MultipartFile image, User user) throws IOException {
+    UploadedImage img = new UploadedImage();
+    img.setUser(user);
+
+    // FIX 2: Use UUID for unique naming.
+    // Using .size() causes overwrites if images are deleted later.
+    String imageName = "user_" + user.getUserId() + "_" + UUID.randomUUID().toString();
+
+    String url = this.uploadImage(image, imageName);
+
+    img.setImageURL(url);
+    img.setName(imageName);
+
+    return imageRepo.save(img);
+  }
+
+  public void deleteImage(String publicId) throws IOException {
+    cloudinary.uploader().destroy(publicId, ObjectUtils.emptyMap());
+  }
+
+  @Transactional
+  public void deleteImageRow(Long imageId, Long userId) throws IOException {
+
+
+    UploadedImage img = imageRepo.findById(imageId)
+            .orElseThrow(() -> new ImageNotFoundException("Image not found"));
+
+    if (!img.getUser().getUserId().equals(userId)) {
+      throw new RuntimeException("Unauthorized: You do not own this Image");
+    }
+
+    this.deleteImage(img.getName());
+
+    imageRepo.delete(img);
+  }
+}

src/main/java/edu/alexu/fitfinder/service/JwtService.java

@@ -0,0 +1,88 @@
+package edu.alexu.fitfinder.service;
+
+import io.jsonwebtoken.Claims;
+import io.jsonwebtoken.Jwts;
+import io.jsonwebtoken.SignatureAlgorithm;
+import io.jsonwebtoken.io.Decoders;
+import io.jsonwebtoken.security.Keys;
+import java.security.Key;
+import java.util.Date;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.function.Function;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.stereotype.Service;
+
+@Service
+public class JwtService {
+  @Value("${security.jwt.secret-key}")
+  private String secretKey;
+
+  @Value("${security.jwt.access-expiration-time}")
+  private long accessExpiration;
+
+  @Value("${security.jwt.refresh-expiration-time}")
+  private long refreshExpiration;
+
+
+  private Claims extractAllClaims(String token) {
+    return Jwts.parserBuilder()
+        .setSigningKey(getSignInKey())
+        .build()
+        .parseClaimsJws(token)
+        .getBody();
+  }
+
+  private <T> T extractClaim(String token, Function<Claims, T> claimsResolver) {
+    final Claims claims = extractAllClaims(token);
+    return claimsResolver.apply(claims);
+  }
+
+  public Date extractExpiration(String token) {
+    return extractClaim(token, Claims::getExpiration);
+  }
+
+  public String extractUserId(String token) {
+    return extractClaim(token, Claims::getSubject);
+  }
+
+  public String generateAccessToken(String id) {
+    return generateToken(new HashMap<>(), id, accessExpiration);
+  }
+
+  public String generateRefreshToken(String id) {
+    return generateToken(new HashMap<>(), id, refreshExpiration);
+  }
+
+  private String generateToken(Map<String, Object> extraClaims, String id, long expiration) {
+    return buildToken(extraClaims, id, expiration);
+  }
+
+  private String buildToken(Map<String, Object> extraClaims, String id, long expiration) {
+    return Jwts.builder()
+        .setClaims(extraClaims)
+        .setSubject(id)
+        .setIssuedAt(new Date(System.currentTimeMillis()))
+        .setExpiration(new Date(System.currentTimeMillis() + expiration))
+        .signWith(getSignInKey(), SignatureAlgorithm.HS256)
+        .compact();
+  }
+
+  public boolean isTokenExpired(String token) {
+    return extractExpiration(token).before(new Date());
+  }
+
+  public Long extractUserFromToken(String token) {
+    if (token == null || !token.startsWith("Bearer ")) {
+      throw new IllegalArgumentException("Invalid Token");
+    }
+    // Clean the token once
+    String cleanToken = token.substring(7);
+    return Long.parseLong(this.extractUserId(cleanToken));
+  }
+
+  private Key getSignInKey() {
+    byte[] keyBytes = Decoders.BASE64.decode(secretKey);
+    return Keys.hmacShaKeyFor(keyBytes);
+  }
+}

src/main/java/edu/alexu/fitfinder/service/SearchService.java

@@ -0,0 +1,74 @@
+package edu.alexu.fitfinder.service;
+
+import com.cloudinary.Cloudinary;
+import com.cloudinary.utils.ObjectUtils;
+import edu.alexu.fitfinder.dto.SearchRequestDTO;
+import edu.alexu.fitfinder.exception.InvalidInputException;
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.http.MediaType;
+import org.springframework.stereotype.Service;
+import org.springframework.web.reactive.function.client.WebClient;
+import edu.alexu.fitfinder.dto.SearchResponseDTO;
+import java.util.LinkedList;
+import java.util.List;
+
+@Service
+public class SearchService {
+
+  @Value("${cloudinary.url}")
+  private Cloudinary cloudinary;
+
+  @Value("${forwarding.url}")
+  private String forwardingUrl;
+
+  private final WebClient webClient;
+
+  public SearchService() {
+    this.webClient = WebClient.builder().baseUrl("https://fitfinder-ai-service.hf.space").build();
+  }
+
+  public List<Long> getSimilarIndices(SearchRequestDTO searchInfo)
+      throws InvalidInputException, Exception {
+
+    // Validate mask
+    if (searchInfo.getMask_json() == null) throw new InvalidInputException("Mask couldn't be null");
+
+    // Validate job_id
+    try {
+      searchInfo.setImage_url(
+          cloudinary
+              .api()
+              .resource(searchInfo.getJob_id(), ObjectUtils.emptyMap())
+              .get("secure_url")
+              .toString());
+    } catch (Exception e) {
+      throw new InvalidInputException("There is no image associated with job Id");
+    }
+    System.out.println(searchInfo.getImage_url());
+
+    // Call AI Service
+    SearchResponseDTO response =
+        webClient
+            .post()
+            .uri("/api/v1/job/search/")
+            .contentType(MediaType.APPLICATION_JSON)
+            .bodyValue(searchInfo)
+            .retrieve()
+            .bodyToMono(SearchResponseDTO.class)
+            .block();
+
+    // Validate response
+    System.out.println("response : " + response);
+    if (response == null) throw new Exception();
+
+    // No matched elements
+    LinkedList<Long> vectorIds = new LinkedList<>();
+    if (response.getResults() == null || response.getResults().length == 0) return vectorIds;
+
+    // Extract Vector Ids
+    for (SearchResponseDTO.Result result : response.getResults())
+      vectorIds.add(result.getFaiss_id());
+    System.out.println(vectorIds);
+    return vectorIds;
+  }
+}