Merge pull request #31 from biek12/main

[fix]增加dockerfile指定配置选项

Dockerfile

@@ -9,9 +9,16 @@ COPY . /app
 
 # 设置环境变量
 ENV PYTHONUNBUFFERED=1
+# 设置工作进程数
+ENV PROCESS_WORKERS=1
+# 设置工作线程数
+ENV PROCESS_THREADS=2
 
 RUN chmod +x /app/start.sh
 
+# 设置 pip 源为清华大学镜像
+RUN pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
+
 # 安装任何所需的依赖项
 RUN pip install --no-cache-dir flask gunicorn requests Pillow flask-cors tiktoken
 

start.sh

@@ -3,8 +3,18 @@
 # 记录当前日期和时间
 NOW=$(date +"%Y-%m-%d-%H-%M")
 
+
+# 如未设置，填充默认值
+if [ -z "$PROCESS_WORKERS" ]; then
+    export PROCESS_WORKERS=1
+fi
+
+if [ -z "$PROCESS_THREADS" ]; then
+    export PROCESS_THREADS=2
+fi
+
 # 启动 Gunicorn 并使用 tee 命令同时输出日志到文件和控制台
-exec gunicorn -w 2 --threads 2 --bind 0.0.0.0:33333 main:app --access-logfile - --error-logfile -
+exec gunicorn -w ${PROCESS_WORKERS} --threads ${PROCESS_THREADS} --bind 0.0.0.0:33333 main:app --access-logfile - --error-logfile -
 
 # python3 ./main.py