Update backend_models.py

backend_models.py

@@ -1,21 +1,23 @@
 import os
-from openai import OpenAI
+import google.generativeai as genai
+
+# Cấu hình API Key của bạn
+# Bạn nên lưu key này trong phần 'Secrets' của Hugging Face với tên GEMINI_API_KEY
+genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
 
 def get_inference_client(model_id: str, provider: str = "auto"):
-    # Ép buộc sử dụng endpoint miễn phí của Hugging Face cho mọi model
-    return OpenAI(
-        base_url="https://api-inference.huggingface.co/v1/",
-        api_key=os.getenv("HF_TOKEN")
-    )
+    """
+    Trả về model Gemini để xử lý yêu cầu.
+    """
+    # Gemini 1.5 Flash rất nhanh và thông minh cho tác vụ code
+    model = genai.GenerativeModel('gemini-1.5-flash')
+    return model
 
 def get_real_model_id(model_id: str) -> str:
-    # Luôn luôn trả về model Llama-3.1-8B-Instruct (Miễn phí 100%)
-    # Dù menu có hiển thị Kimi, Qwen... thì thực tế hệ thống sẽ dùng Llama 3.1
-    return "meta-llama/Llama-3.1-8B-Instruct"
+    return "gemini-1.5-flash"
 
 def is_native_sdk_model(model_id: str) -> bool:
-    return False
+    return True
 
 def is_mistral_model(model_id: str) -> bool:
     return False
-