Update app.py

app.py

@@ -5,23 +5,14 @@ from google import genai
 
 app = Flask(__name__)
 
-# ---- Gemini setup ----
-GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
-if GEMINI_API_KEY:
-    genai.configure(api_key=GEMINI_API_KEY)
-    model = genai.GenerativeModel("gemini-2.5-flash")
-else:
-    model = None
+# Uses GEMINI_API_KEY from env automatically (recommended by Google docs)
+client = genai.Client()
+
+MODEL = os.environ.get("GEMINI_MODEL", "gemini-2.5-flash")  # or gemini-2.5-flash-lite
 
 @app.get("/health")
 def health():
-    return jsonify({"ok": True, "gemini_ready": bool(model)})
-
-@app.post("/v1/echo")
-def echo():
-    data = request.get_json(silent=True) or {}
-    text = data.get("text", "")
-    return jsonify({"input": text, "output": f"Andy heard: {text}"})
+    return jsonify({"ok": True, "model": MODEL})
 
 @app.post("/v1/chat")
 def chat():
@@ -29,20 +20,20 @@ def chat():
     text = (data.get("text") or "").strip()
     if not text:
         return jsonify({"error": "Missing 'text'"}), 400
-    if not model:
-        return jsonify({"error": "Gemini not configured. Set GEMINI_API_KEY secret."}), 500
 
     system_style = (
         "You are Andy, a witty pop-artist bust. "
-        "Keep responses short (1-2 sentences), playful, and conversational."
+        "Keep replies short (1-2 sentences), playful, and conversational."
     )
 
     try:
-        resp = model.generate_content(f"{system_style}\n\nUser: {text}")
-        reply_text = (resp.text or "").strip()
-        return jsonify({"input": text, "reply_text": reply_text})
+        resp = client.models.generate_content(
+            model=MODEL,
+            contents=f"{system_style}\n\nUser: {text}"
+        )
+        return jsonify({"input": text, "reply_text": (resp.text or "").strip()})
     except Exception as e:
-        print("Gemini error:", e)
+        print("Gemini error:", repr(e))
         return jsonify({"error": "Gemini call failed"}), 500
 
 if __name__ == "__main__":