diff --git a/forms-flow-idm/keycloak/docker-compose.yml b/forms-flow-idm/keycloak/docker-compose.yml
index 144cbad8ba..fe620ffcd3 100644
--- a/forms-flow-idm/keycloak/docker-compose.yml
+++ b/forms-flow-idm/keycloak/docker-compose.yml
@@ -25,26 +25,29 @@ services:
       - keycloak-server-network
 
   keycloak:
-    image: quay.io/keycloak/keycloak:14.0.0
+    image: quay.io/keycloak/keycloak:21.1.1
     restart: unless-stopped
     container_name: keycloak
     volumes:
-      - ./imports:/opt/jboss/keycloak/imports
-      - ./themes/formsflow:/opt/jboss/keycloak/themes/formsflow
-      
-    command: 
-      - "-b 0.0.0.0 -Dkeycloak.import=/opt/jboss/keycloak/imports/formsflow-ai-realm.json -Dkeycloak.migration.strategy=OVERWRITE_EXISTING"
+      - ./imports:/opt/keycloak/data/import
+      - ./themes/formsflow:/opt/keycloak/themes/formsflow
+      - ./start-keycloak.sh:/opt/keycloak/bin/start-keycloak.sh
+    entrypoint: ["/bin/bash", "/opt/keycloak/bin/start-keycloak.sh"]
     environment:
       - DB_VENDOR=POSTGRES
       - DB_ADDR=keycloak-db
+      - DB_PORT=5432
       - DB_DATABASE=${KEYCLOAK_JDBC_DB:-keycloak}
       - DB_USER=${KEYCLOAK_JDBC_USER-admin}
       - DB_PASSWORD=${KEYCLOAK_JDBC_PASSWORD:-changeme}
-      - KEYCLOAK_USER=${KEYCLOAK_ADMIN_USER:-admin}
-      - KEYCLOAK_PASSWORD=${KEYCLOAK_ADMIN_PASSWORD:-changeme}
+      - KEYCLOAK_ADMIN=${KEYCLOAK_ADMIN_USER:-admin}
+      - KEYCLOAK_ADMIN_PASSWORD=${KEYCLOAK_ADMIN_PASSWORD:-changeme}
+      - KEYCLOAK_START_MODE=${KEYCLOAK_START_MODE:-start-dev}
+      - KEYCLOAK_HTTP_PATH=${KEYCLOAK_HTTP_PATH:-/auth}
+
     ports:
-      - 8080:8080
-    depends_on:
+      - "8080:8080"
+    links:
       - keycloak-db
     networks:
       - keycloak-server-network
diff --git a/forms-flow-idm/keycloak/sample.env b/forms-flow-idm/keycloak/sample.env
index 4849bc4d3a..3d0f44e042 100644
--- a/forms-flow-idm/keycloak/sample.env
+++ b/forms-flow-idm/keycloak/sample.env
@@ -15,4 +15,7 @@
 #Keycloak admin credentials
 
 #KEYCLOAK_ADMIN_USER=admin
-#KEYCLOAK_ADMIN_PASSWORD=changeme
\ No newline at end of file
+#KEYCLOAK_ADMIN_PASSWORD=changeme
+
+#KEYCLOAK_START_MODE=start
+#KEYCLOAK_HTTP_PATH=/auth
diff --git a/forms-flow-idm/keycloak/start-keycloak.sh b/forms-flow-idm/keycloak/start-keycloak.sh
new file mode 100755
index 0000000000..f3620ee2bc
--- /dev/null
+++ b/forms-flow-idm/keycloak/start-keycloak.sh
@@ -0,0 +1,12 @@
+#!/bin/bash
+
+# Default values if the variables are not set
+START_MODE=${KEYCLOAK_START_MODE:-"start"}
+HTTP_PATH=${KEYCLOAK_HTTP_PATH:-"/auth"}
+
+# Construct the command
+COMMAND="/opt/keycloak/bin/kc.sh $START_MODE --import-realm --http-relative-path $HTTP_PATH"
+
+# Execute the command
+echo "Executing command: $COMMAND"
+exec $COMMAND