Update app.py

app.py

@@ -170,9 +170,14 @@ def generate_image_openai(
         "model": model,
         "prompt": prompt,
         "size": size_param,
-        "quality": quality,
         "n": n_images,
     }
+
+    # Align with API error: supported values are low, medium, high, auto
+    allowed_qualities = {"low", "medium", "high", "auto"}
+    if quality in allowed_qualities:
+        kwargs["quality"] = quality
+
     if seed is not None:
         kwargs["seed"] = seed
 
@@ -220,9 +225,9 @@ def generate_image_google(
     seed: Optional[int],
 ) -> List[Image.Image]:
     """
-    This assumes a Google / Nano-Banana image-capable model that returns
-    inline image bytes in the response. If your model behaves differently,
-    tweak this function.
+    This assumes a Google / Gemini image-capable model that returns
+    inline image bytes in the response. If your Nano-Banana model behaves
+    differently, tweak this function.
     """
     genai = _configure_google(api_key)
     model = genai.GenerativeModel(google_image_model)
@@ -239,7 +244,6 @@ def generate_image_google(
             generation_config=generation_config or None,
         )
 
-        # Extract images from candidates, but safely
         candidates = getattr(resp, "candidates", []) or []
         for cand in candidates:
             content = getattr(cand, "content", None)
@@ -355,7 +359,7 @@ def run_generation(
             def try_google() -> List[Image.Image]:
                 if not google_key.strip():
                     raise ValueError("Google key missing for Google image generation.")
-                model_id = google_image_model.strip() or "nano-banana-pro"
+                model_id = google_image_model.strip() or "gemini-1.5-flash"
                 return generate_image_google(
                     api_key=google_key.strip(),
                     google_image_model=model_id,
@@ -364,8 +368,8 @@ def run_generation(
                     seed=seed_val,
                 )
 
-            # Try primary provider first, then fallback
             image_model_used = None
+
             try:
                 if primary == "OpenAI":
                     images, image_model_used = try_openai()
@@ -402,7 +406,6 @@ def run_generation(
 
     except Exception as e:
         debug_lines.append(f"Exception: {e}")
-        # Surface the error clearly in text output as well
         return f"Error during generation: {e}", [], "\n".join(debug_lines)
 
 
@@ -499,9 +502,9 @@ Multi-provider creator for the ZEN ecosystem:
                     label="Aspect Ratio / Size",
                 )
                 quality = gr.Dropdown(
-                    ["standard", "hd"],
-                    value="hd",
-                    label="Quality",
+                    ["auto", "low", "medium", "high"],
+                    value="high",
+                    label="Quality (OpenAI)",
                 )
                 n_images = gr.Slider(
                     minimum=1,
@@ -526,9 +529,9 @@ Multi-provider creator for the ZEN ecosystem:
 
             gr.Markdown("### 🧪 Google Image / Text Model Hints")
             google_image_model = gr.Textbox(
-                label="Google Image Model (default: nano-banana-pro)",
-                value="nano-banana-pro",
-                placeholder="e.g. nano-banana-pro or your exact model id",
+                label="Google Image Model (default: gemini-1.5-flash)",
+                value="gemini-1.5-flash",
+                placeholder="e.g. your Nano-Banana model id or another image-capable model",
             )
             google_text_model_hint = gr.Textbox(
                 label="Google Text Model Hint",