dockerfile bug fixed

.github/workflows/deploy.yml

@@ -37,9 +37,17 @@ jobs:
         uses: aws-actions/amazon-ecr-login@v1
 
       - name: Build and push Docker image
+        env:
+          ECR_REGISTRY: ${{ steps.login-ecr.outputs.registry }}
+          ECR_REPOSITORY: customer-support-chatbot
+          IMAGE_TAG: latest
         run: |
-          docker build -t ${{ secrets.ACCOUNT_ID }}.dkr.ecr.${{ secrets.AWS_REGION }}.amazonaws.com/customer-support-chatbot:latest .
-          docker push ${{ secrets.ACCOUNT_ID }}.dkr.ecr.${{ secrets.AWS_REGION }}.amazonaws.com/customer-support-chatbot:latest
+          docker build \
+            --build-arg AWS_ACCESS_KEY_ID=${{ secrets.AWS_ACCESS_KEY_ID }} \
+            --build-arg AWS_SECRET_ACCESS_KEY=${{ secrets.AWS_SECRET_ACCESS_KEY }} \
+            --build-arg AWS_DEFAULT_REGION=${{ secrets.AWS_REGION }} \
+            -t $ECR_REGISTRY/$ECR_REPOSITORY:$IMAGE_TAG .
+          docker push $ECR_REGISTRY/$ECR_REPOSITORY:$IMAGE_TAG
 
       - name: Deploy model to SageMaker
         run: |
@@ -48,4 +56,4 @@ jobs:
             --region ${{ secrets.AWS_REGION }} \
             --role_arn ${{ secrets.SAGEMAKER_ROLE_ARN }} \
             --ecr_repo_name "customer-support-chatbot" \
-            --endpoint_name "customer-support-chatbot"
+            --endpoint_name "customer-support-chatbot"

app.py

@@ -6,6 +6,7 @@ import gradio as gr
 import os
 import tarfile
 from typing import List, Tuple
+import boto3
 
 class CustomerSupportBot:
     def __init__(self, model_path="models/customer_support_gpt"):
@@ -18,6 +19,9 @@ class CustomerSupportBot:
         self.process = psutil.Process(os.getpid())
         self.model_path = model_path
         self.model_file_path = os.path.join(self.model_path, "model.tar.gz")
+        self.s3 = boto3.client("s3")
+        self.model_key = "models/model.tar.gz"
+        self.bucket_name = "customer-support-gpt"
         
         # Download and load the model
         self.download_and_load_model()

dockerfile

@@ -39,6 +39,16 @@ ENV PYTHONUNBUFFERED=TRUE
 ENV PYTHONDONTWRITEBYTECODE=TRUE
 ENV PATH="/opt/conda/bin:${PATH}"
 
+# Get args for AWS credentials
+ARG AWS_ACCESS_KEY_ID
+ARG AWS_SECRET_ACCESS_KEY
+ARG AWS_DEFAULT_REGION
+
+# get args as env variables
+ENV AWS_ACCESS_KEY_ID=${AWS_ACCESS_KEY_ID}
+ENV AWS_SECRET_ACCESS_KEY=${AWS_SECRET_ACCESS_KEY}
+ENV AWS_DEFAULT_REGION=${AWS_DEFAULT_REGION}
+
 # Install system dependencies
 RUN apt-get update && \
     DEBIAN_FRONTEND=noninteractive apt-get install -y \
@@ -55,6 +65,7 @@ WORKDIR /opt/ml/code
 
 # Copy requirements and install dependencies
 COPY requirements.txt .
+RUN pip install --no-cache-dir --upgrade pip
 RUN pip install --no-cache-dir torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
 RUN pip install --no-cache-dir -r requirements.txt