Update app.py

app.py

@@ -1,29 +1,50 @@
 import os
 from flask import Flask, request, jsonify
 from waitress import serve
+import google.generativeai as 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-1.5-flash")
+else:
+    model = None
+
 @app.get("/health")
 def health():
-    return jsonify({"ok": True})
+    return jsonify({"ok": True, "gemini_ready": bool(model)})
 
 @app.post("/v1/echo")
 def echo():
-    """
-    JSON body:
-      { "text": "hello" }
-
-    Returns:
-      { "input": "...", "output": "..." }
-    """
     data = request.get_json(silent=True) or {}
     text = data.get("text", "")
-    # Simple test behavior (replace later with Gemini call)
-    output = f"Andy heard: {text}"
-    return jsonify({"input": text, "output": output})
+    return jsonify({"input": text, "output": f"Andy heard: {text}"})
+
+@app.post("/v1/chat")
+def chat():
+    data = request.get_json(silent=True) or {}
+    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."
+    )
+
+    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})
+    except Exception as e:
+        print("Gemini error:", e)
+        return jsonify({"error": "Gemini call failed"}), 500
 
 if __name__ == "__main__":
-    # HF Spaces expects your server to bind to 0.0.0.0 and the PORT env var if present.
     port = int(os.environ.get("PORT", "7860"))
     serve(app, host="0.0.0.0", port=port)