added rollback for both envs

.gitlab-ci.yml

 workflow:
   rules:
-    - if: $CI_PIPELINE_SOURCE == "push" && $CI_COMMIT_BRANCH == "test"
+    - if: $CI_PIPELINE_SOURCE == "push" && $CI_COMMIT_BRANCH == "main"
       when: always
 
 variables:
@@ -170,8 +170,6 @@ deploy_to_staging:
       kubectl apply -f https://raw.githubusercontent.com/kubernetes/ingress-nginx/controller-v1.8.1/deploy/static/provider/cloud/deploy.yaml
       echo "All manifests applied successfully."
 
-  rules:
-    - if: $CI_COMMIT_BRANCH == "test" || $CI_COMMIT_BRANCH == "main"
   needs:
     - create_secrets
 
@@ -210,8 +208,6 @@ deploy_to_production:
       kubectl apply -f https://raw.githubusercontent.com/kubernetes/ingress-nginx/controller-v1.8.1/deploy/static/provider/cloud/deploy.yaml
       echo "All manifests applied successfully."
 
-  rules:
-    - if: $CI_COMMIT_BRANCH == "test" || $CI_COMMIT_BRANCH == "main"
   needs:
     - create_secrets
     - deploy_to_staging
@@ -219,21 +215,17 @@ deploy_to_production:
 rollback:
   <<: *deploy_template
   stage: deploy
-  tags:
-    - gitlab-runner03
+  when: manual
   script:
     - |
-      echo "Performing rollback..."
-      if kubectl get deployment webservice -n $KUBE_NAMESPACE > /dev/null 2>&1; then
+      echo "Performing rollback in ${CI_ENVIRONMENT_NAME} environment..."
+      if kubectl get deployment webservice -n ${KUBE_NAMESPACE} > /dev/null 2>&1; then
         echo "Rolling back webservice deployment"
-        kubectl rollout undo deployment/webservice -n $KUBE_NAMESPACE
+        kubectl rollout undo deployment/webservice -n ${KUBE_NAMESPACE}
       else
-        echo "Deployment 'webservice' not found in namespace $KUBE_NAMESPACE"
+        echo "Deployment 'webservice' not found in namespace ${KUBE_NAMESPACE}"
         echo "Available deployments:"
-        kubectl get deployments -n $KUBE_NAMESPACE
+        kubectl get deployments -n ${KUBE_NAMESPACE}
       fi
-  when: manual
-  environment:
-    name: staging
-  variables:
-    KUBE_NAMESPACE: ${CI_ENVIRONMENT_SLUG}
\ No newline at end of file
+  rules:
+    - if: '$CI_ENVIRONMENT_NAME == "staging" || $CI_ENVIRONMENT_NAME == "production"'
\ No newline at end of file