updated version of import

app.py

@@ -1,5 +1,13 @@
 from openai import OpenAI
-import google.generativeai as genai
+# Prefer the new Google GenAI SDK if available (import style: `from google import genai`).
+# Fall back to the legacy `google.generativeai` package if that's what's installed.
+try:
+  from google import genai  # new `google-genai` package
+except Exception:
+  try:
+    import google.generativeai as genai  # older, deprecated package
+  except Exception:
+    genai = None
 
 import os
 import requests
@@ -281,11 +289,23 @@ def search(query, progress=gr.Progress()):
 
 if GENAI_API == "openai":
   client = OpenAI()
-elif GENAI_API.lower() in ("gemini", "google", "genai"):
-  genai.configure(api_key=os.getenv('GEMINI_API_KEY'))
-  client = genai
+elif GENAI_API and GENAI_API.lower() in ("gemini", "google", "genai"):
+  # If the genai package couldn't be imported earlier, tell the user.
+  if genai is None:
+    raise RuntimeError(
+      "GENAI_API is set to Gemini but no Google GenAI SDK is installed. "
+      "Install `google-genai` (preferred) or `google-generativeai`, or set GENAI_API=openai."
+    )
+
+  # Prefer the new `genai.Client()` style when available (google-genai SDK).
+  if hasattr(genai, "Client"):
+    client = genai.Client(api_key=os.getenv('GEMINI_API_KEY'))
+  else:
+    # Legacy SDK: configure module-level API key and use the module as client.
+    genai.configure(api_key=os.getenv('GEMINI_API_KEY'))
+    client = genai
 else:
-  # Default to OpenAI client if GENAI_API is unrecognized.
+  # Default to OpenAI client if GENAI_API is unrecognized or unset.
   client = OpenAI()
 
 with gr.Blocks() as demo:

requirements.txt

@@ -2,4 +2,4 @@ gradio
 openai
 requests
 httpx<0.28
-google-generativeai
+google-genai