Update app.py

app.py

@@ -3,7 +3,7 @@ import tempfile
 import json
 import re
 import gradio as gr
-from google import genai
+import google.generativeai as genai
 
 print("Google Generative AI SDK version:", genai.__version__)
 
@@ -48,17 +48,20 @@ def process_image(image, prompt):
         print(f"Received image: {type(image)}")
         print(f"Received prompt: {prompt}")
 
-        api_key = get_next_key()
-
-        # INIT CLIENT
-        client = genai.Client(api_key=api_key)
-
-        # Create RAG store
-        RAG_STORE_NAME = "default_rag_store"
-        try:
-            genai.create_rag_store(name=RAG_STORE_NAME)
-        except Exception as e:
-            print("(Info) RAG store exists:", e)
+        api_key = random.choice(API_KEYS)
+        if not api_key:
+            return "ERROR: Missing GOOGLE_API_KEY.", None
+        genai.configure(api_key=api_key)
+        model_name = INTERNAL_MODEL_MAP.get(model_choice, "gemini-2.5-flash")
+        model = genai.GenerativeModel(model_name=model_name,
+                                      generation_config={"temperature": float(0.2), "top_p": float(0.95)})
+
+        # # Create RAG store
+        # RAG_STORE_NAME = "default_rag_store"
+        # try:
+        #     genai.create_rag_store(name=RAG_STORE_NAME)
+        # except Exception as e:
+        #     print("(Info) RAG store exists:", e)
 
         # Create temp file
         with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
@@ -73,7 +76,8 @@ def process_image(image, prompt):
         print("Temp file:", temp_file)
 
         # Upload file
-        uploaded_file = client.files.upload(file=temp_file)
+        #uploaded_file = client.files.upload(file=temp_file)
+        uploaded_file = genai.upload_file(path=temp_file, mime_type="image/jpeg")
         print("Uploaded:", uploaded_file.name)
 
         # === CREATE MODEL USING genai.GenerativeModel ====
@@ -83,15 +87,16 @@ def process_image(image, prompt):
         )
 
         # === GENERATE CONTENT ====
-        response = model.generate_content(
-            contents=[
-                prompt,
-                uploaded_file   # file reference
-            ],
-            request_options={
-                "temperature": 0.2
-            }
-        )
+        response = model.generate_content([prompt, uploaded_file])
+        # response = model.generate_content(
+        #     contents=[
+        #         prompt,
+        #         uploaded_file   # file reference
+        #     ],
+        #     request_options={
+        #         "temperature": 0.2
+        #     }
+        # )
 
         print("Raw response:", response.text[:200], "...")