Update Dockerfile

Dockerfile

@@ -40,9 +40,17 @@ RUN apt-get update && apt-get install -y \
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# 配置 DrissionPage
+RUN cd /usr/local/lib/python3.11/site-packages/DrissionPage/_configs/ && \
+    sed -i "/browser_path/s/$(grep 'browser_path' 'configs.ini' | awk -F '=' '{print $2}')/\/usr\/bin\/google-chrome-stable/" configs.ini && \
+    sed -i "/arguments/s/\[/\[\'--no-sandbox\', \'--headless=new\', /" configs.ini
+
 # Copy the rest of the application
 COPY . .
 
+# 创建并设置 Chrome 数据目录
+RUN mkdir -p /tmp/chrome-data && chmod 777 /tmp/chrome-data
+
 # Expose the port the app runs on
 EXPOSE 7860
 
@@ -50,6 +58,7 @@ EXPOSE 7860
 ENV OPENAI_API_KEY=None
 ENV ENVIRONMENT="production"
 ENV PORT=7860
+ENV CHROME_PATH=/usr/bin/google-chrome-stable
 
 # Command to run the application
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]