Included entrypoint.sh

Dockerfile

@@ -23,6 +23,8 @@ RUN rm requirements.txt
 # Add lambda_entrypoint.py to the container
 COPY lambda_entrypoint.py .
 
+COPY entrypoint.sh .
+
 # Stage 2: Final runtime image
 FROM public.ecr.aws/docker/library/python:3.11.9-slim-bookworm
 
@@ -59,7 +61,7 @@ COPY --from=builder /install /usr/local/lib/python3.11/site-packages/
 #     fi && chmod +x /entrypoint.sh
 
 # Entrypoint helps to switch between Gradio and Lambda mode
-COPY entrypoint.sh /entrypoint.sh
+#COPY entrypoint.sh /entrypoint.sh
 
 # Switch to the "user" user
 USER user

entrypoint.sh

@@ -0,0 +1,9 @@
+#!/bin/sh
+
+if [ "$APP_MODE" = "lambda" ]; then
+    echo "Starting in Lambda mode..."
+    exec python -m awslambdaric "$@"
+else
+    echo "Starting in Gradio mode..."
+    exec python app.py "$@"
+fi

tools/auth.py

@@ -9,8 +9,6 @@ print(f'The value of AWS_CLIENT_ID is {client_id}')
 user_pool_id = get_or_create_env_var('AWS_USER_POOL_ID', '')
 print(f'The value of AWS_USER_POOL_ID is {user_pool_id}')
 
-
-
 def authenticate_user(username:str, password:str, user_pool_id:str=user_pool_id, client_id:str=client_id):
     """Authenticates a user against an AWS Cognito user pool.