Update image_gen.py

image_gen.py

@@ -250,6 +250,28 @@ def generate_image(prompt_text, style, model="hf"):
     # Return a placeholder image in case of failure
     return Image.new('RGB', (1024, 1024), color=(200,200,200)), None
 
+def generate_image_with_retry(prompt_text, style, model="hf", max_retries=3):
+    """
+    Attempt to generate an image using generate_image, retrying up to max_retries if needed.
+    Args:
+        prompt_text (str): The text prompt for image generation.
+        style (str or None): The style of the image (ignored for Pollinations Turbo).
+        model (str): Which model to use ("hf" or "pollinations_turbo").
+        max_retries (int): Maximum number of retries.
+    Returns:
+        tuple: The generated image and its Base64 string.
+    """
+    for attempt in range(max_retries):
+        try:
+            if attempt > 0:
+                time.sleep(2 ** attempt)
+            return generate_image(prompt_text, style, model=model)
+        except Exception as e:
+            logger.error(f"Attempt {attempt+1} failed: {e}")
+            if attempt == max_retries - 1:
+                raise
+    return None, None
+
 # edit image function
 def edit_section_image(image_url: str, gemini_prompt: str):
     """