logging utils

models/business_logic_utils/response_generation.py

@@ -1,5 +1,8 @@
 import requests
 from .config import API_TIMEOUT, SCENARIOS, SUPPORTED_LANGUAGES
+from streamlit.logger import get_logger
+
+logger = get_logger(__name__)
 
 def check_arguments(model_input: dict) -> None:
     """Check if the input arguments are valid."""
@@ -41,6 +44,6 @@ def generate_sim(model_input: dict, endpoint_url: str, endpoint_bearer_token: st
 
     if response.status_code != 200:
         raise ValueError(f"Error in response: {response.status_code} - {response.text}")
-
+    logger.debug(f"Default response is {response.json()}")
     # Return the raw response as a dictionary
     return response.json()