feat: Adjust supported aspect ratios and add Alfred & Frida examples

app.py

@@ -231,21 +231,25 @@ def refine_with_gemini(prompt):
     except Exception as e:
         return f"Error refining prompt with Gemini: {e}"
 
-def generate_image_with_gemini(refined_prompt, technical_prompt):
+def generate_image_with_gemini(refined_prompt, technical_prompt, aspect_ratio):
     if not gemini_active:
-        return None, "Gemini API key not found in .env file."
+        return None, None, "Gemini API key not found in .env file."
     
     # Priority: Refined Prompt -> Technical Prompt
     final_prompt = refined_prompt.strip() if refined_prompt and refined_prompt.strip() else technical_prompt.strip()
     
     if not final_prompt:
-        return None, "No prompt available for generation."
+        return None, None, "No prompt available for generation."
     
     try:
         # Using the new SDK's generate_images method
         response = client.models.generate_images(
             model='imagen-4.0-generate-001',
-            prompt=final_prompt
+            prompt=final_prompt,
+            config=types.GenerateImagesConfig(
+                aspect_ratio=aspect_ratio,
+                output_mime_type='image/png'
+            )
         )
         if response.generated_images:
             img = Image.open(io.BytesIO(response.generated_images[0].image.image_bytes))
@@ -257,7 +261,7 @@ def generate_image_with_gemini(refined_prompt, technical_prompt):
         return None, None, "Gemini Image generation did not return any images."
     except Exception as e:
         traceback.print_exc()
-        return None, f"Image Generation Error: {e}"
+        return None, None, f"Image Generation Error: {e}"
 
 def build_ui():
     with gr.Blocks(title="RPGPortrait Prompt Builder Pro") as demo:
@@ -389,7 +393,7 @@ def build_ui():
         # Gemini Image Generation
         gen_img_btn.click(
             fn=generate_image_with_gemini,
-            inputs=[refined_output, prompt_output], 
+            inputs=[refined_output, prompt_output, dropdowns[-1]], # The last dropdown is Aspect Ratio
             outputs=[image_output, download_img_btn, status_msg]
         ).then(
             fn=lambda x: gr.update(value=x, visible=True) if x else gr.update(visible=False),

examples/rpg_character_alfred.json

@@ -0,0 +1,63 @@
+{
+    "features": {
+        "race": "Human",
+        "class": "Fighter",
+        "gender": "Male",
+        "age": "Elderly",
+        "expression": "Joyful",
+        "pose": "Action",
+        "hair_color": "Grey",
+        "hair_style": "Messy",
+        "eye_color": "Green",
+        "build": "Scrawny",
+        "skin_tone": "Dark",
+        "distinguishing_feature": "Scars",
+        "armor": "Chainmail",
+        "weapon": "Katana",
+        "accessory": "Feathered Hat",
+        "accessory2": "Silk Scarf",
+        "material": "Polished",
+        "background": "Forest",
+        "lighting": "Dramatic Side-lighting",
+        "atmosphere": "Foggy",
+        "vfx": "Arcane Runes",
+        "style": "Realistic Photo",
+        "mood": "Heroic",
+        "camera": "Dynamic3/4",
+        "aspect_ratio": "3:4"
+    },
+    "randomization": {
+        "race": false,
+        "class": false,
+        "gender": false,
+        "age": false,
+        "expression": false,
+        "pose": false,
+        "hair_color": false,
+        "hair_style": false,
+        "eye_color": false,
+        "build": false,
+        "skin_tone": false,
+        "distinguishing_feature": false,
+        "armor": false,
+        "weapon": false,
+        "accessory": false,
+        "accessory2": false,
+        "material": false,
+        "background": false,
+        "lighting": false,
+        "atmosphere": false,
+        "vfx": false,
+        "style": false,
+        "mood": false,
+        "camera": false,
+        "aspect_ratio": false
+    },
+    "extra_info": {
+        "identity": "",
+        "appearance": "",
+        "equipment": "",
+        "environment": "",
+        "style": ""
+    }
+}

examples/rpg_character_frida.json

@@ -0,0 +1,63 @@
+{
+    "features": {
+        "race": "Human",
+        "class": "Rogue",
+        "gender": "Female",
+        "age": "Young Adult",
+        "expression": "Enigmatic",
+        "pose": "Ready Stance",
+        "hair_color": "Blonde",
+        "hair_style": "Ponytail",
+        "eye_color": "Green",
+        "build": "Athletic",
+        "skin_tone": "Pale",
+        "distinguishing_feature": "Freckles",
+        "armor": "Light Leather Armor",
+        "weapon": "Rapier",
+        "accessory": "Cloak",
+        "accessory2": "Hood",
+        "material": "Weathered",
+        "background": "City Street",
+        "lighting": "Soft Golden Hour",
+        "atmosphere": "Rainy",
+        "vfx": "None",
+        "style": "Realistic Photo",
+        "mood": "Heroic",
+        "camera": "Dynamic3/4",
+        "aspect_ratio": "3:4"
+    },
+    "randomization": {
+        "race": false,
+        "class": false,
+        "gender": false,
+        "age": false,
+        "expression": false,
+        "pose": false,
+        "hair_color": false,
+        "hair_style": false,
+        "eye_color": false,
+        "build": false,
+        "skin_tone": false,
+        "distinguishing_feature": false,
+        "armor": false,
+        "weapon": false,
+        "accessory": false,
+        "accessory2": false,
+        "material": false,
+        "background": false,
+        "lighting": false,
+        "atmosphere": false,
+        "vfx": false,
+        "style": false,
+        "mood": false,
+        "camera": false,
+        "aspect_ratio": false
+    },
+    "extra_info": {
+        "identity": "",
+        "appearance": "",
+        "equipment": "",
+        "environment": "",
+        "style": ""
+    }
+}

features.yaml

@@ -206,8 +206,7 @@ technical:
     "16:9": "--ar 16:9"
     "9:16": "--ar 9:16"
     "4:3": "--ar 4:3"
-    "3:2": "--ar 3:2"
-    "2:3": "--ar 2:3"
+    "3:4": "--ar 3:4"
 
 templates:
   default: "A {mood} {style} portrait of {race} {class}, {gender}, {age}. {build} build, {hair_style} {hair_color} hair, {eye_color} eyes, with {distinguishing_feature}. The character is {expression} while {pose}, {armor}, {weapon}{accessories}. All equipment is {material}. The setting is a {background}, {lighting}, {atmosphere}. {vfx}. {camera}. Highly detailed, 8k resolution. {aspect_ratio}"