Add recipe Gemini proxy service

recipe_service/app.py

@@ -0,0 +1,70 @@
+import base64
+import json
+import os
+
+import requests
+from flask import Flask, jsonify, request
+
+
+app = Flask(__name__)
+MAX_IMAGE_BYTES = 10 * 1024 * 1024
+ALLOWED_TYPES = {"image/jpeg", "image/png", "image/webp"}
+GEMINI_URL = "https://generativelanguage.googleapis.com/v1beta/models"
+
+
+def settings() -> tuple[str, dict]:
+    key = os.environ.get("GEMINI_API_KEY")
+    if not key:
+        raise RuntimeError("GEMINI_API_KEY is not configured")
+    try:
+        config = json.loads(os.environ.get("GEMINI_GENERATION_CONFIG", "{}"))
+    except json.JSONDecodeError as exc:
+        raise RuntimeError("GEMINI_GENERATION_CONFIG is invalid") from exc
+    if not isinstance(config, dict):
+        raise RuntimeError("GEMINI_GENERATION_CONFIG must be an object")
+    return key, config
+
+
+@app.get("/health")
+def health():
+    return jsonify(status="ok")
+
+
+@app.post("/recipe")
+def recipe():
+    image = request.files.get("image")
+    prompt = request.form.get("prompt")
+    if image is None or not prompt:
+        return jsonify(error="image and prompt are required"), 400
+    if image.mimetype not in ALLOWED_TYPES:
+        return jsonify(error="unsupported image type"), 415
+    image_data = image.read(MAX_IMAGE_BYTES + 1)
+    if len(image_data) > MAX_IMAGE_BYTES:
+        return jsonify(error="image is too large"), 413
+    try:
+        key, config = settings()
+        response = requests.post(
+            f"{GEMINI_URL}/{os.environ.get('GEMINI_MODEL', 'gemini-3.6-flash')}:generateContent",
+            params={"key": key},
+            json={
+                "contents": [{"parts": [{"text": prompt}, {"inline_data": {
+                    "mime_type": image.mimetype,
+                    "data": base64.b64encode(image_data).decode("ascii"),
+                }}]}],
+                "generationConfig": config,
+            },
+            timeout=180,
+        )
+    except (requests.RequestException, RuntimeError) as exc:
+        return jsonify(error=str(exc) if isinstance(exc, RuntimeError) else "Gemini unavailable"), 503
+    if response.status_code != 200:
+        try:
+            detail = response.json().get("error", {}).get("message", "Gemini request failed")
+        except ValueError:
+            detail = "Gemini request failed"
+        return jsonify(error=detail), 502
+    data = response.json()
+    return jsonify(
+        text=data.get("candidates", [{}])[0].get("content", {}).get("parts", [{}])[0].get("text"),
+        usage=data.get("usageMetadata", {}),
+    )