pluto90 commited on
Commit
4616f86
·
verified ·
1 Parent(s): a367c7d

Update app/api/main.py

Browse files
Files changed (1) hide show
  1. app/api/main.py +10 -2
app/api/main.py CHANGED
@@ -6,8 +6,14 @@ import os
6
  from app.services.inference import detect_license_plate
7
  from fastapi.staticfiles import StaticFiles
8
  from fastapi.middleware.cors import CORSMiddleware
 
 
 
 
 
 
 
9
 
10
- app = FastAPI()
11
 
12
  BASE_DIR = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))
13
  UPLOAD_DIR = os.path.join(BASE_DIR, "temp")
@@ -15,7 +21,9 @@ UPLOAD_DIR = os.path.join(BASE_DIR, "temp")
15
  os.makedirs(UPLOAD_DIR, exist_ok=True)
16
  app.add_middleware(
17
  CORSMiddleware,
18
- allow_origins=["*"], # for development
 
 
19
  allow_credentials=True,
20
  allow_methods=["*"],
21
  allow_headers=["*"],
 
6
  from app.services.inference import detect_license_plate
7
  from fastapi.staticfiles import StaticFiles
8
  from fastapi.middleware.cors import CORSMiddleware
9
+ from pathlib import Path
10
+
11
+ app = FastAPI(title="PlateVision Backend")
12
+
13
+
14
+
15
+ FRONTEND_URL = os.environ.get("FRONTEND_URL")
16
 
 
17
 
18
  BASE_DIR = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))
19
  UPLOAD_DIR = os.path.join(BASE_DIR, "temp")
 
21
  os.makedirs(UPLOAD_DIR, exist_ok=True)
22
  app.add_middleware(
23
  CORSMiddleware,
24
+ allow_origins=[
25
+ FRONTEND_URL
26
+ ],
27
  allow_credentials=True,
28
  allow_methods=["*"],
29
  allow_headers=["*"],