SkyNait commited on
Commit
7d41757
·
verified ·
1 Parent(s): a6f9a33

Add logging

Browse files
Files changed (1) hide show
  1. app.py +38 -30
app.py CHANGED
@@ -1,31 +1,39 @@
1
- #!/usr/bin/env python3
2
- import os
3
- import json
4
- import uuid
5
- import uvicorn
6
- import pika
7
- from fastapi import FastAPI, Body, Header, HTTPException
8
- from fastapi.middleware.cors import CORSMiddleware
9
- from worker import main as rabbit_worker
10
-
11
- app = FastAPI()
12
-
13
- API_KEY = os.getenv("SECRET_KEY")
14
-
15
- app.add_middleware(
16
- CORSMiddleware,
17
- allow_origins=["*"],
18
- allow_credentials=True,
19
- allow_methods=["*"],
20
- allow_headers=["*"],
21
- )
22
-
23
- @app.get("/")
24
- async def root():
25
- return {"status": "ok", "message": "API is running"}
26
-
27
-
28
- if __name__ == "__main__":
29
- os.system('python download_models_hf.py')
30
- rabbit_worker()
 
 
 
 
 
 
 
 
31
  uvicorn.run(app, host="0.0.0.0", port=8000)
 
1
+ #!/usr/bin/env python3
2
+ import os
3
+ import json
4
+ import uuid
5
+ import uvicorn
6
+ import pika
7
+ import logging
8
+ from fastapi import FastAPI, Body, Header, HTTPException
9
+ from fastapi.middleware.cors import CORSMiddleware
10
+ from worker import main as rabbit_worker
11
+
12
+ logging.basicConfig(
13
+ level=logging.INFO,
14
+ format="%(asctime)s [%(levelname)s] %(name)s - %(message)s"
15
+ )
16
+ logger = logging.getLogger(__name__)
17
+
18
+ app = FastAPI()
19
+
20
+ API_KEY = os.getenv("SECRET_KEY")
21
+
22
+ app.add_middleware(
23
+ CORSMiddleware,
24
+ allow_origins=["*"],
25
+ allow_credentials=True,
26
+ allow_methods=["*"],
27
+ allow_headers=["*"],
28
+ )
29
+
30
+ @app.get("/")
31
+ async def root():
32
+ logger.info("Root endpoint called.")
33
+ return {"status": "ok", "message": "API is running"}
34
+
35
+
36
+ if __name__ == "__main__":
37
+ os.system('python download_models_hf.py')
38
+ rabbit_worker()
39
  uvicorn.run(app, host="0.0.0.0", port=8000)