Update main.py

main.py

@@ -17,9 +17,11 @@ import logging
 import traceback
 from video_gen import create_video
 import zipfile
-from fpdf import FPDF
-
+from fpdf2 import FPDF
 
+pdf = FPDF(auto_page_break=True, margin=15)
+pdf.add_page()
+pdf.set_font("Arial", size=12)
 # Initialize Flask app and CORS
 app = Flask(__name__)
 CORS(app)
@@ -680,13 +682,11 @@ def download_story_archive(story_id):
                     blob = bucket.blob(blob_path)
                     image_data = blob.download_as_bytes()
 
-                    # Save the image to a temp file so FPDF can read it
-                    temp_img_path = f"/tmp/{uuid.uuid4().hex}.jpg"
-                    with open(temp_img_path, "wb") as f:
-                        f.write(image_data)
+                    image_stream = io.BytesIO(image_data)
+    
 
                     # Insert the image at (x=10, y=None => next line), width=100
-                    pdf.image(temp_img_path, x=10, y=None, w=100)
+                    pdf.image(image_stream, x=10, y=None, w=100)
                     # Clean up temp file
                     os.remove(temp_img_path)