Update api.py

api.py

@@ -28,16 +28,13 @@ except ImportError as e:
 import google.generativeai as genai
 
 try:
-    from scoring.specificity import load_model
+    from scoring.specificity import load_model as load_specificity_model
 except ImportError as e:
     print(f"Error importing scoring component: {e}")
     print("Please ensure scoring is in your Python path or installed.")
     # You might want to exit or raise a clearer error if imports fail
     raise
 
-status = load_model()
-print(status)
-
 # Configure logging for the API
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
@@ -146,6 +143,21 @@ async def lifespan(app: FastAPI):
             logger.error(f"Failed to configure Gemini client: {e}", exc_info=True)
     else:
         logger.warning("Gemini library not imported. Endpoints requiring Gemini will not work.")
+        
+    # Initialize fine-tuned model for specificity evaluation
+    logger.info("Loading specificity model...")
+    try:
+        from scoring.specificity import load_model as load_specificity_model # Alias pour éviter confusion
+        model_status = load_specificity_model()
+        if model_status:
+            logger.info("Specificity model loaded successfully.")
+        else:
+            logger.error("Failed to load specificity model during startup.")
+            # Décidez si l'application doit échouer au démarrage
+            # raise RuntimeError("Failed to load specificity model.")
+    except Exception as e:
+        logger.error(f"Error loading specificity model during startup: {e}", exc_info=True)
+        # raise RuntimeError("Failed to load specificity model.") from e
 
     yield # API runs here