S01Nour commited on
Commit
60af588
·
1 Parent(s): 25481b5

feat: Add logging, FastAPI imports, and integrate Hugging Face generation model manager.

Browse files
Files changed (1) hide show
  1. main.py +14 -0
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}")