feat: uv로 실행, 명령어 추가 및 포트 외부에서 설정

Author: ParkGyeongTae

Dockerfile

@@ -9,21 +9,24 @@ RUN apt-get update && apt-get install -y \
     libpq-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# 3. 작업 디렉토리 설정
+# 3. uv 설치
+RUN pip install --no-cache-dir uv
+
+# 4. 작업 디렉토리 설정
 WORKDIR /app
 
-# 4. 소스 코드 복사 및 의존성 설치
+# 5. 소스 코드 복사 및 의존성 설치
 COPY pyproject.toml ./
 COPY . .
-RUN pip install --upgrade pip setuptools wheel \
-    && pip install .
+RUN uv pip install --system --upgrade pip setuptools wheel \
+    && uv pip install --system .
 
-# 5. 환경 변수 설정
+# 6. 환경 변수 설정
 ENV PYTHONPATH=/app
 ENV PYTHONUNBUFFERED=1
 
-# 6. 포트 노출
-EXPOSE 8501
+# 7. 포트 설정
+ENV STREAMLIT_SERVER_PORT=8501
 
-# 7. 실행 명령
-CMD streamlit run ./interface/streamlit_app.py --server.port=8501
+# 8. 실행 명령
+CMD ["lang2sql", "run-streamlit-docker"]

cli/__init__.py

@@ -5,8 +5,8 @@
 명령어 예시: lang2sql --datahub_server http://localhost:8080 --run-streamlit
 """
 
-import os
 import logging
+import os
 import subprocess
 
 import click
@@ -238,6 +238,24 @@ def run_streamlit_cli_command(port: int) -> None:
     run_streamlit_command(port)
 
 
+@cli.command(name="run-streamlit-docker")
+def run_streamlit_docker_command() -> None:
+    """
+    Docker 컨테이너 환경에서 Streamlit 애플리케이션을 실행하는 명령어입니다.
+
+    이 명령은 환경 변수 STREAMLIT_SERVER_PORT 를 읽어와
+    해당 포트에서 Streamlit 서버를 구동합니다.
+    docker-compose.yml 에서 설정한 환경 변수를 그대로 활용할 수 있습니다.
+
+    실행 예시:
+        lang2sql run-streamlit-docker
+    """
+
+    port = int(os.environ.get("STREAMLIT_SERVER_PORT", "8501"))
+    logger.info("Executing 'run-streamlit-docker' on port %d...", port)
+    run_streamlit_command(port)
+
+
 @cli.command(name="query")
 @click.argument("question", type=str)
 @click.option(

docker-compose.yml

@@ -1,5 +1,7 @@
 services:
   streamlit:
+    hostname: streamlit
+    container_name: streamlit
     build: .
     ports:
       - "8501:8501"
@@ -8,21 +10,24 @@ services:
     env_file:
       - .env
     environment:
-      - DATABASE_URL=postgresql://postgres:password@db:5432/streamlit_db
+      - STREAMLIT_SERVER_PORT=8501
+      - DATABASE_URL=postgresql://pgvector:pgvector@pgvector:5432/streamlit
     depends_on:
-      - db
+      - pgvector
 
-  db:
+  pgvector:
     image: pgvector/pgvector:pg17
-    container_name: pgvector-db
+    hostname: pgvector
+    container_name: pgvector
     environment:
-      POSTGRES_USER: postgres
-      POSTGRES_PASSWORD: password
-      POSTGRES_DB: streamlit_db
+      POSTGRES_USER: pgvector
+      POSTGRES_PASSWORD: pgvector
+      POSTGRES_DB: streamlit
     ports:
       - "5432:5432"
     volumes:
       - pgdata:/var/lib/postgresql/data
       - ./postgres/schema.sql:/docker-entrypoint-initdb.d/schema.sql
+
 volumes:
-  pgdata: 
+  pgdata: