Update app/genai_client.py

app/genai_client.py

@@ -1,9 +1,16 @@
-import google.generativeai as genai
+from google import genai
 from app.config import GOOGLE_API_KEY, GENAI_MODEL
 
-genai.configure(api_key=GOOGLE_API_KEY)
-model = genai.GenerativeModel(GENAI_MODEL)
+client = genai.Client(api_key=GOOGLE_API_KEY)
+
+def generate_answer_from_prompt(prompt: str) -> str:
+    response = client.models.generate_content(
+        model=GENAI_MODEL,   # เช่น "gemini-1.5-flash"
+        contents=prompt
+    )
+
+    if not response or not response.text:
+        raise RuntimeError("Empty response from Gemini")
 
-def generate_answer_from_prompt(prompt):
-    response = model.generate_content(prompt)
     return response.text
+