Add logging utility and refactor app structure

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
@@ -95,16 +95,7 @@ def get_svg_base64(svg_content):
     </style>
 """, 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()

tests/test_streamlit_app.py

@@ -147,6 +147,7 @@ def test_translate_military_code_not_found(self, mock_job_codes):
         assert isinstance(result["data"]["tech_focus"], list)
 
 class TestChatFunctionality:
+    @patch.dict(os.environ, {"OPENAI_API_KEY": "test_api_key"}) 
     @patch('openai.OpenAI')
     def test_get_chat_response(self, mock_openai):
         mock_client = MagicMock()

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