Change model Gemini

app/services/text_normalizer.py

@@ -6,16 +6,30 @@ CACHE_TTL = 60 * 60 * 24 * 3  # 3 days
 # Simple in-memory cache (có thể thay bằng Redis, v.v. sau này)
 # _normalize_cache = {}
 
-# --- Gemini client thực tế (Google GenerativeAI) ---
+# --- Gemini client (use new `google.genai` if available) ---
+try:
+	import google.genai as genai
+	from google.api_core.exceptions import GoogleAPIError  # optional but useful
+except Exception:
+	genai = None
+	class GoogleAPIError(Exception):
+		pass
 
-import google.generativeai as genai
 from app.config.settings import GEMINI_API_KEY
 
-if GEMINI_API_KEY:
-	genai.configure(api_key=GEMINI_API_KEY)
-	_gemini_model = genai.GenerativeModel("gemini-pro")
+_gemini_client = None
+_GEMINI_MODEL = "gemini-1.5-flash"  # hoặc "gemini-1.5-flash-latest"
+
+if GEMINI_API_KEY and genai is not None:
+	try:
+		_gemini_client = genai.Client(api_key=GEMINI_API_KEY)
+	except Exception:
+		_gemini_client = None
+elif GEMINI_API_KEY and genai is None:
+	# package not installed
+	_gemini_client = None
 else:
-	_gemini_model = None
+	_gemini_client = None
 
 async def normalize_text(raw_text: str) -> str:
 	cache_key = f"normalize:{sha256(raw_text)}"
@@ -33,14 +47,25 @@ Văn bản:
 {raw_text}
 """
 	result = raw_text
-	if _gemini_model:
-		# Google GenerativeAI Gemini API (synchronous, wrap in thread for async)
+	if _gemini_client:
 		import asyncio
 		loop = asyncio.get_event_loop()
+
 		def call_gemini():
-			response = _gemini_model.generate_content(prompt)
-			return response.text.strip() if hasattr(response, 'text') else str(response)
-		result = await loop.run_in_executor(None, call_gemini)
+			resp = _gemini_client.models.generate_content(
+				model=_GEMINI_MODEL,
+				contents=prompt,
+			)
+			return resp.text if hasattr(resp, 'text') else str(resp)
+
+		try:
+			result = await loop.run_in_executor(None, call_gemini)
+			if isinstance(result, str):
+				result = result.strip()
+		except GoogleAPIError:
+			result = raw_text
+		except Exception:
+			result = raw_text
 	else:
 		# Nếu chưa cấu hình Gemini, trả về text gốc
 		result = raw_text