Update app.py

app.py

@@ -1,8 +1,8 @@
 """
-Gradio Space: GPT‑Image‑1 – BYOT playground
+Gradio Space: GPT-Image-1 – BYOT playground
 Generate · Edit (paint mask!) · Variations
 ==========================================
-Adds an **in‑browser paint tool** for the edit / inpaint workflow so users can
+Adds an **in-browser paint tool** for the edit / inpaint workflow so users can
 draw the mask directly instead of uploading one.
 
 ### How mask painting works
@@ -41,8 +41,9 @@ def _client(key: str) -> openai.OpenAI:
     return openai.OpenAI(api_key=api_key)
 
 
-def _img_list(resp, *, fmt: str, transparent: bool) -> List[str]:
-    mime = "image/png" if fmt == "png" or transparent else f"image/{fmt}"
+def _img_list(resp, *, fmt: str) -> List[str]:
+    """Return list of data URLs or direct URLs depending on API response."""
+    mime = f"image/{fmt}"
     return [
         f"data:{mime};base64,{d.b64_json}" if hasattr(d, "b64_json") else d.url
         for d in resp.data
@@ -58,19 +59,27 @@ def _common_kwargs(
     compression: int,
     transparent_bg: bool,
 ):
-    kwargs = dict(
+    """Prepare keyword arguments for Images API based on latest OpenAI spec."""
+    kwargs: Dict[str, Any] = dict(
         model=MODEL,
         n=n,
         size=size,
         quality=quality,
         output_format=out_fmt,
-        transparent_background=transparent_bg,
-        response_format="url" if out_fmt == "png" and not transparent_bg else "b64_json",
     )
+
+    # Prompt is optional for variations
     if prompt is not None:
         kwargs["prompt"] = prompt
+
+    # Transparency via background parameter (png & webp only)
+    if transparent_bg:
+        kwargs["background"] = "transparent"
+
+    # Compression for lossy formats
     if out_fmt in {"jpeg", "webp"}:
-        kwargs["compression"] = f"{compression}%"
+        kwargs["output_compression"] = f"{compression}%"
+
     return kwargs
 
 
@@ -91,7 +100,7 @@ def generate(
         resp = client.images.generate(**_common_kwargs(prompt, n, size, quality, out_fmt, compression, transparent_bg))
     except Exception as e:
         raise gr.Error(f"OpenAI error: {e}")
-    return _img_list(resp, fmt=out_fmt, transparent=transparent_bg)
+    return _img_list(resp, fmt=out_fmt)
 
 
 # ---------- Edit / Inpaint ---------- #
@@ -147,7 +156,7 @@ def edit_image(
     mask_numpy = _extract_mask_array(mask_value)
 
     if mask_numpy is not None:
-        # Convert painted area (any non‑zero pixel) to white, else black; 1‑channel.
+        # Convert painted area (any non-zero pixel) to white, else black; 1‑channel.
         if mask_numpy.shape[-1] == 4:  # RGBA (has alpha channel)
             alpha = mask_numpy[:, :, 3]
         else:  # RGB or grayscale
@@ -164,7 +173,7 @@ def edit_image(
         )
     except Exception as e:
         raise gr.Error(f"OpenAI error: {e}")
-    return _img_list(resp, fmt=out_fmt, transparent=transparent_bg)
+    return _img_list(resp, fmt=out_fmt)
 
 
 # ---------- Variations ---------- #
@@ -190,17 +199,17 @@ def variation_image(
         )
     except Exception as e:
         raise gr.Error(f"OpenAI error: {e}")
-    return _img_list(resp, fmt=out_fmt, transparent=transparent_bg)
+    return _img_list(resp, fmt=out_fmt)
 
 
 # ---------- UI ---------- #
 
 def build_ui():
-    with gr.Blocks(title="GPT‑Image‑1 (BYOT)") as demo:
-        gr.Markdown("""# GPT‑Image‑1 Playground 🖼️🔑\nGenerate • Edit (paint mask) • Variations""")
+    with gr.Blocks(title="GPT-Image-1 (BYOT)") as demo:
+        gr.Markdown("""# GPT-Image-1 Playground 🖼️🔑\nGenerate • Edit (paint mask) • Variations""")
 
         with gr.Accordion("🔐 API key", open=False):
-            api = gr.Textbox(label="OpenAI API key", type="password", placeholder="sk‑…")
+            api = gr.Textbox(label="OpenAI API key", type="password", placeholder="sk-…")
 
         # Common controls
         n_slider = gr.Slider(1, 10, value=1, step=1, label="Number of images (n)")
@@ -208,7 +217,7 @@ def build_ui():
         quality = gr.Dropdown(QUALITY_CHOICES, value="auto", label="Quality")
         out_fmt = gr.Radio(FORMAT_CHOICES, value="png", label="Format")
         compression = gr.Slider(0, 100, value=75, step=1, label="Compression (JPEG/WebP)")
-        transparent = gr.Checkbox(False, label="Transparent background (PNG only)")
+        transparent = gr.Checkbox(False, label="Transparent background (PNG/WebP only)")
 
         def _toggle_compression(fmt):
             return gr.update(visible=fmt in {"jpeg", "webp"})
@@ -242,19 +251,4 @@ def build_ui():
 
             # ----- Variations Tab ----- #
             with gr.TabItem("Variations"):
-                img_var = gr.Image(label="Source image", type="numpy")
-                btn_var = gr.Button("Variations 🔄")
-                gallery_var = gr.Gallery(columns=2, height="auto")
-                btn_var.click(
-                    variation_image,
-                    inputs=[api, img_var, n_slider, size, quality, out_fmt, compression, transparent],
-                    outputs=gallery_var,
-                )
-
-    return demo
-
-
-demo = build_ui()
-
-if __name__ == "__main__":
-    demo.launch()
+                img_var =