Refactor generate_image and generate_post_html functions to remove category parameter and improve URL encoding

post_blog.py

@@ -1,7 +1,7 @@
 import re
 import os
 import time
-import base64
+from urllib.parse import quote
 import requests
 import dotenv
 import mistune
@@ -20,20 +20,21 @@ def extract_summary(text):
         return match.group(1).replace("#", "").replace("\n", "").strip()
     return None
 
-def generate_image(title, summary, category):
+def generate_image(title, summary):
     extracted_summary = extract_summary(summary)
     if extracted_summary is None:
         extracted_summary = title
-    url = f"https://image.pollinations.ai/prompt/{title}%20%3A%20{extracted_summary}?width=1280&height=720&seed=623862700&nologo=true&model=turbo"
+    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
 
-def generate_post_html(title, summary, mindmap, category, citation):
+def generate_post_html(title, summary, mindmap, citation):
     title = title.replace("{", r"{{").replace("}", r"}}")
     summary = summary.replace("{", r"{{").replace("}", r"}}")
     mindmap = mindmap.replace("{", r"{{").replace("}", r"}}")
     citation = citation.replace("{", r"{{").replace("}", r"}}")
-    image = generate_image(title, summary, category)
+    image = generate_image(title, summary)
     html_summary = mistune.html(summary)
     post = f"""
 <div>
@@ -71,7 +72,7 @@ height: 80dvh;
 def create_post(title, category, summary, mindmap, citation):
     post_title = title
     post_category = f"{category}"
-    post_body, post_image = generate_post_html(title, summary, mindmap, category, citation)
+    post_body, post_image = generate_post_html(title, summary, mindmap, citation)
     return post_title, post_category, post_body, post_image
 
 def post_post(title, category, body, image):