Route recipe responses through Gemini proxy
This commit is contained in:
+9
-19
@@ -1,4 +1,3 @@
|
||||
import base64
|
||||
import json
|
||||
import os
|
||||
|
||||
@@ -9,20 +8,17 @@ 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"
|
||||
PROXY_URL = "http://95.179.252.111:8768/gemini"
|
||||
|
||||
|
||||
def settings() -> tuple[str, dict]:
|
||||
key = os.environ.get("GEMINI_API_KEY")
|
||||
if not key:
|
||||
raise RuntimeError("GEMINI_API_KEY is not configured")
|
||||
def settings() -> dict:
|
||||
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
|
||||
return config
|
||||
|
||||
|
||||
def authorized() -> bool:
|
||||
@@ -50,17 +46,11 @@ def recipe():
|
||||
if len(image_data) > MAX_IMAGE_BYTES:
|
||||
return jsonify(error="image is too large"), 413
|
||||
try:
|
||||
key, config = settings()
|
||||
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,
|
||||
},
|
||||
os.environ.get("GEMINI_PROXY_URL", PROXY_URL),
|
||||
files={"image": (image.filename or "image", image_data, image.mimetype)},
|
||||
data={"prompt": prompt, "generation_config": json.dumps(config)},
|
||||
timeout=180,
|
||||
)
|
||||
except (requests.RequestException, RuntimeError) as exc:
|
||||
@@ -73,6 +63,6 @@ def recipe():
|
||||
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", {}),
|
||||
text=data.get("text"),
|
||||
usage=data.get("usage", {}),
|
||||
)
|
||||
Reference in New Issue
Block a user