Update src/image_gen.py

src/image_gen.py

@@ -2,7 +2,7 @@ from huggingface_hub import InferenceClient
 import os
 import glob
 from collections import defaultdict
-import google.generativeai as genai
+from google import genai
 from tqdm import tqdm
 from huggingface_hub.utils import HfHubHTTPError
 import random
@@ -12,7 +12,7 @@ from dotenv import load_dotenv
 load_dotenv()
 HF_TOKEN = os.getenv("HF_TOKEN")
 GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
-genai.configure(api_key=GOOGLE_API_KEY)
+client_gemini = genai.Client(api_key = GOOGLE_API_KEY)
 client = InferenceClient(provider="hf-inference", api_key=HF_TOKEN)
 
 def split_text_for_images(number_of_images):
@@ -70,8 +70,9 @@ def describe_image(description, detail_level="short", perspective="neutral", emo
     """
 
     try:
-        model = genai.GenerativeModel("gemini-pro")
-        response = model.generate_content(prompt)
+        response = client_gemini.models.generate_content(
+            model = "gemini-2.0-flash", contents = prompt
+        )
         return response.text.strip()
     except Exception as e:
         print(f"Lỗi khi gọi API Gemini: {e}")