handler: support image_format/webp/quality and guidance alias for flowcore client compat

worker/handler.py

@@ -122,7 +122,7 @@ def handler(job):
     w, h = max(64, w - w % 16), max(64, h - h % 16)
 
     steps = int(inp.get("num_inference_steps", inp.get("steps", 8)))
-    cfg = float(inp.get("true_cfg_scale", inp.get("cfg", 1.0)))
+    cfg = float(inp.get("true_cfg_scale", inp.get("cfg", inp.get("guidance", 1.0))))
     num_images = min(int(inp.get("num_images", 1)), 4)
     seed = inp.get("seed")
     if seed is None or int(seed) < 0:
@@ -163,12 +163,13 @@ def handler(job):
         ).images
         timings["generate_s"] = round(time.time() - t, 1)
 
-        fmt = str(inp.get("output_format", "png")).lower()
-        fmt = "JPEG" if fmt in ("jpg", "jpeg") else "PNG"
+        fmt = str(inp.get("output_format") or inp.get("image_format") or "png").lower()
+        fmt = {"jpg": "JPEG", "jpeg": "JPEG", "webp": "WEBP"}.get(fmt, "PNG")
+        quality = int(inp.get("image_quality", 95))
         out = []
         for img in images:
             buf = io.BytesIO()
-            img.save(buf, format=fmt, quality=95)
+            img.save(buf, format=fmt, quality=quality)
             out.append(base64.b64encode(buf.getvalue()).decode())
         return {"images": out, "seed": seed, "width": w, "height": h, "timings": timings}
     except Exception as e: