Update app.py

app.py

@@ -1,21 +1,4 @@
-from huggingface_hub import login
 import os
-import time
-import gradio as gr
-from typing import *  # Ensure that the typing module is imported
-from pillow_heif import register_heif_opener
-register_heif_opener()
-import vision_agent as va
-from vision_agent.tools import register_tool, load_image, owl_v2, overlay_bounding_boxes, save_image
-
-# Perform login using the token
-hf_token = os.getenv("HF_TOKEN")
-login(token=hf_token, add_to_git_credential=True)
-
-import numpy as np
-from PIL import Image
-
-# Set up logging
 import logging
 import sys
 
@@ -34,13 +17,34 @@ def get_logging_level(default_level=logging.INFO):
     else:
         return default_level
 
+# Set up logging before importing the library
 logging_level = get_logging_level()
 logging.basicConfig(stream=sys.stdout, level=logging_level)
 _LOGGER = logging.getLogger(__name__)
 
+# Explicitly set logging level for the vision-agent library
+vision_agent_logger = logging.getLogger('vision_agent')
+vision_agent_logger.setLevel(logging_level)
+
 # Print the logging level to verify it's set correctly
 print(f"Logging level set to: {logging.getLevelName(logging_level)}")
 
+from huggingface_hub import login
+import time
+import gradio as gr
+from typing import *
+from pillow_heif import register_heif_opener
+register_heif_opener()
+import vision_agent as va
+from vision_agent.tools import register_tool, load_image, owl_v2, overlay_bounding_boxes, save_image
+
+# Perform login using the token
+hf_token = os.getenv("HF_TOKEN")
+login(token=hf_token, add_to_git_credential=True)
+
+import numpy as np
+from PIL import Image
+
 def detect_brain_tumor(image, seg_input, debug: bool = True):
     """
     Detects a brain tumor in the given image and returns the annotated image.