pvanand commited on
Commit
b1f421c
·
verified ·
1 Parent(s): 5d8720e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -1
main.py CHANGED
@@ -7,6 +7,8 @@ import uvicorn
7
  from fastapi.middleware.cors import CORSMiddleware
8
  from utils import read_file, fetch_job_description, optimize_resume_api
9
  import logging
 
 
10
 
11
  # Set up logging
12
  logging.basicConfig(level=logging.DEBUG)
@@ -27,7 +29,7 @@ class ErrorResponse(BaseModel):
27
 
28
  def verify_token(credentials: HTTPAuthorizationCredentials = Depends(security)):
29
  token = credentials.credentials
30
- if token != "your_secret_token":
31
  raise HTTPException(status_code=401, detail="Invalid authentication token")
32
  return token
33
 
 
7
  from fastapi.middleware.cors import CORSMiddleware
8
  from utils import read_file, fetch_job_description, optimize_resume_api
9
  import logging
10
+ import os
11
+ API_SECRET = os.getenv("API_SECRET")
12
 
13
  # Set up logging
14
  logging.basicConfig(level=logging.DEBUG)
 
29
 
30
  def verify_token(credentials: HTTPAuthorizationCredentials = Depends(security)):
31
  token = credentials.credentials
32
+ if token != API_SECRET:
33
  raise HTTPException(status_code=401, detail="Invalid authentication token")
34
  return token
35