Refactor image generation to use API endpoint and remove pollinations dependency

image.py

@@ -3,8 +3,8 @@ import io
 import os
 import re
 import requests
+from urllib.parse import quote
 from PIL import Image
-import pollinations as ai
 
 def extract_summary(text):
     text = text.replace("#", "").strip().lower()
@@ -22,27 +22,14 @@ def fix_base64_padding(data):
 def generate_image(title, summary):
     try:
         extracted_summary = extract_summary(summary)
-        prompt = f"{title.strip()}: {extracted_summary.strip()}"
-        model_obj = ai.ImageModel(
-            model=ai.flux_pro,
-            width=1280,
-            height=720,
-            seed=2342342340,
-            enhance=True,
-            nologo=True
-        )
-        image = model_obj.generate(
-            prompt=prompt,
-            negative="low quality, blurry, pixelated, bad anatomy, bad hands, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, poorly rendered hands, bad face, fused face, cloned face, worst face, three crus, extra crus, fused crus, worst feet, three feet, fused feet, fused thigh, three thigh, extra thigh, worst thigh, missing fingers, extra fingers, ugly fingers, long fingers, bad composition, horn, extra eyes, huge eyes, 2girl, amputation, disconnected limbs, cartoon, cg, 3d, unreal, animate, cgi, render, artwork, illustration, 3d render, cinema 4d, artstation, octane render, mutated body parts, painting, oil painting, 2d, sketch, bad photography, bad photo, deviant art, aberrations, abstract, anime, black and white, collapsed, conjoined, creative, drawing, extra windows, harsh lighting, jpeg artifacts, low saturation, monochrome, multiple levels, overexposed, oversaturated, photoshop, rotten, surreal, twisted, UI, underexposed, unnatural, unreal engine, unrealistic, video game, deformed body features",
-            save=True,
-            file="image.png"
-        )
-        if type(image) == str:
-            print(f"An error occurred during image generation: {image}")
-            return None
-        image_url = image.params.get("url")
-        if image_url:
-            img_data = requests.get(image_url, timeout=60).content
+        negative="low quality, blurry, pixelated, bad anatomy, bad hands, three hands, three legs, bad arms, missing legs, missing arms, poorly drawn face, poorly rendered hands, bad face, fused face, cloned face, worst face, three crus, extra crus, fused crus, worst feet, three feet, fused feet, fused thigh, three thigh, extra thigh, worst thigh, missing fingers, extra fingers, ugly fingers, long fingers, bad composition, horn, extra eyes, huge eyes, 2girl, amputation, disconnected limbs, cartoon, cg, 3d, unreal, animate, cgi, render, artwork, illustration, 3d render, cinema 4d, artstation, octane render, mutated body parts, painting, oil painting, 2d, sketch, bad photography, bad photo, deviant art, aberrations, abstract, anime, black and white, collapsed, conjoined, creative, drawing, extra windows, harsh lighting, jpeg artifacts, low saturation, monochrome, multiple levels, overexposed, oversaturated, photoshop, rotten, surreal, twisted, UI, underexposed, unnatural, unreal engine, unrealistic, video game, deformed body features",
+        image_api_endpoint = "image.pollinations.ai"
+        params = f"negative={quote(str(negative))}&seed={2342342340}&width={1024}&height={576}&nologo={True}&private={False}&model={"turbo"}&enhance={True}"
+        data_header = {"Content-Type": "application/json"}
+        prompt = quote(f"{title.strip()}: {extracted_summary.strip()}")
+        image_url = f"https://{image_api_endpoint}/prompt/{prompt}?{params}"
+        img_data = requests.get(image_url, headers=data_header, timeout=60).content
+        if img_data:
             base64_encoded = base64.b64encode(img_data).decode("utf-8")
             return f"data:image/png;base64,{base64_encoded}"
         return None
@@ -94,4 +81,10 @@ def fetch_image(title, summary, api_key):
     finally:
         if os.path.exists("image.png"):
             os.remove("image.png")
-        return image_url
+        return image_url
+    
+
+if __name__ == "__main__":
+    title = "A beautiful sunset over the mountains"
+    summary = "A beautiful sunset over the mountains with vibrant colors and a clear sky."
+    print(fetch_image(title, summary, "aa38b04047587c609f5c7e22f9d840f0"))

requirements.txt

@@ -5,5 +5,4 @@ mistune==3.0.2
 nbconvert==7.16.4
 requests==2.32.3
 pillow==11.0.0
-pollinations==2.1
 sib-api-v3-sdk==7.6.0