pluto90 commited on
Commit
8e5eef4
·
verified ·
1 Parent(s): 36604c8

Update app/api/main.py

Browse files
Files changed (1) hide show
  1. app/api/main.py +11 -2
app/api/main.py CHANGED
@@ -18,10 +18,11 @@ FRONTEND_URL = os.environ.get("FRONTEND_URL")
18
  # UPLOAD_DIR = os.path.join(BASE_DIR, "temp")
19
 
20
  BASE_DIR = Path(__file__).resolve().parent.parent.parent
21
- UPLOAD_DIR = BASE_DIR / "temp"
22
-
23
 
24
  os.makedirs(UPLOAD_DIR, exist_ok=True)
 
25
  app.add_middleware(
26
  CORSMiddleware,
27
  allow_origins=[
@@ -35,6 +36,14 @@ app.add_middleware(
35
 
36
  app.mount("/temp", StaticFiles(directory=UPLOAD_DIR), name="temp")
37
 
 
 
 
 
 
 
 
 
38
  @app.post("/api/detect")
39
  async def detect(file: UploadFile = File(...)):
40
  file_path = os.path.join(UPLOAD_DIR, file.filename)
 
18
  # UPLOAD_DIR = os.path.join(BASE_DIR, "temp")
19
 
20
  BASE_DIR = Path(__file__).resolve().parent.parent.parent
21
+ # UPLOAD_DIR = BASE_DIR / "temp"
22
+ UPLOAD_DIR = "/app/temp"
23
 
24
  os.makedirs(UPLOAD_DIR, exist_ok=True)
25
+
26
  app.add_middleware(
27
  CORSMiddleware,
28
  allow_origins=[
 
36
 
37
  app.mount("/temp", StaticFiles(directory=UPLOAD_DIR), name="temp")
38
 
39
+
40
+
41
+ @app.get("/")
42
+ async def root():
43
+ return {"status": "ok", "message": "PlateVision API is running"}
44
+
45
+
46
+
47
  @app.post("/api/detect")
48
  async def detect(file: UploadFile = File(...)):
49
  file_path = os.path.join(UPLOAD_DIR, file.filename)