Update app.py

app.py

@@ -146,50 +146,115 @@ async def v1_chat(request: Request, authorization: str = Header(None)):
         headers={"Access-Control-Allow-Origin": "*"},
     )
 
-#---------------------------------------------------------------------
-#--------------------------/v1 Dall-e---------------------------------
-#---------------------------------------------------------------------
-
+# ---------------------------------------------------------------------
+# --------------------- /v1/images/generations (FIXED) ---------------
+# ---------------------------------------------------------------------
 @api.post("/v1/images/generations")
-async def image_generation(request: Request, authorization: str = Header(None)):
+async def v1_images_generations(request: Request, authorization: str = Header(None)):
+    """
+    Fully OpenAI-compatible DALL·E-3 via Bytez
+    → Accepts `model` field (required by Continue.dev, Cursor, etc.)
+    → Ignores it safely (since Bytez uses URL path, not model name)
+    → Returns proper OpenAI format with url + b64_json
+    """
     check_key(authorization)
-    if not BYTEZ_AUTH:
-        raise HTTPException(status_code=500, detail="Server BYTEZ_API_KEY not configured")
 
-    # Parse incoming OpenAI-style payload
-    payload = await request.json()
-    prompt = payload.get("prompt")
-    if not prompt:
-        raise HTTPException(status_code=400, detail="Missing 'prompt' in request body")
+    if not BYTEZ_AUTH_2:
+        raise HTTPException(status_code=500, detail="Server BYTEZ_API_KEY_2 not configured")
+
+    try:
+        payload = await request.json()
+    except json.JSONDecodeError:
+        raise HTTPException(status_code=400, detail="Invalid JSON")
 
-    # Prepare Bytez API payload
-    bytez_payload = {"text": prompt}
+    prompt = payload.get("prompt")
+    if not prompt or not prompt.strip():
+        raise HTTPException(status_code=400, detail="Field 'prompt' is required and cannot be empty")
+
+    # These fields are REQUIRED by UI tools even if we ignore some
+    model_name = payload.get("model", "dall-e-3")  # just for show — we ignore it
+    n = payload.get("n", 1)
+    size = payload.get("size", "1024x1024")
+    quality = payload.get("quality", "standard")
+    style = payload.get("style")  # vivid or natural
+    response_format = payload.get("response_format", "url")  # url or b64_json
+
+    # Map OpenAI sizes → Bytez accepts the same strings
+    if size not in ["1024x1024", "1024x1792", "1792x1024"]:
+        size = "1024x1024"  # fallback
+
+    bytez_payload = {
+        "text": prompt,
+        "num_outputs": n,
+        "size": size,
+    }
+    if quality in ["standard", "hd"]:
+        bytez_payload["quality"] = quality
+    if style in ["vivid", "natural"]:
+        bytez_payload["style"] = style
 
     headers = {
-        "Authorization": BYTEZ_AUTH,
-        "Content-Type": "application/json"
+        "Authorization": BYTEZ_AUTH_2,
+        "Content-Type": "application/json",
     }
 
-    # Call Bytez image API
-    async with httpx.AsyncClient(timeout=60) as client:
-        r = await client.post(BYTEZ_IMAGE_URL, headers=headers, json=bytez_payload)
-    try:
-        bytez_resp = r.json()
-    except Exception:
-        raise HTTPException(status_code=502, detail="Upstream returned invalid JSON")
-    
-    # Map Bytez output to OpenAI response format
-    # Adjust this according to actual Bytez API response content/structure
-    image_url = bytez_resp.get("url") or bytez_resp.get("image") or bytez_resp.get("data", [{}])[0].get("url")
-    if not image_url:
-        raise HTTPException(status_code=502, detail="Bytez API did not return an image URL")
-
-    # OpenAI's format
-    openai_resp = {
+    async with httpx.AsyncClient(timeout=200) as client:
+        try:
+            resp = await client.post(
+                "https://api.bytez.com/models/v2/openai/dall-e-3",
+                json=bytez_payload,
+                headers=headers,
+            )
+            resp.raise_for_status()
+        except httpx.HTTPStatusError as e:
+            try:
+                error_detail = e.response.json()
+            except:
+                error_detail = e.response.text
+            raise HTTPException(status_code=e.response.status_code, detail=error_detail)
+        except Exception as e:
+            raise HTTPException(status_code=502, detail=f"Bytez unreachable: {str(e)}")
+
+        try:
+            bytez_data = resp.json()
+        except json.JSONDecodeError:
+            raise HTTPException(status_code=502, detail="Bytez returned invalid JSON")
+
+    # Handle different possible response shapes from Bytez
+    images = bytez_data.get("images") or bytez_data.get("data") or []
+    if isinstance(images, str):
+        images = [images]
+    if not images:
+        raise HTTPException(status_code=500, detail="No images returned from Bytez")
+
+    # Build proper OpenAI response
+    openai_images = []
+    for img_data in images:
+        if img_data.startswith("data:image"):
+            b64 = img_data.split("base64,")[-1]
+            url = img_data
+        else:
+            b64 = img_data
+            url = f"data:image/png;base64,{img_data}"
+
+        item = {}
+        if response_format == "b64_json" or response_format is None:
+            item["b64_json"] = b64
+        else:
+            item["url"] = url
+
+        # Optional: include revised_prompt if Bytez returns it
+        if "revised_prompt" in bytez_data:
+            item["revised_prompt"] = bytez_data["revised_prompt"]
+
+        openai_images.append(item)
+
+    final_response = {
         "created": int(time.time()),
-        "data": [{"url": image_url}]
+        "data": openai_images
     }
-    return JSONResponse(openai_resp, headers={"Access-Control-Allow-Origin": "*"})
+
+    return JSONResponse(final_response, headers={"Access-Control-Allow-Origin": "*"})