Update app.py

app.py

@@ -3,7 +3,8 @@ import tempfile
 import json
 import re
 import gradio as gr
-import google.generativeai as genai
+# import google.generativeai as genai # Layton Luan - edit
+from google import genai              # Layton Luan - edit
 
 # ==== DEBUG VERSION ====
 print("Google Generative AI SDK version:", genai.__version__)
@@ -54,7 +55,10 @@ def process_image(image, prompt):
 
         # === API key rotation ===
         api_key = get_next_key()
-        genai.configure(api_key=api_key)
+
+        # Layton Luan - edit
+        # genai.configure(api_key=api_key)
+        client = genai.Client(api_key=api_key)
 
         # === Ensure RAG store exists ===
         RAG_STORE_NAME = "default_rag_store"
@@ -76,7 +80,10 @@ def process_image(image, prompt):
         print(f"Created temp file: {temp_file}")
 
         # === Upload to Gemini with RAG ===
-        uploaded_file = genai.upload_file(
+
+        # Layton Luan - edit
+        # uploaded_file = genai.upload_file(
+        uploaded_file = client.files.upload(
             path=temp_file,
             display_name=os.path.basename(temp_file),
             mime_type="image/png",
@@ -88,10 +95,21 @@ def process_image(image, prompt):
         model = genai.GenerativeModel("gemini-2.5-flash")
 
         # === Generate content using RAG context ===
-        response = model.generate_content(
-            [prompt],
-            request_options={"rag_store_name": RAG_STORE_NAME},  # ✅ cho phép truy vấn dựa theo store
+
+        # Layton Luan - edit
+        # response = model.generate_content(
+        #     [prompt],
+        #     request_options={"rag_store_name": RAG_STORE_NAME},  # ✅ cho phép truy vấn dựa theo store
+        # )
+        response = client.models.generate_content(
+            model='gemini-2.0-flash',
+            contents=[
+                [prompt],
+                # request_options={"rag_store_name": RAG_STORE_NAME},  # ✅ cho phép truy vấn dựa theo store
+                uploaded_file
+            ]
         )
+        
         print(f"Got response: {response.text[:200]}...")
 
         result = extract_json(response.text)