gavinzli commited on
Commit
f06d4d5
·
1 Parent(s): 5b954ed

Update Dockerfile and refactor FastAPI routers for consistency

Browse files
Files changed (4) hide show
  1. Dockerfile +1 -0
  2. app/main.py +2 -2
  3. app/router/content.py +1 -1
  4. app/router/mail.py +1 -1
Dockerfile CHANGED
@@ -9,6 +9,7 @@ WORKDIR /app
9
 
10
  # Create a cache directory and set permissions
11
  RUN mkdir -p /app/cache && chmod -R 777 /app/cache
 
12
 
13
  # Install requirements.txt
14
  RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
9
 
10
  # Create a cache directory and set permissions
11
  RUN mkdir -p /app/cache && chmod -R 777 /app/cache
12
+ RUN mkdir -p /app/cache && chmod -R 777 /nltk_data
13
 
14
  # Install requirements.txt
15
  RUN pip install --no-cache-dir --upgrade -r requirements.txt
app/main.py CHANGED
@@ -7,8 +7,8 @@ from router import content, mail
7
 
8
  app = FastAPI(docs_url="/")
9
 
10
- app.include_router(content.router, tags=["content"])
11
- app.include_router(mail.router, tags=["mail"])
12
 
13
  origins = [
14
  "*"
 
7
 
8
  app = FastAPI(docs_url="/")
9
 
10
+ app.include_router(content.router)
11
+ app.include_router(mail.router)
12
 
13
  origins = [
14
  "*"
app/router/content.py CHANGED
@@ -4,7 +4,7 @@ from fastapi.responses import StreamingResponse
4
  from schema import ReqData
5
  from utils import generate
6
 
7
- router = APIRouter()
8
 
9
  @router.post("/stream")
10
  async def stream(query: ReqData):
 
4
  from schema import ReqData
5
  from utils import generate
6
 
7
+ router = APIRouter(tags=["content"])
8
 
9
  @router.post("/stream")
10
  async def stream(query: ReqData):
app/router/mail.py CHANGED
@@ -3,7 +3,7 @@ from fastapi import APIRouter
3
  from fastapi.responses import JSONResponse
4
  from controllers import mail
5
 
6
- router = APIRouter(prefix="/mail", tags=["Mail"])
7
 
8
  @router.post("")
9
  def collect():
 
3
  from fastapi.responses import JSONResponse
4
  from controllers import mail
5
 
6
+ router = APIRouter(prefix="/mail", tags=["mail"])
7
 
8
  @router.post("")
9
  def collect():