updated dockerfile

Dockerfile

@@ -8,31 +8,23 @@ WORKDIR /app
 RUN apt-get update && apt-get install -y \
     git \
     xvfb \
+    # Clean up apt caches to reduce image size
     && rm -rf /var/lib/apt/lists/*
 
-# Copy common Python dependencies first
-# These are now at the root of your local Git repo
+# Copy essential files from the repo root to /app in the container
 COPY requirements.txt .
-
-# Copy your Flask app.py to the root of /app
-# This is also now at the root of your local Git repo
 COPY app.py .
 
-# Copy the entire 'src' directory from proxy-lite-demo-v2
-# This ensures 'proxy_lite' module is available for imports
-COPY proxy-lite-demo-v2/src/ /app/src/
+# Copy the entire 'proxy-lite-demo-v2' directory into /app/proxy-lite-demo-v2
+# This will ensure src, pyproject.toml, etc., are in their original nested structure within the container.
+COPY proxy-lite-demo-v2 /app/proxy-lite-demo-v2/
 
-# Copy other relevant sub-directories if your app needs them
-# For example, if you access 'gifs' or 'screenshots' from your app.py
-COPY proxy-lite-demo-v2/gifs/ /app/proxy-lite-demo-v2/gifs/
-COPY proxy-lite-demo-v2/screenshots/ /app/proxy-lite-demo-v2/screenshots/
-# Add other top-level files from proxy-lite-demo-v2 if your app references them directly
-COPY proxy-lite-demo-v2/README.md .
-# And other directories like proxy-lite-work if your app uses them
-COPY proxy-lite-work /app/proxy-lite-work
+# Copy the 'proxy-lite-work' directory if it's needed by the app
+COPY proxy-lite-work /app/proxy-lite-work/
 
-# Ensure the 'src' directory is on the Python path for imports like 'from proxy_lite import ...'
-ENV PYTHONPATH=/app/src:$PYTHONPATH
+# --- IMPORTANT: Add proxy-lite-demo-v2/src to PYTHONPATH ---
+# This is crucial so that 'app.py' (now at /app) can find 'from proxy_lite import ...'
+ENV PYTHONPATH=/app/proxy-lite-demo-v2/src:$PYTHONPATH
 
 # --- START: Directory permission workaround ---
 # Create the directory proxy-lite's recorder insists on writing to
@@ -46,17 +38,8 @@ RUN mkdir -p /app/local_trajectories \
 RUN pip install --no-cache-dir --upgrade pip setuptools wheel
 
 # Install your local proxy-lite package in editable mode.
-# Assuming your setup.py/pyproject.toml for proxy-lite is within src/proxy_lite
-# If it's at proxy-lite-demo-v2/ (which is now copied to /app/proxy-lite-demo-v2),
-# you might need to adjust this, but for now, this assumes it's within /app/src/
-# If 'pip install -e .' expects it at the root of the original project,
-# we need to be careful here. Let's make it explicit if it's a sub-package.
-# From your original commit log: create mode 100644 proxy-lite-demo-v2/pyproject.toml
-# This means proxy-lite itself might be defined by pyproject.toml at the *sub-root*.
-# So, for 'pip install -e .', the '.' would need to point to /app/proxy-lite-demo-v2.
-# Let's adjust this.
-
-# Install the proxy_lite package from its expected location inside the container
+# Now that proxy-lite-demo-v2 (containing pyproject.toml and src/) is at /app/proxy-lite-demo-v2,
+# we explicitly tell pip to install from that directory.
 RUN pip install --no-cache-dir --no-input -e /app/proxy-lite-demo-v2
 
 # Install the rest of the Python dependencies from requirements.txt
@@ -78,5 +61,5 @@ RUN echo "-----------------------------------"
 EXPOSE 7860
 
 # Define the command to run your Flask application using Gunicorn for production.
-# This runs the 'app' instance from the 'app.py' file located at the root of /app
+# This assumes app.py is at the root of /app, which is true because we copied it there.
 CMD exec gunicorn --bind 0.0.0.0:7860 --workers 2 --worker-class gevent app:app --timeout 300