Update app.py

app.py

@@ -10,34 +10,34 @@ import os
 client = genai.Client(api_key=os.environ['GEMINI_API_KEY'])
 
 def generate_item(tag):
-    """
-    Generate a single feed item consisting of text from Gemini LLM and an image from Imagen.
-    
-    Args:
-        tag (str): The tag to base the content on.
-    
-    Returns:
-        dict: A dictionary with 'text' (str) and 'image_base64' (str).
-    """
     # Generate text using Gemini LLM
     prompt = f"Generate a short, engaging post about {tag} in the style of a TikTok caption."
-    text_response = client.models.generate_content(model='gemini-2.5-flash-preview-04-17',contents=[prompt])
+    text_response = client.models.generate_content(
+        model='gemini-2.5-flash-preview-04-17',
+        contents=[prompt]
+    )
     text = text_response.text.strip()
 
-    # Generate an image using Imagen based on the generated text
+    # Generate an image based on the text or tag
     image_response = client.models.generate_images(
         model='imagen-3.0-generate-002',
-        prompt=text,
+        prompt=text,  # Using the generated text as the prompt
         config=types.GenerateImagesConfig(
             number_of_images=1,
-            aspect_ratio="9:16",  # TikTok-style vertical video ratio
-            person_generation="DONT_ALLOW"  # Avoid generating people
+            aspect_ratio="9:16",
+            person_generation="DONT_ALLOW"
         )
     )
-    generated_image = image_response.generated_images[0]
-    image = Image.open(BytesIO(generated_image.image.image_bytes))
 
-    # Convert the image to base64 for HTML display
+    # Check if images were generated
+    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))
+    else:
+        # Fallback to a placeholder image if no images are generated
+        image = Image.new('RGB', (300, 533), color='gray')  # Size matches 9:16 aspect ratio
+
+    # Convert the image to base64
     buffered = BytesIO()
     image.save(buffered, format="PNG")
     img_str = base64.b64encode(buffered.getvalue()).decode()