fix: find deceit_env package location and copy data correctly

Dockerfile

@@ -4,7 +4,7 @@ ENV PYTHONUNBUFFERED=1
 ENV HF_HOME=/tmp/huggingface
 ENV HOME=/tmp
 ENV TORCHINDUCTOR_CACHE_DIR=/tmp/torch_cache
-ENV PYTHONPATH=/usr/local/lib/python3.10/site-packages
+ENV PYTHONPATH=/app:/usr/local/lib/python3.10/site-packages:/home/trainer/.local/lib/python3.10/site-packages
 
 RUN apt-get update && apt-get install -y git build-essential && rm -rf /var/lib/apt/lists/*
 
@@ -25,11 +25,30 @@ RUN pip install --no-cache-dir \
 RUN pip install --no-cache-dir \
     git+https://github.com/Jayant-kernel/DECEIT-the-ai-truth-environment-.git
 
-RUN python -c "import deceit_env; import os; print(os.path.dirname(deceit_env.__file__))"
+RUN python -c "
+import subprocess
+result = subprocess.run(['find', '/', '-name', 'environment.py', '-path', '*/deceit_env/*'],
+    capture_output=True, text=True)
+print(result.stdout)
+"
+
+RUN python -c "
+import importlib.util, os
+spec = importlib.util.find_spec('deceit_env')
+if spec:
+    pkg_dir = os.path.dirname(spec.origin)
+    data_dst = os.path.join(pkg_dir, 'data')
+    print(f'Package at: {pkg_dir}')
+    os.makedirs(data_dst, exist_ok=True)
+    print(f'Data dir created: {data_dst}')
+" || echo "deceit_env not found"
+
+RUN mkdir -p /usr/local/lib/python3.10/site-packages/deceit_env/data/ && \
+    mkdir -p /usr/local/lib/python3.10/site-packages/deceit_env/server/ && \
+    touch /usr/local/lib/python3.10/site-packages/deceit_env/server/__init__.py || true
 
-RUN mkdir -p /usr/local/lib/python3.10/site-packages/deceit_env/data/
-COPY data/ /usr/local/lib/python3.10/site-packages/deceit_env/data/
 COPY data/ /app/data/
+COPY data/ /usr/local/lib/python3.10/site-packages/deceit_env/data/
 COPY train.py .
 COPY evaluate.py .