Update app.py

app.py

@@ -119,18 +119,21 @@ def generate_item(tag, ideas, max_retries=3):
 
         # Attempt to generate the image
         try:
+            # Pass parameters directly or adjust based on the correct API schema
             image_response = client.models.generate_images(
                 model='imagen-3.0-generate-002',
                 prompt=image_prompt,
-                config=types.GenerateContentConfig(
-                    number_of_images=1,
-                    aspect_ratio="9:16",
-                    person_generation="DONT_ALLOW"
-                )
+                number_of_images=1,  # Pass directly instead of in config
+                aspect_ratio="9:16",
+                person_generation="DONT_ALLOW"
             )
             if image_response.generated_images and len(image_response.generated_images) > 0:
                 generated_image = image_response.generated_images[0]
                 image = Image.open(BytesIO(generated_image.image.image_bytes))
+                # Ensure the image matches the desired aspect ratio (9:16 = 0.5625)
+                target_width = 360
+                target_height = int(target_width / 9 * 16)  # 9:16 aspect ratio
+                image = image.resize((target_width, target_height), Image.LANCZOS)
                 # Successfully generated an image, proceed
                 buffered = BytesIO()
                 image.save(buffered, format="PNG")
@@ -558,9 +561,9 @@ with gr.Blocks(
     with gr.Column(elem_classes="gr-form"):
         gr.Markdown("### Create Your Feed")
         tag_input = gr.Textbox(
-            label="Enter Concept or Ideas",  # Renamed label
+            label="Enter Concept or Ideas",
             value="",
-            placeholder="e.g., sushi adventure, neon tech",  # Updated placeholder
+            placeholder="e.g., sushi adventure, neon tech",
             submit_btn=False
         )
         magic_button = gr.Button("✨ Generate Next Item", elem_classes="gr-button")