fix: docker compose

Dockerfile

@@ -5,7 +5,7 @@ WORKDIR /home/user/app
 
 COPY requirements.txt .
 
-RUN apt-get update && apt-get install -y --no-install-recommends \
+RUN apt-get update && apt-get install -y --no-install-recommends && \
     wget gnupg ca-certificates && \
     pip install --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt && \

app/util/parameter_utils.py

@@ -0,0 +1,27 @@
+# parameter_utils.py
+import os
+
+import boto3
+
+ssm = boto3.client('ssm', region_name='ap-southeast-3')
+
+def init_secret(path):
+    next_token = None
+    while True:
+        response = ssm.get_parameters_by_path(
+            Path=path,
+            Recursive=True,
+            WithDecryption=True,
+            NextToken=next_token if next_token else "",
+        )
+        parameters = response.get("Parameters", [])
+        for parameter in parameters:
+            key = parameter.get("Name").split("/")[-1]
+            value = parameter.get("Value")
+            os.environ[key] = value
+            print(f"Set env variable {key}")
+
+        next_token = response.get("NextToken")
+        if not next_token:
+            break
+

deploy.sh

@@ -10,21 +10,20 @@ else
     exit 1
 fi
 
-APP_ENV=${DEPLOYMENT_GROUP_NAME}
-
-if [[ "$APP_ENV" == "development" ]]; then
-    API_PORT=5001
-    ES_PORT=9200
-else
-    API_PORT=5002
-    ES_PORT=9201
-fi
 
 echo "=== Deployment Variables ==="
-echo "APP_ENV=${APP_ENV}"
+echo "GOOGLE_AI_STUDIO_API_KEY=${GOOGLE_AI_STUDIO_API_KEY}"
 echo "IMAGE_TAG=${IMAGE_TAG}"
-echo "API_PORT=${API_PORT}"
-echo "ES_PORT=${ES_PORT}"
+echo "GOOGLE_AI_STUDIO_MODEL=${GOOGLE_AI_STUDIO_MODEL}"
+echo "DB_HOST=${DB_HOST}"
+echo "DB_NAME=${DB_NAME}"
+echo "DB_USERNAME=${DB_USERNAME}"
+echo "DB_PORT=${DB_PORT}"
+echo "DB_PASSWORD=${DB_PASSWORD}"
+
+echo "PORT=${PORT}"
+echo "PLAYWRIGHT_BROWSERS_PATH=${PLAYWRIGHT_BROWSERS_PATH}"
+
 export APP_ENV IMAGE_TAG API_PORT ES_PORT
 
 echo "=== Login AWS ==="

docker-compose.yaml

@@ -7,7 +7,6 @@ services:
       - "7860:7860"
     env_file:
       - .env
-    environment:
-      - PORT=7860
-      - PLAYWRIGHT_BROWSERS_PATH=/home/user/.cache/ms-playwright
+    volumes:
+      - ./.env:/.env
     restart: always

requirements.txt

@@ -15,4 +15,6 @@ fpdf
 pandas
 
 SQLAlchemy
-psycopg2-binary
+psycopg2-binary
+
+boto3

server.py

@@ -13,7 +13,8 @@ import io
 
 from app.util.gen_ai_base import GenAIBaseClient
 from app.util.browser_agent import BrowserAgent
-from app.util.japan_multientry_visa_letter_generator   import JapanMultiEntryVisaLetterGenerator
+# from app.util.japan_multientry_visa_letter_generator   import JapanMultiEntryVisaLetterGenerator
+from app.util.parameter_utils import init_secret
 import sys
 sys.stdout.reconfigure(line_buffering=True)
 API = "https://api-dev.spun.global"
@@ -28,6 +29,14 @@ def create_app() -> Flask:
     app = Flask(__name__)
     logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 
+    try:
+        aws_param_path = os.getenv("AWS_PARAMETER_STORE_PATH")
+        if aws_param_path:
+            init_secret(aws_param_path)
+            logging.info("Secrets loaded from AWS SSM Parameter Store.")
+    except Exception as e:
+        logging.exception(f"Could not load secrets from SSM: {e}")
+            
     @app.route('/scrape', methods=['POST'])
     async def scrape():
         try: