S01Nour
commited on
Commit
·
60af588
1
Parent(s):
25481b5
feat: Add logging, FastAPI imports, and integrate Hugging Face generation model manager.
Browse files
main.py
CHANGED
|
@@ -4,10 +4,22 @@ import logging
|
|
| 4 |
from contextlib import asynccontextmanager
|
| 5 |
import atexit
|
| 6 |
import shutil
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
# --- Config ---
|
| 8 |
from config import (
|
| 9 |
API_TITLE, API_DESCRIPTION, API_VERSION,
|
| 10 |
HUGGINGFACE_API_KEY, HUGGINGFACE_STANCE_MODEL_ID, HUGGINGFACE_LABEL_MODEL_ID,
|
|
|
|
| 11 |
HOST, PORT, RELOAD,
|
| 12 |
CORS_ORIGINS, CORS_METHODS, CORS_HEADERS, CORS_CREDENTIALS,
|
| 13 |
PRELOAD_MODELS_ON_STARTUP, LOAD_STANCE_MODEL, LOAD_KPA_MODEL,
|
|
@@ -42,9 +54,11 @@ def cleanup_on_exit():
|
|
| 42 |
# --- Import des singletons de services ---
|
| 43 |
stance_model_manager = None
|
| 44 |
kpa_model_manager = None
|
|
|
|
| 45 |
try:
|
| 46 |
from services.stance_model_manager import stance_model_manager
|
| 47 |
from services.label_model_manager import kpa_model_manager
|
|
|
|
| 48 |
logger.info("✓ Gestionnaires de modèles importés")
|
| 49 |
except ImportError as e:
|
| 50 |
logger.warning(f"⚠ Impossible d'importer les gestionnaires de modèles: {e}")
|
|
|
|
| 4 |
from contextlib import asynccontextmanager
|
| 5 |
import atexit
|
| 6 |
import shutil
|
| 7 |
+
import uvicorn
|
| 8 |
+
from fastapi import FastAPI
|
| 9 |
+
from fastapi.middleware.cors import CORSMiddleware
|
| 10 |
+
from fastapi.responses import RedirectResponse
|
| 11 |
+
|
| 12 |
+
# Setup logging
|
| 13 |
+
logging.basicConfig(
|
| 14 |
+
level=logging.INFO,
|
| 15 |
+
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
|
| 16 |
+
)
|
| 17 |
+
logger = logging.getLogger(__name__)
|
| 18 |
# --- Config ---
|
| 19 |
from config import (
|
| 20 |
API_TITLE, API_DESCRIPTION, API_VERSION,
|
| 21 |
HUGGINGFACE_API_KEY, HUGGINGFACE_STANCE_MODEL_ID, HUGGINGFACE_LABEL_MODEL_ID,
|
| 22 |
+
HUGGINGFACE_GENERATE_MODEL_ID,
|
| 23 |
HOST, PORT, RELOAD,
|
| 24 |
CORS_ORIGINS, CORS_METHODS, CORS_HEADERS, CORS_CREDENTIALS,
|
| 25 |
PRELOAD_MODELS_ON_STARTUP, LOAD_STANCE_MODEL, LOAD_KPA_MODEL,
|
|
|
|
| 54 |
# --- Import des singletons de services ---
|
| 55 |
stance_model_manager = None
|
| 56 |
kpa_model_manager = None
|
| 57 |
+
generate_model_manager = None
|
| 58 |
try:
|
| 59 |
from services.stance_model_manager import stance_model_manager
|
| 60 |
from services.label_model_manager import kpa_model_manager
|
| 61 |
+
from services.generate_model_manager import generate_model_manager
|
| 62 |
logger.info("✓ Gestionnaires de modèles importés")
|
| 63 |
except ImportError as e:
|
| 64 |
logger.warning(f"⚠ Impossible d'importer les gestionnaires de modèles: {e}")
|