From d7213716e5fce65832ddaf26d8f805d4fc682c6c Mon Sep 17 00:00:00 2001 From: Sherika Date: Thu, 21 Nov 2024 06:19:05 -0500 Subject: [PATCH] Add logging utility and refactor app structure --- app.py | 13 ++----------- utils/__init__.py | 0 utils/logging_utils.py | 15 +++++++++++++++ 3 files changed, 17 insertions(+), 11 deletions(-) create mode 100644 utils/__init__.py create mode 100644 utils/logging_utils.py diff --git a/app.py b/app.py index 95040d8..c47951a 100644 --- a/app.py +++ b/app.py @@ -1,6 +1,6 @@ import streamlit as st import os -import logging +from utils.logging_utils import setup_logging from typing import Dict, List from datetime import datetime from dotenv import load_dotenv @@ -93,16 +93,7 @@ def get_svg_base64(svg_content): """, unsafe_allow_html=True) -# Configure logging -logging.basicConfig( - level=logging.INFO, - format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', - handlers=[ - logging.FileHandler('vetsai.log'), - logging.StreamHandler() - ] -) -logger = logging.getLogger(__name__) +logger = setup_logging() # Load environment variables load_dotenv() diff --git a/utils/__init__.py b/utils/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/utils/logging_utils.py b/utils/logging_utils.py new file mode 100644 index 0000000..2c1f91c --- /dev/null +++ b/utils/logging_utils.py @@ -0,0 +1,15 @@ +import logging + +def setup_logging(): + + # Configure logging + logging.basicConfig( + level=logging.INFO, + format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', + handlers=[ + logging.FileHandler('vetsai.log'), + logging.StreamHandler() + ] + ) + logger = logging.getLogger(__name__) + return logger