Update app.py

app.py

@@ -277,30 +277,36 @@ async def v1_images_generations(request: Request, authorization: str = Header(No
     images = images[:n]
 
     openai_data = []
-    for img in images:
-        # If it looks like a data URL already
-        if isinstance(img, str) and img.startswith("data:image"):
-            # Already a data URL; extract base64 if needed
-            b64_part = img.split("base64,", 1)[-1]
-            if response_format == "b64_json":
-                openai_data.append({"b64_json": b64_part})
-            else:
-                openai_data.append({"url": img})
 
-        else:
-            # Raw URL or base64 – we can't be sure, so:
-            if response_format == "b64_json":
-                # Assume it's base64 or treat as a string anyway
-                openai_data.append({"b64_json": str(img)})
-            else:
-                # Assume it's a URL or make it one if needed
-                openai_data.append({"url": str(img)})
+async def fetch_and_b64(url: str) -> str:
+    async with httpx.AsyncClient(timeout=60) as client:
+        r = await client.get(url)
+        r.raise_for_status()
+        return base64.b64encode(r.content).decode("utf-8")
+
+for img in images:
+    # Case 1: Already base64
+    if isinstance(img, str) and img.startswith("data:image"):
+        b64_part = img.split("base64,", 1)[1]
+        openai_data.append({"b64_json": b64_part})
+        continue
+
+    # Case 2: Looks like base64 (no URL)
+    if isinstance(img, str) and not img.startswith("http"):
+        openai_data.append({"b64_json": img})
+        continue
+
+    # Case 3: It is a URL → fetch + convert to base64
+    if isinstance(img, str) and img.startswith("http"):
+        try:
+            b64 = await fetch_and_b64(img)
+            openai_data.append({"b64_json": b64})
+        except Exception as e:
+            raise HTTPException(
+                status_code=502,
+                detail=f"Failed to fetch image from URL returned by Bytez: {str(e)}"
+            )
 
-    result = {
-        "created": int(time.time()),
-        "data": openai_data,
-    }
-    return JSONResponse(result, headers={"Access-Control-Allow-Origin": "*"})