Add API endpoint to unload models from memory without database deletion

api/management.py

@@ -76,3 +76,43 @@ async def update_single_model(model_db_id: Any):
         if model_db_id in model_pipelines:
             del model_pipelines[model_db_id] # Assurer le déchargement en cas d'erreur imprévue
         raise HTTPException(status_code=500, detail=f"Internal server error while updating model {model_db_id}. Model has been unloaded.")
+
+@router.delete("/unload_model/{model_db_id}", summary="Unload a model from memory")
+async def delete_single_model(model_db_id: Any):
+    """Décharge un modèle de la mémoire sans le supprimer de la base de données.
+    Ceci permet de libérer des ressources système.
+    
+    Args:
+        model_db_id: L'ID du modèle à décharger.
+        
+    Returns:
+        Un message de confirmation du déchargement.
+        
+    Raises:
+        HTTPException: Si le modèle n'est pas trouvé en mémoire.
+    """
+    logger.info(f"Request to unload model {model_db_id} from memory")
+    
+    if model_db_id not in model_pipelines:
+        raise HTTPException(status_code=404, detail=f"Model ID {model_db_id} is not currently loaded.")
+    
+    try:
+        # Récupérer le nom du fichier pour le logging avant de supprimer
+        filename = model_pipelines[model_db_id]['metadata'].get('hf_filename', 'unknown')
+        
+        # Supprimer le modèle de la mémoire
+        del model_pipelines[model_db_id]
+        
+        # Force le garbage collector pour libérer la mémoire
+        import gc
+        gc.collect()
+        
+        logger.info(f"Successfully unloaded model ID {model_db_id} ({filename}) from memory")
+        return {"message": f"Model {model_db_id} successfully unloaded from memory"}
+    
+    except Exception as e:
+        logger.exception(f"Error unloading model {model_db_id}: {e}")
+        # Si une erreur se produit pendant le déchargement, on tente quand même de supprimer
+        if model_db_id in model_pipelines:
+            del model_pipelines[model_db_id]
+        raise HTTPException(status_code=500, detail=f"Internal server error while unloading model {model_db_id}")