Add image verification and error handling in generate_image function

post_blog.py

@@ -2,6 +2,7 @@ import re
 import os
 import time
 from urllib.parse import quote
+import base64
 import requests
 import dotenv
 import mistune
@@ -20,14 +21,50 @@ def extract_summary(text):
         return match.group(1).replace("#", "").replace("\n", "").strip()
     return None
 
+from PIL import Image
+import io
+
+def verify_image(image_data):
+    try:
+        image_stream = io.BytesIO(image_data)
+        image = Image.open(image_stream)
+        image.verify()
+        image_stream.seek(0)
+        image = Image.open(image_stream)
+        return True
+    except Exception as e:
+        print(f"Error verifying image: {e}")
+        return False
+
 def generate_image(title, summary):
+    default_image = "https://i.ibb.co/qdC1nSx/landscape-placeholder-e1608289113759.png"
     extracted_summary = extract_summary(summary)
     if extracted_summary is None:
         extracted_summary = title
     data = quote(f"{title} : {extracted_summary}")
     url = f"https://image.pollinations.ai/prompt/{data}?width=1280&height=720&seed=623862700&nologo=true&model=turbo"
-    requests.get(url).content
-    return url
+    headers = {
+        "accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7",
+        "accept-language": "en-US,en-GB;q=0.9,en;q=0.8",
+        "cache-control": "max-age=0",
+        "priority": "u=0, i",
+        "sec-ch-ua": "\"Google Chrome\";v=\"131\", \"Chromium\";v=\"131\", \"Not_A Brand\";v=\"24\"",
+        "sec-ch-ua-mobile": "?1",
+        "sec-ch-ua-platform": "\"Android\"",
+        "sec-fetch-dest": "document",
+        "sec-fetch-mode": "navigate",
+        "sec-fetch-site": "none",
+        "sec-fetch-user": "?1",
+        "upgrade-insecure-requests": "1",
+    }
+    response = requests.get(url, headers=headers)
+    if response.status_code != 200:
+        print(f"Failed to fetch image. Status code: {response.status_code}")
+        return default_image
+    elif not verify_image(response.content):
+        return default_image
+    else:
+        return url    
 
 def generate_post_html(title, summary, mindmap, citation):
     title = title.replace("{", r"{{").replace("}", r"}}")
@@ -162,4 +199,5 @@ def test(access_key):
                     return status
                 
 if __name__ == '__main__':
-    test(access_key)
+    # test(access_key)
+    print(generate_image("title", "summary summaryyyyyy highlights"))