Fix test-env initialisation

app.py

@@ -6,13 +6,19 @@ This module serves as the entry point for the Lyrics Analyzer application, which
 uses a system of specialized agents to search for and analyze song lyrics.  
 """
 import os
+
 from loguru import logger
-from Gradio_UI import GradioUI
 from smolagents import LiteLLMModel
 
-from config import setup_logger, load_api_keys, get_model_id, get_gradio_config
-from agents.manager_agent import create_manager_agent
 from agents.single_agent import create_single_agent
+from config import (
+    get_gradio_config,
+    get_model_id,
+    get_ollama_api_base,
+    load_api_keys,
+    setup_logger,
+)
+from Gradio_UI import GradioUI
 
 
 def main():
@@ -28,13 +34,11 @@ def main():
     
     is_test = os.environ.get('SPACE_ID') is None
 
+    # If using Ollama, we need to specify the API base URL
     # Initialize the LLM model based on configuration
     model_id = get_model_id(is_test=is_test)
     logger.info(f"Initializing with model: {model_id}")
-    
-    # If using Ollama, we need to specify the API base URL
-    if "ollama" in model_id:
-        from config import get_ollama_api_base
+    if is_test:
         api_base = get_ollama_api_base()
         logger.info(f"Using Ollama API base: {api_base}")
         model = LiteLLMModel(model_id=model_id, api_base=api_base)
@@ -50,7 +54,6 @@ def main():
     # Determine if we're in test mode (local) or production (HuggingFace)
     # HuggingFace environment has SPACE_ID environment variable
     
-    
     gradio_config = get_gradio_config(is_test)
     
     # Launch with appropriate configuration