Spaces:
Build error
Build error
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)
|