ishworrsubedii commited on
Commit
7956aa1
·
1 Parent(s): 3eefc8f

fix: import issue

Browse files
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 = 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)
 
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)