Update app.py

app.py

@@ -313,29 +313,18 @@ def unified_image_generator(prompt: str, images: Optional[List[str]], previous_v
         # Select model based on radio selection
         model_name = GEMINI_PRO_MODEL_NAME if use_pro_model else GEMINI_MODEL_NAME
         
-        # Create config with aspect ratio (omit image_config if Auto is selected)
-        # Note: PRO model uses image_size instead of aspect_ratio
-        if use_pro_model:
-            # Gemini 3 Pro uses image_size parameter
+        # Create config with aspect ratio
+        if aspect_ratio == "Auto":
+            generate_content_config = types.GenerateContentConfig(
+                response_modalities=["IMAGE", "TEXT"],
+            )
+        else:
             generate_content_config = types.GenerateContentConfig(
                 response_modalities=["IMAGE", "TEXT"],
                 image_config=types.ImageConfig(
-                    image_size="1K",  # Options: "1K", "2K", "4K"
+                    aspect_ratio=aspect_ratio,
                 ),
             )
-        else:
-            # Gemini 2.5 Flash uses aspect_ratio parameter
-            if aspect_ratio == "Auto":
-                generate_content_config = types.GenerateContentConfig(
-                    response_modalities=["IMAGE", "TEXT"],
-                )
-            else:
-                generate_content_config = types.GenerateContentConfig(
-                    response_modalities=["IMAGE", "TEXT"],
-                    image_config=types.ImageConfig(
-                        aspect_ratio=aspect_ratio,
-                    ),
-                )
         
         response = client.models.generate_content(
             model=model_name, 
@@ -416,9 +405,12 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
                 image_input_gallery = gr.Gallery(label="Upload one or more images here. Leave empty for text-to-image", file_types=["image"], height="auto")
                 prompt_input = gr.Textbox(label="Prompt", placeholder="Turns this photo into a masterpiece")
                 
+                # Model selection radio
                 model_radio = gr.Radio(
                     choices=["Nano Banana", "Nano Banana PRO"],
-                    value="Nano Banana PRO",
+                    value="Nano Banana",
+                    label="Model Selection",
+                    info="PRO: Higher quality, slower generation (50/day) | Standard: Faster generation (75/day)"
                 )
                 
                 aspect_ratio_dropdown = gr.Dropdown(