change docker

Dockerfile

@@ -1,9 +1,21 @@
-FROM python:3.10.9
+FROM python:3.9-slim
 
-COPY . .
+WORKDIR /code
 
-WORKDIR /
+# 复制项目文件
+COPY ./requirements.txt /code/requirements.txt
+COPY ./app /code/app
+COPY ./utils /code/utils
+COPY ./main.py /code/main.py
 
-RUN pip install --no-cache-dir --upgrade -r /requirements.txt
+# 安装依赖
+RUN pip install --no-cache-dir -r requirements.txt
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+# 设置环境变量
+ENV PYTHONPATH=/code
+
+# 暴露端口
+EXPOSE 7860
+
+# 启动命令
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

app.py → main.py

@@ -10,15 +10,19 @@ import time
 from app.backend.constant import Navigation, ModelProvider, EvaluationMetric, EmbdDtype, EmbdDim, Similarity
 from app.backend.data_engine import DataEngine
 
+# 创建 FastAPI 应用
 app = FastAPI()
 
-# Mount static files
-app.mount("/static", StaticFiles(directory="app/static"), name="static")
+# 获取项目根目录
+BASE_DIR = Path(__file__).resolve().parent
 
-# Templates
-templates = Jinja2Templates(directory="app/templates")
+# 挂载静态文件
+app.mount("/static", StaticFiles(directory=str(BASE_DIR / "app" / "static")), name="static")
 
-# Initialize data engine
+# 设置模板
+templates = Jinja2Templates(directory=str(BASE_DIR / "app" / "templates"))
+
+# 初始化数据引擎
 data_engine = None
 initialization_start_time = None
 
@@ -105,7 +109,3 @@ async def filter_data(filters: dict):
             status_code=500,
             content={"error": str(e)}
         )
-
-# if __name__ == "__main__":
-#     import uvicorn
-#     uvicorn.run(app, host="0.0.0.0", port=7891)