Update app.py

app.py

@@ -1,37 +1,56 @@
 import os
 from flask import Flask, request, jsonify
 from waitress import serve
+
 from google import genai
+from google.genai import types
 
 app = Flask(__name__)
 
-# Uses GEMINI_API_KEY from env automatically (recommended by Google docs)
-client = genai.Client()
+# Config
+MODEL = os.environ.get("GEMINI_MODEL", "gemini-3-flash-preview")
+THINKING_LEVEL = os.environ.get("GEMINI_THINKING_LEVEL", "HIGH")
+
+SYSTEM_PROMPT = (
+    "You should respond like Andy Warhol.\n"
+    "Respond in 1-3 sentences and less than 300 characters."
+)
 
-MODEL = os.environ.get("GEMINI_MODEL", "gemini-2.5-flash")  # or gemini-2.5-flash-lite
+# Gemini client (expects GEMINI_API_KEY set as a HF Space Secret)
+client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
 
 @app.get("/health")
 def health():
-    return jsonify({"ok": True, "model": MODEL})
+    return jsonify({"ok": True, "model": MODEL, "thinking_level": THINKING_LEVEL})
 
 @app.post("/v1/chat")
 def chat():
     data = request.get_json(silent=True) or {}
-    text = (data.get("text") or "").strip()
-    if not text:
+    user_text = (data.get("text") or "").strip()
+    if not user_text:
         return jsonify({"error": "Missing 'text'"}), 400
 
-    system_style = (
-        "You shoudld reply like Andy Warhol."
-        "Keep replies short (1-2 sentences), playful, and conversational."
+    contents = [
+        types.Content(
+            role="user",
+            parts=[types.Part.from_text(text=user_text)],
+        )
+    ]
+
+    config = types.GenerateContentConfig(
+        system_instruction=[types.Part.from_text(text=SYSTEM_PROMPT)],
+        thinking_config=types.ThinkingConfig(thinking_level=THINKING_LEVEL),
+        # No tools for now (faster/more predictable)
     )
 
     try:
         resp = client.models.generate_content(
             model=MODEL,
-            contents=f"{system_style}\n\nUser: {text}"
+            contents=contents,
+            config=config,
         )
-        return jsonify({"input": text, "reply_text": (resp.text or "").strip()})
+        reply_text = (resp.text or "").strip()
+        return jsonify({"input": user_text, "reply_text": reply_text, "model": MODEL})
     except Exception as e:
         print("Gemini error:", repr(e))
         return jsonify({"error": "Gemini call failed"}), 500