Update app.py

app.py

@@ -4,11 +4,22 @@ from urllib.parse import quote
 import os
 import openai
 from gradio_client import Client
+import requests
+from PIL import Image
+from io import BytesIO
+
 
 def generate_image_url(keywords):
     truncated_keywords = keywords[:20]
     return f"https://image.pollinations.ai/prompt/{quote(truncated_keywords)}"
 
+def download_image(image_url):
+    response = requests.get(image_url)
+    img = Image.open(BytesIO(response.content))
+    local_image_path = f"/tmp/{image_url.split('/')[-1]}.png"
+    img.save(local_image_path)
+    return local_image_path
+    
 class PDF(FPDF):
     def chapter_title(self, num, label):
         self.set_font('Arial', 'B', 12)
@@ -22,9 +33,10 @@ class PDF(FPDF):
         self.ln(60)
 
     def add_chapter(self, num, title, body, image_url):
+        local_image_path = download_image(image_url)
         self.add_page()
         self.chapter_title(num, title)
-        self.chapter_body(body, image_url)
+        self.chapter_body(body, local_image_path)
 
 def save_as_pdf():
     global chapters