Update backend_models.py

backend_models.py

@@ -1,23 +1,24 @@
 import os
-import google.generativeai as genai
+from google import genai
 
-# Cấu hình API Key cho Gemini
-# Đảm bảo bạn đã thêm biến GEMINI_API_KEY trong phần Settings của Space
-genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
+# Khởi tạo client mới (thay thế cho việc cấu hình global cũ)
+client = genai.Client(api_key=os.getenv("GEMINI_API_KEY"))
 
-def get_gemini_model(model_id: str):
+def get_gemini_model_id(model_id: str) -> str:
     """
-    Khởi tạo model Gemini dựa trên ID được truyền vào.
-    Các ID hợp lệ: 'gemini-1.5-flash', 'gemini-1.5-pro'
+    Chuyển đổi tên hiển thị sang ID model chuẩn cho thư viện google-genai
     """
-    # Nếu người dùng gửi tên thân thiện, ta tự chuyển đổi nhẹ
     mapping = {
-        "Gemini 1.5 Flash": "gemini-1.5-flash",
-        "Gemini 1.5 Pro": "gemini-1.5-pro"
+        "Gemini 3.5 Flash": "gemini-3.5-flash",
+        "Gemini 3.1 Pro Preview": "gemini-3.1-pro-preview",
+        "Gemini 2.0 Flash": "gemini-2.0-flash"
     }
-    actual_id = mapping.get(model_id, "gemini-1.5-flash")
-    
-    return genai.GenerativeModel(actual_id)
+    return mapping.get(model_id, "gemini-3.5-flash")
+
+# Hàm này trả về client để bạn sử dụng trong backend_api.py
+def get_genai_client():
+    return client
+
 
 # Các hàm cũ không còn cần thiết cho Gemini
 def is_native_sdk_model(model_id: str) -> bool: return True