Spaces:
Sleeping
Sleeping
Commit
·
7956aa1
1
Parent(s):
3eefc8f
fix: import issue
Browse files- src/api/image_prep_api.py +1 -1
- src/api/image_regeneration_api.py +1 -1
- src/utils/logger.py +4 -4
src/api/image_prep_api.py
CHANGED
@@ -19,7 +19,7 @@ from fastapi.responses import JSONResponse
|
|
19 |
from src.components.auto_crop import crop_transparent_image
|
20 |
from src.components.color_extraction import ColorExtractionRMBG
|
21 |
from src.components.title_des_gen import NecklaceProductListing
|
22 |
-
from src.utils import logger
|
23 |
|
24 |
preprocessing_router = APIRouter()
|
25 |
|
|
|
19 |
from src.components.auto_crop import crop_transparent_image
|
20 |
from src.components.color_extraction import ColorExtractionRMBG
|
21 |
from src.components.title_des_gen import NecklaceProductListing
|
22 |
+
from src.utils.logger import logger
|
23 |
|
24 |
preprocessing_router = APIRouter()
|
25 |
|
src/api/image_regeneration_api.py
CHANGED
@@ -13,7 +13,7 @@ import requests
|
|
13 |
from PIL import Image
|
14 |
from fastapi import APIRouter, UploadFile, File, Form
|
15 |
from fastapi.responses import JSONResponse
|
16 |
-
from src.utils import logger
|
17 |
|
18 |
image_regeneration_router = APIRouter()
|
19 |
|
|
|
13 |
from PIL import Image
|
14 |
from fastapi import APIRouter, UploadFile, File, Form
|
15 |
from fastapi.responses import JSONResponse
|
16 |
+
from src.utils.logger import logger
|
17 |
|
18 |
image_regeneration_router = APIRouter()
|
19 |
|
src/utils/logger.py
CHANGED
@@ -5,18 +5,18 @@ logger = logging.getLogger(__name__)
|
|
5 |
logger.setLevel(logging.INFO)
|
6 |
|
7 |
log_dir = os.path.join(os.getcwd(), "logs")
|
8 |
-
os.makedirs(log_dir, exist_ok
|
9 |
|
10 |
LOG_FILE = os.path.join(log_dir, "running_logs.log")
|
11 |
|
12 |
logFormat = "[%(asctime)s: %(levelname)s: %(module)s: %(message)s]"
|
13 |
-
logFormatter = logging.Formatter(fmt
|
14 |
|
15 |
streamHandler = logging.StreamHandler()
|
16 |
streamHandler.setFormatter(logFormatter)
|
17 |
|
18 |
-
fileHandler = logging.FileHandler(filename
|
19 |
fileHandler.setFormatter(logFormatter)
|
20 |
|
21 |
logger.addHandler(streamHandler)
|
22 |
-
logger.addHandler(fileHandler)
|
|
|
5 |
logger.setLevel(logging.INFO)
|
6 |
|
7 |
log_dir = os.path.join(os.getcwd(), "logs")
|
8 |
+
os.makedirs(log_dir, exist_ok=True)
|
9 |
|
10 |
LOG_FILE = os.path.join(log_dir, "running_logs.log")
|
11 |
|
12 |
logFormat = "[%(asctime)s: %(levelname)s: %(module)s: %(message)s]"
|
13 |
+
logFormatter = logging.Formatter(fmt=logFormat, style="%")
|
14 |
|
15 |
streamHandler = logging.StreamHandler()
|
16 |
streamHandler.setFormatter(logFormatter)
|
17 |
|
18 |
+
fileHandler = logging.FileHandler(filename=LOG_FILE)
|
19 |
fileHandler.setFormatter(logFormatter)
|
20 |
|
21 |
logger.addHandler(streamHandler)
|
22 |
+
logger.addHandler(fileHandler)
|