Dockerized version

Dockerfile

@@ -0,0 +1,34 @@
+# Use Python 3.10 slim image as base
+FROM python:3.10-slim
+
+# Set working directory
+WORKDIR /app
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    gcc \
+    python3-dev \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN curl -sSL https://install.python-poetry.org | python3 -
+ENV PATH="/root/.local/bin:$PATH"
+
+# Copy requirements first to leverage Docker cache
+# Copy source code
+COPY src/ src/
+COPY .env .
+COPY .gitignore .
+COPY README.md .
+COPY LICENSE .
+COPY pyproject.toml .
+
+# Install Python dependencies
+RUN poetry install
+
+# Set environment variable for Python to run in unbuffered mode
+ENV PYTHONUNBUFFERED=1
+
+# Set default command with -u flag for unbuffered output
+ENTRYPOINT ["poetry", "run", "python", "-u", "src/main.py"]
+CMD ["--ticker", "AAPL,MSFT,NVDA"]  # Optional default arguments

README.md

@@ -126,6 +126,18 @@ You can optionally specify the start and end dates to backtest over a specific t
 poetry run python src/backtester.py --ticker AAPL,MSFT,NVDA --start-date 2024-01-01 --end-date 2024-03-01
 ```
 
+## Using Docker
+
+To build the docker container:
+```bash
+docker build .
+```
+
+To run the docker container:
+```bash
+docker run -it <container_name> --ticker AAPL,MSFT,NVDA
+```
+
 ## Project Structure 
 ```
 ai-hedge-fund/