Niansuh commited on
Commit
e4d9983
·
verified ·
1 Parent(s): 32ba13d

Update api/app.py

Browse files
Files changed (1) hide show
  1. api/app.py +40 -41
api/app.py CHANGED
@@ -1,41 +1,40 @@
1
- from fastapi import FastAPI, Request
2
- from starlette.middleware.cors import CORSMiddleware
3
- from fastapi.responses import JSONResponse
4
- from api.logger import setup_logger
5
- from api.routes import router
6
-
7
-
8
- logger = setup_logger(__name__)
9
-
10
- def create_app():
11
- app = FastAPI(
12
- title="NiansuhAI API Gateway",
13
- docs_url=None, # Disable Swagger UI
14
- redoc_url=None, # Disable ReDoc
15
- openapi_url=None, # Disable OpenAPI schema
16
- )
17
-
18
- # CORS settings
19
- app.add_middleware(
20
- CORSMiddleware,
21
- allow_origins=["*"], # Adjust as needed for security
22
- allow_credentials=True,
23
- allow_methods=["*"],
24
- allow_headers=["*"],
25
- )
26
-
27
- # Include routes
28
- app.include_router(router)
29
-
30
- # Global exception handler for better error reporting
31
- @app.exception_handler(Exception)
32
- async def global_exception_handler(request: Request, exc: Exception):
33
- logger.error(f"An error occurred: {str(exc)}")
34
- return JSONResponse(
35
- status_code=500,
36
- content={"message": "An internal server error occurred."},
37
- )
38
-
39
- return app
40
-
41
- app = create_app()
 
1
+ from fastapi import FastAPI, Request
2
+ from starlette.middleware.cors import CORSMiddleware
3
+ from fastapi.responses import JSONResponse
4
+ from api.logger import setup_logger
5
+ from api.routes import router
6
+
7
+ logger = setup_logger(__name__)
8
+
9
+ def create_app():
10
+ app = FastAPI(
11
+ title="NiansuhAI API Gateway",
12
+ docs_url=None, # Disable Swagger UI
13
+ redoc_url=None, # Disable ReDoc
14
+ openapi_url=None, # Disable OpenAPI schema
15
+ )
16
+
17
+ # CORS settings
18
+ app.add_middleware(
19
+ CORSMiddleware,
20
+ allow_origins=["*"], # Adjust as needed for security
21
+ allow_credentials=True,
22
+ allow_methods=["*"],
23
+ allow_headers=["*"],
24
+ )
25
+
26
+ # Include routes
27
+ app.include_router(router)
28
+
29
+ # Global exception handler for better error reporting
30
+ @app.exception_handler(Exception)
31
+ async def global_exception_handler(request: Request, exc: Exception):
32
+ logger.error(f"An error occurred: {str(exc)}")
33
+ return JSONResponse(
34
+ status_code=500,
35
+ content={"message": "An internal server error occurred."},
36
+ )
37
+
38
+ return app
39
+
40
+ app = create_app()